9a7b8acd5a
* Centralize timezone validation into one spot + add serializer validation for schedules and oncall shifts (both public and internal API) * add engine-manage make command
97 lines
3 KiB
Python
97 lines
3 KiB
Python
import pytest
|
|
import pytz
|
|
from rest_framework import serializers
|
|
from rest_framework.exceptions import APIException
|
|
|
|
import common.timezones as tz
|
|
from common.api_helpers.exceptions import BadRequest
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"input,expected",
|
|
[
|
|
("UTC", pytz.timezone("UTC")),
|
|
("asdfasdfasdf", False),
|
|
],
|
|
)
|
|
def test_is_valid_timezone(input, expected):
|
|
assert tz.is_valid_timezone(input) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"input,raises_exception",
|
|
[
|
|
("UTC", False),
|
|
("asdfasdfasdf", True),
|
|
],
|
|
)
|
|
def test_raise_exception_if_not_valid_timezone(input, raises_exception):
|
|
if raises_exception:
|
|
with pytest.raises(BadRequest, match="Invalid timezone"):
|
|
tz.raise_exception_if_not_valid_timezone(input)
|
|
else:
|
|
try:
|
|
tz.raise_exception_if_not_valid_timezone(input)
|
|
except Exception:
|
|
pytest.fail()
|
|
|
|
|
|
def test_raise_exception_if_not_valid_timezone_custom_exception():
|
|
class MyCustomException(APIException):
|
|
"asdfasdf"
|
|
|
|
with pytest.raises(MyCustomException, match="Invalid timezone"):
|
|
tz.raise_exception_if_not_valid_timezone("asdfasfd", exception_class=MyCustomException)
|
|
|
|
|
|
class TestTimeZoneField:
|
|
@pytest.mark.parametrize("tz", pytz.all_timezones)
|
|
def test_valid_timezones(self, tz):
|
|
class MySerializer(serializers.Serializer):
|
|
tz = tz.TimeZoneField()
|
|
|
|
try:
|
|
serializer = MySerializer(data={"tz": tz})
|
|
serializer.is_valid(raise_exception=True)
|
|
|
|
assert serializer.validated_data["tz"] == tz
|
|
except Exception:
|
|
pytest.fail()
|
|
|
|
def test_invalid_timezone(self):
|
|
class MySerializer(serializers.Serializer):
|
|
tz = tz.TimeZoneField()
|
|
|
|
with pytest.raises(serializers.ValidationError, match="Invalid timezone"):
|
|
serializer = MySerializer(data={"tz": "potato"})
|
|
serializer.is_valid(raise_exception=True)
|
|
|
|
def test_it_works_with_allow_null(self):
|
|
class MySerializer(serializers.Serializer):
|
|
tz = tz.TimeZoneField(allow_null=True)
|
|
|
|
try:
|
|
serializer = MySerializer(data={"tz": None})
|
|
serializer.is_valid(raise_exception=True)
|
|
assert serializer.validated_data["tz"] is None
|
|
|
|
serializer = MySerializer(data={"tz": "UTC"})
|
|
serializer.is_valid(raise_exception=True)
|
|
assert serializer.validated_data["tz"] == "UTC"
|
|
except Exception:
|
|
pytest.fail()
|
|
|
|
def test_it_works_with_required(self):
|
|
class MySerializer(serializers.Serializer):
|
|
tz = tz.TimeZoneField(required=True)
|
|
|
|
with pytest.raises(serializers.ValidationError, match="This field is required"):
|
|
serializer = MySerializer(data={})
|
|
serializer.is_valid(raise_exception=True)
|
|
|
|
try:
|
|
serializer = MySerializer(data={"tz": "UTC"})
|
|
serializer.is_valid(raise_exception=True)
|
|
assert serializer.validated_data["tz"] == "UTC"
|
|
except Exception:
|
|
pytest.fail()
|