Updated the test cases based on the CR comments

This commit is contained in:
Ion Scerbatiuc 2015-07-25 07:52:05 -07:00
parent 085c3e8a2b
commit bbd44ae94b

View File

@ -73,84 +73,55 @@ class TestNotRequiredNestedSerializer:
class TestNestedSerializerWithMany: class TestNestedSerializerWithMany:
def setup(self): def setup(self):
class PaymentInfoSerializer(serializers.Serializer):
url = serializers.URLField()
amount = serializers.DecimalField(max_digits=6, decimal_places=2)
class NestedSerializer(serializers.Serializer): class NestedSerializer(serializers.Serializer):
foo = serializers.CharField() example = serializers.IntegerField(max_value=10)
class AcceptRequestSerializer(serializers.Serializer): class TestSerializer(serializers.Serializer):
ERROR_MESSAGE = '`payment_info` is required under this condition' allow_null = NestedSerializer(many=True, allow_null=True)
not_allow_null = NestedSerializer(many=True)
nested = NestedSerializer(many=True) self.Serializer = TestSerializer
payment_info = PaymentInfoSerializer(many=True, allow_null=True)
def validate_payment_info(self, value):
if value is None and not self.context['condition_to_allow_null']:
raise serializers.ValidationError(self.ERROR_MESSAGE)
return value
self.Serializer = AcceptRequestSerializer
def get_serializer(self, data, condition_to_allow_null=True):
return self.Serializer(
data=data,
context={'condition_to_allow_null': condition_to_allow_null})
def test_null_allowed_if_allow_null_is_set(self): def test_null_allowed_if_allow_null_is_set(self):
input_data = { input_data = {
'nested': [{'foo': 'bar'}], 'allow_null': None,
'payment_info': None 'not_allow_null': [{'example': '2'}, {'example': '3'}]
} }
expected_data = { expected_data = {
'nested': [{'foo': 'bar'}], 'allow_null': None,
'payment_info': None 'not_allow_null': [{'example': 2}, {'example': 3}]
} }
serializer = self.get_serializer(input_data) serializer = self.Serializer(data=input_data)
assert serializer.is_valid(), serializer.errors assert serializer.is_valid(), serializer.errors
assert serializer.validated_data == expected_data assert serializer.validated_data == expected_data
def test_null_is_not_allowed_if_allow_null_is_not_set(self): def test_null_is_not_allowed_if_allow_null_is_not_set(self):
input_data = { input_data = {
'nested': None, 'allow_null': None,
'payment_info': None 'not_allow_null': None
} }
serializer = self.get_serializer(input_data) serializer = self.Serializer(data=input_data)
assert not serializer.is_valid() assert not serializer.is_valid()
assert set(serializer.errors) == set(['nested'])
assert serializer.errors['nested'][0] == serializer.error_messages['null'] expected_errors = {'not_allow_null': [serializer.error_messages['null']]}
assert serializer.errors == expected_errors
def test_run_the_field_validation_even_if_the_field_is_null(self): def test_run_the_field_validation_even_if_the_field_is_null(self):
input_data = { class TestSerializer(self.Serializer):
'nested': [{'foo': 'bar'}], validation_was_run = False
'payment_info': None,
}
serializer = self.get_serializer(input_data, condition_to_allow_null=False)
assert not serializer.is_valid() def validate_allow_null(self, value):
assert set(serializer.errors) == set(['payment_info']) TestSerializer.validation_was_run = True
assert serializer.errors['payment_info'][0] == serializer.ERROR_MESSAGE return value
def test_expected_results_if_not_null(self):
input_data = { input_data = {
'nested': [{'foo': 'bar'}], 'allow_null': None,
'payment_info': [ 'not_allow_null': [{'example': 2}]
{'url': 'https://domain.org/api/payment-method/1/', 'amount': '22'},
{'url': 'https://domain.org/api/payment-method/2/', 'amount': '33'},
]
} }
expected_data = { serializer = TestSerializer(data=input_data)
'nested': [{'foo': 'bar'}],
'payment_info': [
{'url': 'https://domain.org/api/payment-method/1/', 'amount': 22},
{'url': 'https://domain.org/api/payment-method/2/', 'amount': 33},
]
}
serializer = self.get_serializer(input_data)
assert serializer.is_valid() assert serializer.is_valid()
assert serializer.validated_data == expected_data assert serializer.validated_data == input_data
assert TestSerializer.validation_was_run