mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-09-21 11:29:01 +03:00
93 lines
3.5 KiB
Python
93 lines
3.5 KiB
Python
# from __future__ import unicode_literals
|
|
# from django.test import TestCase
|
|
# from django.utils import six
|
|
# from rest_framework import serializers
|
|
# from rest_framework.compat import BytesIO
|
|
# import datetime
|
|
|
|
|
|
# class UploadedFile(object):
|
|
# def __init__(self, file=None, created=None):
|
|
# self.file = file
|
|
# self.created = created or datetime.datetime.now()
|
|
|
|
|
|
# class UploadedFileSerializer(serializers.Serializer):
|
|
# file = serializers.FileField(required=False)
|
|
# created = serializers.DateTimeField()
|
|
|
|
# def restore_object(self, attrs, instance=None):
|
|
# if instance:
|
|
# instance.file = attrs['file']
|
|
# instance.created = attrs['created']
|
|
# return instance
|
|
# return UploadedFile(**attrs)
|
|
|
|
|
|
# class FileSerializerTests(TestCase):
|
|
# def test_create(self):
|
|
# now = datetime.datetime.now()
|
|
# file = BytesIO(six.b('stuff'))
|
|
# file.name = 'stuff.txt'
|
|
# file.size = len(file.getvalue())
|
|
# serializer = UploadedFileSerializer(data={'created': now}, files={'file': file})
|
|
# uploaded_file = UploadedFile(file=file, created=now)
|
|
# self.assertTrue(serializer.is_valid())
|
|
# self.assertEqual(serializer.object.created, uploaded_file.created)
|
|
# self.assertEqual(serializer.object.file, uploaded_file.file)
|
|
# self.assertFalse(serializer.object is uploaded_file)
|
|
|
|
# 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()
|
|
|
|
# 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.assertEqual(serializer.errors, {'file': [errmsg]})
|
|
|
|
# def test_validation_with_no_data(self):
|
|
# """
|
|
# Validation should still function when no data dictionary is provided.
|
|
# """
|
|
# uploaded_file = BytesIO(six.b('stuff'))
|
|
# uploaded_file.name = 'stuff.txt'
|
|
# uploaded_file.size = len(uploaded_file.getvalue())
|
|
# serializer = UploadedFileSerializer(files={'file': uploaded_file})
|
|
# self.assertFalse(serializer.is_valid())
|