django-rest-framework/rest_framework/tests/test_files.py

52 lines
1.7 KiB
Python
Raw Normal View History

from __future__ import unicode_literals
2012-11-15 02:04:46 +04:00
from django.test import TestCase
from rest_framework import serializers
from rest_framework.compat import BytesIO
from rest_framework.compat import six
import datetime
2012-11-15 02:04:46 +04:00
class UploadedFile(object):
def __init__(self, file, created=None):
self.file = file
self.created = created or datetime.datetime.now()
2012-11-15 02:04:46 +04:00
class UploadedFileSerializer(serializers.Serializer):
file = serializers.FileField()
created = serializers.DateTimeField()
2012-11-15 02:04:46 +04:00
def restore_object(self, attrs, instance=None):
if instance:
instance.file = attrs['file']
instance.created = attrs['created']
return instance
return UploadedFile(**attrs)
2012-11-15 02:04:46 +04:00
class FileSerializerTests(TestCase):
def test_create(self):
now = datetime.datetime.now()
file = BytesIO(six.b('stuff'))
2012-11-15 02:04:46 +04:00
file.name = 'stuff.txt'
file.size = len(file.getvalue())
2012-11-15 02:04:46 +04:00
serializer = UploadedFileSerializer(data={'created': now}, files={'file': file})
uploaded_file = UploadedFile(file=file, created=now)
self.assertTrue(serializer.is_valid())
2013-02-28 01:15:00 +04:00
self.assertEqual(serializer.object.created, uploaded_file.created)
self.assertEqual(serializer.object.file, uploaded_file.file)
2012-11-15 02:04:46 +04:00
self.assertFalse(serializer.object is uploaded_file)
2013-01-03 00:07:23 +04:00
def test_creation_failure(self):
"""
Passing files=None should result in an ValidationError
Regression test for:
https://github.com/tomchristie/django-rest-framework/issues/542
"""
now = datetime.datetime.now()
2013-01-03 10:28:17 +04:00
serializer = UploadedFileSerializer(data={'created': now})
2013-01-03 00:07:23 +04:00
self.assertFalse(serializer.is_valid())
self.assertIn('file', serializer.errors)