mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-23 18:13:57 +03:00
412 lines
17 KiB
Python
412 lines
17 KiB
Python
"""
|
|
General serializer field tests.
|
|
"""
|
|
from __future__ import unicode_literals
|
|
import datetime
|
|
|
|
import django
|
|
from django.db import models
|
|
from django.test import TestCase
|
|
from django.core import validators
|
|
from django.utils import unittest
|
|
|
|
from rest_framework import serializers
|
|
|
|
|
|
class TimestampedModel(models.Model):
|
|
added = models.DateTimeField(auto_now_add=True)
|
|
updated = models.DateTimeField(auto_now=True)
|
|
|
|
|
|
class CharPrimaryKeyModel(models.Model):
|
|
id = models.CharField(max_length=20, primary_key=True)
|
|
|
|
|
|
class DateObject(object):
|
|
def __init__(self, date):
|
|
self.date = date
|
|
|
|
|
|
class DateTimeObject(object):
|
|
def __init__(self, date_time):
|
|
self.date_time = date_time
|
|
|
|
|
|
class TimeObject(object):
|
|
def __init__(self, time):
|
|
self.time = time
|
|
|
|
|
|
class TimestampedModelSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = TimestampedModel
|
|
|
|
|
|
class CharPrimaryKeyModelSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = CharPrimaryKeyModel
|
|
|
|
|
|
class DateObjectSerializer(serializers.Serializer):
|
|
date = serializers.DateField()
|
|
|
|
def restore_object(self, attrs, instance=None):
|
|
if instance is not None:
|
|
instance.date = attrs['date']
|
|
return instance
|
|
return DateObject(**attrs)
|
|
|
|
|
|
class DateObjectCustomFormatSerializer(serializers.Serializer):
|
|
date = serializers.DateField(format=("%Y", "%Y -- %m"))
|
|
|
|
def restore_object(self, attrs, instance=None):
|
|
if instance is not None:
|
|
instance.date = attrs['date']
|
|
return instance
|
|
return DateObject(**attrs)
|
|
|
|
|
|
class DateTimeObjectSerializer(serializers.Serializer):
|
|
date_time = serializers.DateTimeField()
|
|
|
|
def restore_object(self, attrs, instance=None):
|
|
if instance is not None:
|
|
instance.date_time = attrs['date_time']
|
|
return instance
|
|
return DateTimeObject(**attrs)
|
|
|
|
|
|
class DateTimeObjectCustomFormatSerializer(serializers.Serializer):
|
|
date_time = serializers.DateTimeField(format=("%Y", "%Y %H:%M"))
|
|
|
|
def restore_object(self, attrs, instance=None):
|
|
if instance is not None:
|
|
instance.date_time = attrs['date_time']
|
|
return instance
|
|
return DateTimeObject(**attrs)
|
|
|
|
|
|
class TimeObjectSerializer(serializers.Serializer):
|
|
time = serializers.TimeField()
|
|
|
|
def restore_object(self, attrs, instance=None):
|
|
if instance is not None:
|
|
instance.time = attrs['time']
|
|
return instance
|
|
return TimeObject(**attrs)
|
|
|
|
|
|
class TimeObjectCustomFormatSerializer(serializers.Serializer):
|
|
time = serializers.TimeField(format=("%H -- %M", "%H%M%S"))
|
|
|
|
def restore_object(self, attrs, instance=None):
|
|
if instance is not None:
|
|
instance.time = attrs['time']
|
|
return instance
|
|
return TimeObject(**attrs)
|
|
|
|
|
|
class TimeFieldModel(models.Model):
|
|
clock = models.TimeField()
|
|
|
|
|
|
class TimeFieldModelSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = TimeFieldModel
|
|
|
|
|
|
class BasicFieldTests(TestCase):
|
|
def test_auto_now_fields_read_only(self):
|
|
"""
|
|
auto_now and auto_now_add fields should be read_only by default.
|
|
"""
|
|
serializer = TimestampedModelSerializer()
|
|
self.assertEquals(serializer.fields['added'].read_only, True)
|
|
|
|
def test_auto_pk_fields_read_only(self):
|
|
"""
|
|
AutoField fields should be read_only by default.
|
|
"""
|
|
serializer = TimestampedModelSerializer()
|
|
self.assertEquals(serializer.fields['id'].read_only, True)
|
|
|
|
def test_non_auto_pk_fields_not_read_only(self):
|
|
"""
|
|
PK fields other than AutoField fields should not be read_only by default.
|
|
"""
|
|
serializer = CharPrimaryKeyModelSerializer()
|
|
self.assertEquals(serializer.fields['id'].read_only, False)
|
|
|
|
|
|
class DateFieldTest(TestCase):
|
|
def test_valid_default_date_input_formats(self):
|
|
serializer = DateObjectSerializer(data={'date': '1984-07-31'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateObjectSerializer(data={'date': '07/31/1984'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateObjectSerializer(data={'date': '07/31/84'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateObjectSerializer(data={'date': 'Jul 31 1984'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateObjectSerializer(data={'date': 'Jul 31, 1984'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateObjectSerializer(data={'date': '31 Jul 1984'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateObjectSerializer(data={'date': '31 Jul 1984'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateObjectSerializer(data={'date': 'July 31 1984'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateObjectSerializer(data={'date': 'July 31, 1984'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateObjectSerializer(data={'date': '31 July 1984'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateObjectSerializer(data={'date': '31 July, 1984'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
def test_valid_custom_date_input_formats(self):
|
|
serializer = DateObjectCustomFormatSerializer(data={'date': '1984'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateObjectCustomFormatSerializer(data={'date': '1984 -- 07'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
def test_wrong_default_date_input_format(self):
|
|
serializer = DateObjectSerializer(data={'date': 'something wrong'})
|
|
self.assertFalse(serializer.is_valid())
|
|
self.assertEquals(serializer.errors, {'date': ['Date has wrong format. Use one of these formats instead: '
|
|
'YYYY-MM-DD; MM/DD/YYYY; MM/DD/YY; [Jan through Dec] DD YYYY; '
|
|
'[Jan through Dec] DD, YYYY; DD [Jan through Dec] YYYY; '
|
|
'DD [Jan through Dec], YYYY; [January through December] DD YYYY; '
|
|
'[January through December] DD, YYYY; DD [January through December] YYYY; '
|
|
'DD [January through December], YYYY']})
|
|
|
|
def test_wrong_custom_date_input_format(self):
|
|
serializer = DateObjectCustomFormatSerializer(data={'date': '07/31/1984'})
|
|
self.assertFalse(serializer.is_valid())
|
|
self.assertEquals(serializer.errors, {'date': ['Date has wrong format. Use one of these formats instead: YYYY; YYYY -- MM']})
|
|
|
|
def test_from_native(self):
|
|
f = serializers.DateField()
|
|
result = f.from_native('1984-07-31')
|
|
|
|
self.assertEqual(datetime.date(1984, 7, 31), result)
|
|
|
|
def test_from_native_datetime_date(self):
|
|
"""
|
|
Make sure from_native() accepts a datetime.date instance.
|
|
"""
|
|
f = serializers.DateField()
|
|
result = f.from_native(datetime.date(1984, 7, 31))
|
|
|
|
self.assertEqual(result, datetime.date(1984, 7, 31))
|
|
|
|
def test_from_native_empty(self):
|
|
f = serializers.DateField()
|
|
result = f.from_native('')
|
|
|
|
self.assertEqual(result, None)
|
|
|
|
def test_from_native_invalid_date(self):
|
|
f = serializers.DateField()
|
|
|
|
try:
|
|
f.from_native('1984-42-31')
|
|
except validators.ValidationError as e:
|
|
self.assertEqual(e.messages, ['Date has wrong format. Use one of these formats instead: '
|
|
'YYYY-MM-DD; MM/DD/YYYY; MM/DD/YY; [Jan through Dec] DD YYYY; '
|
|
'[Jan through Dec] DD, YYYY; DD [Jan through Dec] YYYY; '
|
|
'DD [Jan through Dec], YYYY; [January through December] DD YYYY; '
|
|
'[January through December] DD, YYYY; DD [January through December] YYYY; '
|
|
'DD [January through December], YYYY'])
|
|
else:
|
|
self.fail("ValidationError was not properly raised")
|
|
|
|
|
|
class DateTimeFieldTest(TestCase):
|
|
def test_valid_default_date_time_input_formats(self):
|
|
serializer = DateTimeObjectSerializer(data={'date_time': '1984-07-31 04:31:59'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateTimeObjectSerializer(data={'date_time': '1984-07-31 04:31'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateTimeObjectSerializer(data={'date_time': '1984-07-31'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateTimeObjectSerializer(data={'date_time': '07/31/1984 04:31:59'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateTimeObjectSerializer(data={'date_time': '07/31/1984 04:31'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateTimeObjectSerializer(data={'date_time': '07/31/1984'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateTimeObjectSerializer(data={'date_time': '07/31/84 04:31:59'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateTimeObjectSerializer(data={'date_time': '07/31/84 04:31'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateTimeObjectSerializer(data={'date_time': '07/31/84'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
@unittest.skipUnless(django.VERSION >= (1, 4), "django < 1.4 don't have microseconds in default settings")
|
|
def test_valid_default_date_time_input_formats_for_django_gte_1_4(self):
|
|
serializer = DateTimeObjectSerializer(data={'date_time': '1984-07-31 04:31:59.123456'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateTimeObjectSerializer(data={'date_time': '07/31/1984 04:31:59.123456'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateTimeObjectSerializer(data={'date_time': '07/31/84 04:31:59.123456'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
def test_valid_custom_date_time_input_formats(self):
|
|
serializer = DateTimeObjectCustomFormatSerializer(data={'date_time': '1984'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = DateTimeObjectCustomFormatSerializer(data={'date_time': '1984 04:31'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
@unittest.skipUnless(django.VERSION >= (1, 4), "django < 1.4 don't have microseconds in default settings")
|
|
def test_wrong_default_date_time_input_format_for_django_gte_1_4(self):
|
|
serializer = DateTimeObjectSerializer(data={'date_time': 'something wrong'})
|
|
self.assertFalse(serializer.is_valid())
|
|
self.assertEquals(serializer.errors, {'date_time': ['Datetime has wrong format. Use one of these formats instead: '
|
|
'YYYY-MM-DD HH:MM:SS; YYYY-MM-DD HH:MM:SS.uuuuuu; YYYY-MM-DD HH:MM; '
|
|
'YYYY-MM-DD; MM/DD/YYYY HH:MM:SS; MM/DD/YYYY HH:MM:SS.uuuuuu; '
|
|
'MM/DD/YYYY HH:MM; MM/DD/YYYY; MM/DD/YY HH:MM:SS; '
|
|
'MM/DD/YY HH:MM:SS.uuuuuu; MM/DD/YY HH:MM; MM/DD/YY']})
|
|
|
|
@unittest.skipUnless(django.VERSION < (1, 4), "django >= 1.4 have microseconds in default settings")
|
|
def test_wrong_default_date_time_input_format_for_django_lt_1_4(self):
|
|
serializer = DateTimeObjectSerializer(data={'date_time': 'something wrong'})
|
|
self.assertFalse(serializer.is_valid())
|
|
self.assertEquals(serializer.errors, {'date_time': ['Datetime has wrong format. Use one of these formats instead: '
|
|
'YYYY-MM-DD HH:MM:SS; YYYY-MM-DD HH:MM; YYYY-MM-DD; '
|
|
'MM/DD/YYYY HH:MM:SS; MM/DD/YYYY HH:MM; MM/DD/YYYY; '
|
|
'MM/DD/YY HH:MM:SS; MM/DD/YY HH:MM; MM/DD/YY']})
|
|
|
|
def test_wrong_custom_date_time_input_format(self):
|
|
serializer = DateTimeObjectCustomFormatSerializer(data={'date_time': '07/31/84 04:31'})
|
|
self.assertFalse(serializer.is_valid())
|
|
self.assertEquals(serializer.errors, {'date_time': ['Datetime has wrong format. Use one of these formats instead: YYYY; YYYY HH:MM']})
|
|
|
|
def test_from_native(self):
|
|
f = serializers.DateTimeField()
|
|
result = f.from_native('1984-07-31 04:31')
|
|
|
|
self.assertEqual(datetime.datetime(1984, 7, 31, 4, 31), result)
|
|
|
|
def test_from_native_datetime_datetime(self):
|
|
"""
|
|
Make sure from_native() accepts a datetime.date instance.
|
|
"""
|
|
f = serializers.DateTimeField()
|
|
result = f.from_native(datetime.datetime(1984, 7, 31))
|
|
|
|
self.assertEqual(result, datetime.datetime(1984, 7, 31))
|
|
|
|
def test_from_native_empty(self):
|
|
f = serializers.DateTimeField()
|
|
result = f.from_native('')
|
|
|
|
self.assertEqual(result, None)
|
|
|
|
@unittest.skipUnless(django.VERSION >= (1, 4), "django < 1.4 don't have microseconds in default settings")
|
|
def test_from_native_invalid_datetime_for_django_gte_1_4(self):
|
|
f = serializers.DateTimeField()
|
|
|
|
try:
|
|
f.from_native('1984-42-31 04:31')
|
|
except validators.ValidationError as e:
|
|
self.assertEqual(e.messages, ['Datetime has wrong format. Use one of these formats instead: '
|
|
'YYYY-MM-DD HH:MM:SS; YYYY-MM-DD HH:MM:SS.uuuuuu; YYYY-MM-DD HH:MM; '
|
|
'YYYY-MM-DD; MM/DD/YYYY HH:MM:SS; MM/DD/YYYY HH:MM:SS.uuuuuu; '
|
|
'MM/DD/YYYY HH:MM; MM/DD/YYYY; MM/DD/YY HH:MM:SS; '
|
|
'MM/DD/YY HH:MM:SS.uuuuuu; MM/DD/YY HH:MM; MM/DD/YY'])
|
|
else:
|
|
self.fail("ValidationError was not properly raised")
|
|
|
|
@unittest.skipUnless(django.VERSION < (1, 4), "django >= 1.4 have microseconds in default settings")
|
|
def test_from_native_invalid_datetime_for_django_lt_1_4(self):
|
|
f = serializers.DateTimeField()
|
|
|
|
try:
|
|
f.from_native('1984-42-31 04:31')
|
|
except validators.ValidationError as e:
|
|
self.assertEqual(e.messages, ['Datetime has wrong format. Use one of these formats instead: '
|
|
'YYYY-MM-DD HH:MM:SS; YYYY-MM-DD HH:MM; YYYY-MM-DD; '
|
|
'MM/DD/YYYY HH:MM:SS; MM/DD/YYYY HH:MM; MM/DD/YYYY; '
|
|
'MM/DD/YY HH:MM:SS; MM/DD/YY HH:MM; MM/DD/YY'])
|
|
else:
|
|
self.fail("ValidationError was not properly raised")
|
|
|
|
|
|
class TimeFieldTest(TestCase):
|
|
def test_valid_default_time_input_formats(self):
|
|
serializer = TimeObjectSerializer(data={'time': '04:31'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = TimeObjectSerializer(data={'time': '04:31:59'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
def test_valid_custom_time_input_formats(self):
|
|
serializer = TimeObjectCustomFormatSerializer(data={'time': '04 -- 31'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
serializer = TimeObjectCustomFormatSerializer(data={'time': '043159'})
|
|
self.assertTrue(serializer.is_valid())
|
|
|
|
def test_wrong_default_time_input_format(self):
|
|
serializer = TimeObjectSerializer(data={'time': 'something wrong'})
|
|
self.assertFalse(serializer.is_valid())
|
|
self.assertEquals(serializer.errors, {'time': ['Time has wrong format. Use one of these formats instead: HH:MM:SS; HH:MM']})
|
|
|
|
def test_wrong_custom_time_input_format(self):
|
|
serializer = TimeObjectCustomFormatSerializer(data={'time': '04:31'})
|
|
self.assertFalse(serializer.is_valid())
|
|
self.assertEquals(serializer.errors, {'time': ['Time has wrong format. Use one of these formats instead: HH -- MM; HHMMSS']})
|
|
|
|
def test_from_native(self):
|
|
f = serializers.TimeField()
|
|
result = f.from_native('12:34:56')
|
|
|
|
self.assertEqual(datetime.time(12, 34, 56), result)
|
|
|
|
def test_from_native_datetime_time(self):
|
|
"""
|
|
Make sure from_native() accepts a datetime.time instance.
|
|
"""
|
|
f = serializers.TimeField()
|
|
result = f.from_native(datetime.time(12, 34, 56))
|
|
|
|
self.assertEqual(result, datetime.time(12, 34, 56))
|
|
|
|
def test_from_native_empty(self):
|
|
f = serializers.TimeField()
|
|
result = f.from_native('')
|
|
|
|
self.assertEqual(result, None)
|
|
|
|
def test_from_native_invalid_time(self):
|
|
f = serializers.TimeField()
|
|
|
|
try:
|
|
f.from_native('12:69:12')
|
|
except validators.ValidationError as e:
|
|
self.assertEqual(e.messages, ["Time has wrong format. Use one of these formats instead: HH:MM:SS; HH:MM"])
|
|
else:
|
|
self.fail("ValidationError was not properly raised") |