diff --git a/tests/regressiontests/serializers_regress/__init__.py b/tests/regressiontests/serializers_regress/__init__.py new file mode 100644 index 00000000000..e69de29bb2d diff --git a/tests/regressiontests/serializers_regress/models.py b/tests/regressiontests/serializers_regress/models.py new file mode 100644 index 00000000000..b5580500227 --- /dev/null +++ b/tests/regressiontests/serializers_regress/models.py @@ -0,0 +1,170 @@ +""" +A test spanning all the capabilities of all the serializers. + +This class sets up a model for each model field type +(except for image types, because of the PIL dependency). +""" + +from django.db import models + +# The following classes are for testing basic data +# marshalling, including NULL values. + +class BooleanData(models.Model): + data = models.BooleanField(null=True) + +class CharData(models.Model): + data = models.CharField(maxlength=30, null=True) + +class DateData(models.Model): + data = models.DateField(null=True) + +class DateTimeData(models.Model): + data = models.DateTimeField(null=True) + +class EmailData(models.Model): + data = models.EmailField(null=True) + +class FileData(models.Model): + data = models.FileField(null=True, upload_to='/foo/bar') + +class FilePathData(models.Model): + data = models.FilePathField(null=True) + +class FloatData(models.Model): + data = models.FloatField(null=True, decimal_places=3, max_digits=5) + +class IntegerData(models.Model): + data = models.IntegerField(null=True) + +# class ImageData(models.Model): +# data = models.ImageField(null=True) + +class IPAddressData(models.Model): + data = models.IPAddressField(null=True) + +class NullBooleanData(models.Model): + data = models.NullBooleanField(null=True) + +class PhoneData(models.Model): + data = models.PhoneNumberField(null=True) + +class PositiveIntegerData(models.Model): + data = models.PositiveIntegerField(null=True) + +class PositiveSmallIntegerData(models.Model): + data = models.PositiveSmallIntegerField(null=True) + +class SlugData(models.Model): + data = models.SlugField(null=True) + +class SmallData(models.Model): + data = models.SmallIntegerField(null=True) + +class TextData(models.Model): + data = models.TextField(null=True) + +class TimeData(models.Model): + data = models.TimeField(null=True) + +class USStateData(models.Model): + data = models.USStateField(null=True) + +class XMLData(models.Model): + data = models.XMLField(null=True) + +# The following test classes are all for validation +# of related objects; in particular, forward, backward, +# and self references. + +class Anchor(models.Model): + """This is a model that can be used as + something for other models to point at""" + + data = models.CharField(maxlength=30) + +class FKData(models.Model): + data = models.ForeignKey(Anchor, null=True) + +class M2MData(models.Model): + data = models.ManyToManyField(Anchor, null=True) + +class O2OData(models.Model): + data = models.OneToOneField(Anchor, null=True) + +class FKSelfData(models.Model): + data = models.ForeignKey('self', null=True) + +class M2MSelfData(models.Model): + data = models.ManyToManyField('self', null=True, symmetrical=False) + +# The following test classes are for validating the +# deserialization of objects that use a user-defined +# field as the primary key. +# Some of these data types have been commented out +# because they can't be used as a primary key on one +# or all database backends. + +class BooleanPKData(models.Model): + data = models.BooleanField(primary_key=True) + +class CharPKData(models.Model): + data = models.CharField(maxlength=30, primary_key=True) + +# class DatePKData(models.Model): +# data = models.DateField(primary_key=True) + +# class DateTimePKData(models.Model): +# data = models.DateTimeField(primary_key=True) + +class EmailPKData(models.Model): + data = models.EmailField(primary_key=True) + +class FilePKData(models.Model): + data = models.FileField(primary_key=True, upload_to='/foo/bar') + +class FilePathPKData(models.Model): + data = models.FilePathField(primary_key=True) + +class FloatPKData(models.Model): + data = models.FloatField(primary_key=True, decimal_places=3, max_digits=5) + +class IntegerPKData(models.Model): + data = models.IntegerField(primary_key=True) + +# class ImagePKData(models.Model): +# data = models.ImageField(primary_key=True) + +class IPAddressPKData(models.Model): + data = models.IPAddressField(primary_key=True) + +class NullBooleanPKData(models.Model): + data = models.NullBooleanField(primary_key=True) + +class PhonePKData(models.Model): + data = models.PhoneNumberField(primary_key=True) + +class PositiveIntegerPKData(models.Model): + data = models.PositiveIntegerField(primary_key=True) + +class PositiveSmallIntegerPKData(models.Model): + data = models.PositiveSmallIntegerField(primary_key=True) + +class SlugPKData(models.Model): + data = models.SlugField(primary_key=True) + +class SmallPKData(models.Model): + data = models.SmallIntegerField(primary_key=True) + +# class TextPKData(models.Model): +# data = models.TextField(primary_key=True) + +# class TimePKData(models.Model): +# data = models.TimeField(primary_key=True) + +class USStatePKData(models.Model): + data = models.USStateField(primary_key=True) + +# class XMLPKData(models.Model): +# data = models.XMLField(primary_key=True) + diff --git a/tests/regressiontests/serializers_regress/tests.py b/tests/regressiontests/serializers_regress/tests.py new file mode 100644 index 00000000000..05f320bfa8f --- /dev/null +++ b/tests/regressiontests/serializers_regress/tests.py @@ -0,0 +1,243 @@ +""" +A test spanning all the capabilities of all the serializers. + +This class defines sample data and a dynamically generated +test case that is capable of testing the capabilities of +the serializers. This includes all valid data values, plus +forward, backwards and self references. +""" + + +import unittest, datetime + +from django.utils.functional import curry +from django.core import serializers +from django.db import transaction +from django.core import management + +from models import * + +# A set of functions that can be used to recreate +# test data objects of various kinds +def data_create(pk, klass, data): + instance = klass(id=pk) + instance.data = data + instance.save() + return instance + +def fk_create(pk, klass, data): + instance = klass(id=pk) + setattr(instance, 'data_id', data) + instance.save() + return instance + +def m2m_create(pk, klass, data): + instance = klass(id=pk) + instance.save() + instance.data = data + return instance + +def o2o_create(pk, klass, data): + instance = klass() + instance.data_id = data + instance.save() + return instance + +def pk_create(pk, klass, data): + instance = klass() + instance.data = data + instance.save() + return instance + +# A set of functions that can be used to compare +# test data objects of various kinds +def data_compare(testcase, pk, klass, data): + instance = klass.objects.get(id=pk) + testcase.assertEqual(data, instance.data, + "Objects with PK=%d not equal; expected '%s', got '%s'" % (pk,data,instance.data)) + +def fk_compare(testcase, pk, klass, data): + instance = klass.objects.get(id=pk) + testcase.assertEqual(data, instance.data_id) + +def m2m_compare(testcase, pk, klass, data): + instance = klass.objects.get(id=pk) + testcase.assertEqual(data, [obj.id for obj in instance.data.all()]) + +def o2o_compare(testcase, pk, klass, data): + instance = klass.objects.get(data=data) + testcase.assertEqual(data, instance.data_id) + +def pk_compare(testcase, pk, klass, data): + instance = klass.objects.get(data=data) + testcase.assertEqual(data, instance.data) + +# Define some data types. Each data type is +# actually a pair of functions; one to create +# and one to compare objects of that type +data_obj = (data_create, data_compare) +fk_obj = (fk_create, fk_compare) +m2m_obj = (m2m_create, m2m_compare) +o2o_obj = (o2o_create, o2o_compare) +pk_obj = (pk_create, pk_compare) + +test_data = [ + # Format: (data type, PK value, Model Class, data) + (data_obj, 1, BooleanData, True), + (data_obj, 2, BooleanData, False), + (data_obj, 10, CharData, "Test Char Data"), + (data_obj, 11, CharData, ""), + (data_obj, 12, CharData, "None"), + (data_obj, 13, CharData, "null"), + (data_obj, 14, CharData, "NULL"), + (data_obj, 15, CharData, None), + (data_obj, 20, DateData, datetime.date(2006,6,16)), + (data_obj, 21, DateData, None), + (data_obj, 30, DateTimeData, datetime.datetime(2006,6,16,10,42,37)), + (data_obj, 31, DateTimeData, None), + (data_obj, 40, EmailData, "hovercraft@example.com"), + (data_obj, 41, EmailData, None), + (data_obj, 50, FileData, 'file:///foo/bar/whiz.txt'), + (data_obj, 51, FileData, None), + (data_obj, 60, FilePathData, "/foo/bar/whiz.txt"), + (data_obj, 61, FilePathData, None), + (data_obj, 70, FloatData, 12.345), + (data_obj, 71, FloatData, -12.345), + (data_obj, 72, FloatData, 0.0), + (data_obj, 73, FloatData, None), + (data_obj, 80, IntegerData, 123456789), + (data_obj, 81, IntegerData, -123456789), + (data_obj, 82, IntegerData, 0), + (data_obj, 83, IntegerData, None), + #(XX, ImageData + (data_obj, 90, IPAddressData, "127.0.0.1"), + (data_obj, 91, IPAddressData, None), + (data_obj, 100, NullBooleanData, True), + (data_obj, 101, NullBooleanData, False), + (data_obj, 102, NullBooleanData, None), + (data_obj, 110, PhoneData, "212-634-5789"), + (data_obj, 111, PhoneData, None), + (data_obj, 120, PositiveIntegerData, 123456789), + (data_obj, 121, PositiveIntegerData, None), + (data_obj, 130, PositiveSmallIntegerData, 12), + (data_obj, 131, PositiveSmallIntegerData, None), + (data_obj, 140, SlugData, "this-is-a-slug"), + (data_obj, 141, SlugData, None), + (data_obj, 150, SmallData, 12), + (data_obj, 151, SmallData, -12), + (data_obj, 152, SmallData, 0), + (data_obj, 153, SmallData, None), + (data_obj, 160, TextData, """This is a long piece of text. +It contains line breaks. +Several of them. +The end."""), + (data_obj, 161, TextData, ""), + (data_obj, 162, TextData, None), + (data_obj, 170, TimeData, datetime.time(10,42,37)), + (data_obj, 171, TimeData, None), + (data_obj, 180, USStateData, "MA"), + (data_obj, 181, USStateData, None), + (data_obj, 190, XMLData, ""), + (data_obj, 191, XMLData, None), + + (data_obj, 300, Anchor, "Anchor 1"), + (data_obj, 301, Anchor, "Anchor 2"), + + (fk_obj, 400, FKData, 300), # Post reference + (fk_obj, 401, FKData, 500), # Pre reference + (fk_obj, 402, FKData, None), # Empty reference + + (m2m_obj, 410, M2MData, []), # Empty set + (m2m_obj, 411, M2MData, [300,301]), # Post reference + (m2m_obj, 412, M2MData, [500,501]), # Pre reference + (m2m_obj, 413, M2MData, [300,301,500,501]), # Pre and Post reference + + (o2o_obj, None, O2OData, 300), # Post reference + (o2o_obj, None, O2OData, 500), # Pre reference + + (fk_obj, 430, FKSelfData, 431), # Pre reference + (fk_obj, 431, FKSelfData, 430), # Post reference + (fk_obj, 432, FKSelfData, None), # Empty reference + + (m2m_obj, 440, M2MSelfData, []), + (m2m_obj, 441, M2MSelfData, []), + (m2m_obj, 442, M2MSelfData, [440, 441]), + (m2m_obj, 443, M2MSelfData, [445, 446]), + (m2m_obj, 444, M2MSelfData, [440, 441, 445, 446]), + (m2m_obj, 445, M2MSelfData, []), + (m2m_obj, 446, M2MSelfData, []), + + (data_obj, 500, Anchor, "Anchor 3"), + (data_obj, 501, Anchor, "Anchor 4"), + + (pk_obj, 601, BooleanPKData, True), + (pk_obj, 602, BooleanPKData, False), + (pk_obj, 610, CharPKData, "Test Char PKData"), +# (pk_obj, 620, DatePKData, datetime.date(2006,6,16)), +# (pk_obj, 630, DateTimePKData, datetime.datetime(2006,6,16,10,42,37)), + (pk_obj, 640, EmailPKData, "hovercraft@example.com"), + (pk_obj, 650, FilePKData, 'file:///foo/bar/whiz.txt'), + (pk_obj, 660, FilePathPKData, "/foo/bar/whiz.txt"), + (pk_obj, 670, FloatPKData, 12.345), + (pk_obj, 671, FloatPKData, -12.345), + (pk_obj, 672, FloatPKData, 0.0), + (pk_obj, 680, IntegerPKData, 123456789), + (pk_obj, 681, IntegerPKData, -123456789), + (pk_obj, 682, IntegerPKData, 0), +# (XX, ImagePKData + (pk_obj, 690, IPAddressPKData, "127.0.0.1"), + (pk_obj, 700, NullBooleanPKData, True), + (pk_obj, 701, NullBooleanPKData, False), + (pk_obj, 710, PhonePKData, "212-634-5789"), + (pk_obj, 720, PositiveIntegerPKData, 123456789), + (pk_obj, 730, PositiveSmallIntegerPKData, 12), + (pk_obj, 740, SlugPKData, "this-is-a-slug"), + (pk_obj, 750, SmallPKData, 12), + (pk_obj, 751, SmallPKData, -12), + (pk_obj, 752, SmallPKData, 0), +# (pk_obj, 760, TextPKData, """This is a long piece of text. +# It contains line breaks. +# Several of them. +# The end."""), +# (pk_obj, 770, TimePKData, datetime.time(10,42,37)), + (pk_obj, 780, USStatePKData, "MA"), +# (pk_obj, 790, XMLPKData, ""), +] + +# Dynamically create serializer tests to ensure that all +# registered serializers are automatically tested. +class SerializerTests(unittest.TestCase): + pass + +def serializerTest(format, self): + # Clear the database first + management.flush(verbosity=0, interactive=False) + + # Create all the objects defined in the test data + objects = [] + transaction.enter_transaction_management() + transaction.managed(True) + for (func, pk, klass, datum) in test_data: + objects.append(func[0](pk, klass, datum)) + transaction.commit() + transaction.leave_transaction_management() + + # Serialize the test database + serialized_data = serializers.serialize(format, objects, indent=2) + + # Flush the database and recreate from the serialized data + management.flush(verbosity=0, interactive=False) + transaction.enter_transaction_management() + transaction.managed(True) + for obj in serializers.deserialize(format, serialized_data): + obj.save() + transaction.commit() + transaction.leave_transaction_management() + + # Assert that the deserialized data is the same + # as the original source + for (func, pk, klass, datum) in test_data: + func[1](self, pk, klass, datum) + +for format in serializers.get_serializer_formats(): + setattr(SerializerTests, 'test_'+format+'_serializer', curry(serializerTest, format))