diff options
| author | Xavier Ordoquy | 2014-04-17 09:53:44 +0200 | 
|---|---|---|
| committer | Xavier Ordoquy | 2014-04-17 09:53:44 +0200 | 
| commit | 1797a74e828c9fdfbfb46bb4de049100b18db875 (patch) | |
| tree | a91e8f4ff22e8eb63a990a439995632e9198d1c3 /tests/test_validation.py | |
| parent | 1d404874b3f0f5b16f3b38ba322f31a18c41aad6 (diff) | |
| parent | 971578ca345c3d3bae7fd93b87c41d43483b6f05 (diff) | |
| download | django-rest-framework-1797a74e828c9fdfbfb46bb4de049100b18db875.tar.bz2 | |
Merge remote-tracking branch 'pelme/pytest' into feature/pytest
Conflicts:
	.travis.yml
	rest_framework/runtests/runtests.py
	tests/test_filters.py
	tests/test_pagination.py
	tox.ini
Diffstat (limited to 'tests/test_validation.py')
| -rw-r--r-- | tests/test_validation.py | 148 | 
1 files changed, 148 insertions, 0 deletions
| diff --git a/tests/test_validation.py b/tests/test_validation.py new file mode 100644 index 00000000..e13e4078 --- /dev/null +++ b/tests/test_validation.py @@ -0,0 +1,148 @@ +from __future__ import unicode_literals +from django.core.validators import MaxValueValidator +from django.db import models +from django.test import TestCase +from rest_framework import generics, serializers, status +from rest_framework.test import APIRequestFactory + +factory = APIRequestFactory() + + +# Regression for #666 + +class ValidationModel(models.Model): +    blank_validated_field = models.CharField(max_length=255) + + +class ValidationModelSerializer(serializers.ModelSerializer): +    class Meta: +        model = ValidationModel +        fields = ('blank_validated_field',) +        read_only_fields = ('blank_validated_field',) + + +class UpdateValidationModel(generics.RetrieveUpdateDestroyAPIView): +    model = ValidationModel +    serializer_class = ValidationModelSerializer + + +class TestPreSaveValidationExclusions(TestCase): +    def test_pre_save_validation_exclusions(self): +        """ +        Somewhat weird test case to ensure that we don't perform model +        validation on read only fields. +        """ +        obj = ValidationModel.objects.create(blank_validated_field='') +        request = factory.put('/', {}, format='json') +        view = UpdateValidationModel().as_view() +        response = view(request, pk=obj.pk).render() +        self.assertEqual(response.status_code, status.HTTP_200_OK) + + +# Regression for #653 + +class ShouldValidateModel(models.Model): +    should_validate_field = models.CharField(max_length=255) + + +class ShouldValidateModelSerializer(serializers.ModelSerializer): +    renamed = serializers.CharField(source='should_validate_field', required=False) + +    def validate_renamed(self, attrs, source): +        value = attrs[source] +        if len(value) < 3: +            raise serializers.ValidationError('Minimum 3 characters.') +        return attrs + +    class Meta: +        model = ShouldValidateModel +        fields = ('renamed',) + + +class TestPreSaveValidationExclusionsSerializer(TestCase): +    def test_renamed_fields_are_model_validated(self): +        """ +        Ensure fields with 'source' applied do get still get model validation. +        """ +        # We've set `required=False` on the serializer, but the model +        # does not have `blank=True`, so this serializer should not validate. +        serializer = ShouldValidateModelSerializer(data={'renamed': ''}) +        self.assertEqual(serializer.is_valid(), False) +        self.assertIn('renamed', serializer.errors) +        self.assertNotIn('should_validate_field', serializer.errors) + + +class TestCustomValidationMethods(TestCase): +    def test_custom_validation_method_is_executed(self): +        serializer = ShouldValidateModelSerializer(data={'renamed': 'fo'}) +        self.assertFalse(serializer.is_valid()) +        self.assertIn('renamed', serializer.errors) + +    def test_custom_validation_method_passing(self): +        serializer = ShouldValidateModelSerializer(data={'renamed': 'foo'}) +        self.assertTrue(serializer.is_valid()) + + +class ValidationSerializer(serializers.Serializer): +    foo = serializers.CharField() + +    def validate_foo(self, attrs, source): +        raise serializers.ValidationError("foo invalid") + +    def validate(self, attrs): +        raise serializers.ValidationError("serializer invalid") + + +class TestAvoidValidation(TestCase): +    """ +    If serializer was initialized with invalid data (None or non dict-like), it +    should avoid validation layer (validate_<field> and validate methods) +    """ +    def test_serializer_errors_has_only_invalid_data_error(self): +        serializer = ValidationSerializer(data='invalid data') +        self.assertFalse(serializer.is_valid()) +        self.assertDictEqual(serializer.errors, +                             {'non_field_errors': ['Invalid data']}) + + +# regression tests for issue: 1493 + +class ValidationMaxValueValidatorModel(models.Model): +    number_value = models.PositiveIntegerField(validators=[MaxValueValidator(100)]) + + +class ValidationMaxValueValidatorModelSerializer(serializers.ModelSerializer): +    class Meta: +        model = ValidationMaxValueValidatorModel + + +class UpdateMaxValueValidationModel(generics.RetrieveUpdateDestroyAPIView): +    model = ValidationMaxValueValidatorModel +    serializer_class = ValidationMaxValueValidatorModelSerializer + + +class TestMaxValueValidatorValidation(TestCase): + +    def test_max_value_validation_serializer_success(self): +        serializer = ValidationMaxValueValidatorModelSerializer(data={'number_value': 99}) +        self.assertTrue(serializer.is_valid()) + +    def test_max_value_validation_serializer_fails(self): +        serializer = ValidationMaxValueValidatorModelSerializer(data={'number_value': 101}) +        self.assertFalse(serializer.is_valid()) +        self.assertDictEqual({'number_value': ['Ensure this value is less than or equal to 100.']}, serializer.errors) + +    def test_max_value_validation_success(self): +        obj = ValidationMaxValueValidatorModel.objects.create(number_value=100) +        request = factory.patch('/{0}'.format(obj.pk), {'number_value': 98}, format='json') +        view = UpdateMaxValueValidationModel().as_view() +        response = view(request, pk=obj.pk).render() +        self.assertEqual(response.status_code, status.HTTP_200_OK) + +    def test_max_value_validation_fail(self): +        obj = ValidationMaxValueValidatorModel.objects.create(number_value=100) +        request = factory.patch('/{0}'.format(obj.pk), {'number_value': 101}, format='json') +        view = UpdateMaxValueValidationModel().as_view() +        response = view(request, pk=obj.pk).render() +        self.assertEqual(response.content, b'{"number_value": ["Ensure this value is less than or equal to 100."]}') +        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST) | 
