diff options
Diffstat (limited to 'rest_framework/tests/test_files.py')
| -rw-r--r-- | rest_framework/tests/test_files.py | 37 | 
1 files changed, 34 insertions, 3 deletions
| diff --git a/rest_framework/tests/test_files.py b/rest_framework/tests/test_files.py index 487046ac..c13c38b8 100644 --- a/rest_framework/tests/test_files.py +++ b/rest_framework/tests/test_files.py @@ -7,13 +7,13 @@ import datetime  class UploadedFile(object): -    def __init__(self, file, created=None): +    def __init__(self, file=None, created=None):          self.file = file          self.created = created or datetime.datetime.now()  class UploadedFileSerializer(serializers.Serializer): -    file = serializers.FileField() +    file = serializers.FileField(required=False)      created = serializers.DateTimeField()      def restore_object(self, attrs, instance=None): @@ -47,5 +47,36 @@ class FileSerializerTests(TestCase):          now = datetime.datetime.now()          serializer = UploadedFileSerializer(data={'created': now}) +        self.assertTrue(serializer.is_valid()) +        self.assertEqual(serializer.object.created, now) +        self.assertIsNone(serializer.object.file) + +    def test_remove_with_empty_string(self): +        """ +        Passing empty string as data should cause file to be removed + +        Test for: +        https://github.com/tomchristie/django-rest-framework/issues/937 +        """ +        now = datetime.datetime.now() +        file = BytesIO(six.b('stuff')) +        file.name = 'stuff.txt' +        file.size = len(file.getvalue()) + +        uploaded_file = UploadedFile(file=file, created=now) + +        serializer = UploadedFileSerializer(instance=uploaded_file, data={'created': now, 'file': ''}) +        self.assertTrue(serializer.is_valid()) +        self.assertEqual(serializer.object.created, uploaded_file.created) +        self.assertIsNone(serializer.object.file) + +    def test_validation_error_with_non_file(self): +        """ +        Passing non-files should raise a validation error. +        """ +        now = datetime.datetime.now() +        errmsg = 'No file was submitted. Check the encoding type on the form.' + +        serializer = UploadedFileSerializer(data={'created': now, 'file': 'abc'})          self.assertFalse(serializer.is_valid()) -        self.assertIn('file', serializer.errors) +        self.assertEqual(serializer.errors, {'file': [errmsg]}) | 
