diff --git a/tests/serializers/models/__init__.py b/tests/serializers/models/__init__.py new file mode 100644 index 0000000000..a786b54be9 --- /dev/null +++ b/tests/serializers/models/__init__.py @@ -0,0 +1,3 @@ +from .base import * # NOQA +from .data import * # NOQA +from .natural import * # NOQA diff --git a/tests/serializers/models/base.py b/tests/serializers/models/base.py new file mode 100644 index 0000000000..4889377e09 --- /dev/null +++ b/tests/serializers/models/base.py @@ -0,0 +1,178 @@ +""" +Serialization + +``django.core.serializers`` provides interfaces to converting Django +``QuerySet`` objects to and from "flat" data (i.e. strings). +""" +from __future__ import unicode_literals + +from decimal import Decimal + +from django.db import models +from django.utils import six +from django.utils.encoding import python_2_unicode_compatible + + +class CategoryMetaDataManager(models.Manager): + + def get_by_natural_key(self, kind, name): + return self.get(kind=kind, name=name) + + +@python_2_unicode_compatible +class CategoryMetaData(models.Model): + kind = models.CharField(max_length=10) + name = models.CharField(max_length=10) + value = models.CharField(max_length=10) + objects = CategoryMetaDataManager() + + class Meta: + unique_together = (('kind', 'name'),) + + def __str__(self): + return '[%s:%s]=%s' % (self.kind, self.name, self.value) + + def natural_key(self): + return (self.kind, self.name) + + +@python_2_unicode_compatible +class Category(models.Model): + name = models.CharField(max_length=20) + meta_data = models.ForeignKey(CategoryMetaData, models.SET_NULL, null=True, default=None) + + class Meta: + ordering = ('name',) + + def __str__(self): + return self.name + + +@python_2_unicode_compatible +class Author(models.Model): + name = models.CharField(max_length=20) + + class Meta: + ordering = ('name',) + + def __str__(self): + return self.name + + +@python_2_unicode_compatible +class Article(models.Model): + author = models.ForeignKey(Author, models.CASCADE) + headline = models.CharField(max_length=50) + pub_date = models.DateTimeField() + categories = models.ManyToManyField(Category) + meta_data = models.ManyToManyField(CategoryMetaData) + + class Meta: + ordering = ('pub_date',) + + def __str__(self): + return self.headline + + +@python_2_unicode_compatible +class AuthorProfile(models.Model): + author = models.OneToOneField(Author, models.CASCADE, primary_key=True) + date_of_birth = models.DateField() + + def __str__(self): + return "Profile of %s" % self.author + + +@python_2_unicode_compatible +class Actor(models.Model): + name = models.CharField(max_length=20, primary_key=True) + + class Meta: + ordering = ('name',) + + def __str__(self): + return self.name + + +@python_2_unicode_compatible +class Movie(models.Model): + actor = models.ForeignKey(Actor, models.CASCADE) + title = models.CharField(max_length=50) + price = models.DecimalField(max_digits=6, decimal_places=2, default=Decimal('0.00')) + + class Meta: + ordering = ('title',) + + def __str__(self): + return self.title + + +class Score(models.Model): + score = models.FloatField() + + +@python_2_unicode_compatible +class Team(object): + def __init__(self, title): + self.title = title + + def __str__(self): + raise NotImplementedError("Not so simple") + + def to_string(self): + return "%s" % self.title + + +class TeamField(models.CharField): + + def __init__(self): + super(TeamField, self).__init__(max_length=100) + + def get_db_prep_save(self, value, connection): + return six.text_type(value.title) + + def to_python(self, value): + if isinstance(value, Team): + return value + return Team(value) + + def from_db_value(self, value, expression, connection, context): + return Team(value) + + def value_to_string(self, obj): + return self.value_from_object(obj).to_string() + + def deconstruct(self): + name, path, args, kwargs = super(TeamField, self).deconstruct() + del kwargs['max_length'] + return name, path, args, kwargs + + +@python_2_unicode_compatible +class Player(models.Model): + name = models.CharField(max_length=50) + rank = models.IntegerField() + team = TeamField() + + def __str__(self): + return '%s (%d) playing for %s' % (self.name, self.rank, self.team.to_string()) + + +class BaseModel(models.Model): + parent_data = models.IntegerField() + + +class ProxyBaseModel(BaseModel): + class Meta: + proxy = True + + +class ProxyProxyBaseModel(ProxyBaseModel): + class Meta: + proxy = True + + +class ComplexModel(models.Model): + field1 = models.CharField(max_length=10) + field2 = models.CharField(max_length=10) + field3 = models.CharField(max_length=10) diff --git a/tests/serializers/models.py b/tests/serializers/models/data.py similarity index 59% rename from tests/serializers/models.py rename to tests/serializers/models/data.py index 023a2e8f85..98d59622d7 100644 --- a/tests/serializers/models.py +++ b/tests/serializers/models/data.py @@ -1,213 +1,19 @@ -# -*- coding: utf-8 -*- """ -Serialization - -``django.core.serializers`` provides interfaces to converting Django -``QuerySet`` objects to and from "flat" data (i.e. strings). +******** Models for test_data.py *********** +The following classes are for testing basic data marshalling, including +NULL values, where allowed. +The basic idea is to have a model for each Django data type. """ from __future__ import unicode_literals -from decimal import Decimal - from django.contrib.contenttypes.fields import ( GenericForeignKey, GenericRelation, ) from django.contrib.contenttypes.models import ContentType from django.db import models -from django.utils import six -from django.utils.encoding import python_2_unicode_compatible +from .base import BaseModel -class CategoryMetaDataManager(models.Manager): - - def get_by_natural_key(self, kind, name): - return self.get(kind=kind, name=name) - - -@python_2_unicode_compatible -class CategoryMetaData(models.Model): - kind = models.CharField(max_length=10) - name = models.CharField(max_length=10) - value = models.CharField(max_length=10) - objects = CategoryMetaDataManager() - - class Meta: - unique_together = (('kind', 'name'),) - - def __str__(self): - return '[%s:%s]=%s' % (self.kind, self.name, self.value) - - def natural_key(self): - return (self.kind, self.name) - - -@python_2_unicode_compatible -class Category(models.Model): - name = models.CharField(max_length=20) - meta_data = models.ForeignKey(CategoryMetaData, models.SET_NULL, null=True, default=None) - - class Meta: - ordering = ('name',) - - def __str__(self): - return self.name - - -@python_2_unicode_compatible -class Author(models.Model): - name = models.CharField(max_length=20) - - class Meta: - ordering = ('name',) - - def __str__(self): - return self.name - - -@python_2_unicode_compatible -class Article(models.Model): - author = models.ForeignKey(Author, models.CASCADE) - headline = models.CharField(max_length=50) - pub_date = models.DateTimeField() - categories = models.ManyToManyField(Category) - meta_data = models.ManyToManyField(CategoryMetaData) - - class Meta: - ordering = ('pub_date',) - - def __str__(self): - return self.headline - - -@python_2_unicode_compatible -class AuthorProfile(models.Model): - author = models.OneToOneField(Author, models.CASCADE, primary_key=True) - date_of_birth = models.DateField() - - def __str__(self): - return "Profile of %s" % self.author - - -@python_2_unicode_compatible -class Actor(models.Model): - name = models.CharField(max_length=20, primary_key=True) - - class Meta: - ordering = ('name',) - - def __str__(self): - return self.name - - -@python_2_unicode_compatible -class Movie(models.Model): - actor = models.ForeignKey(Actor, models.CASCADE) - title = models.CharField(max_length=50) - price = models.DecimalField(max_digits=6, decimal_places=2, default=Decimal('0.00')) - - class Meta: - ordering = ('title',) - - def __str__(self): - return self.title - - -class Score(models.Model): - score = models.FloatField() - - -@python_2_unicode_compatible -class Team(object): - def __init__(self, title): - self.title = title - - def __str__(self): - raise NotImplementedError("Not so simple") - - def to_string(self): - return "%s" % self.title - - -class TeamField(models.CharField): - - def __init__(self): - super(TeamField, self).__init__(max_length=100) - - def get_db_prep_save(self, value, connection): - return six.text_type(value.title) - - def to_python(self, value): - if isinstance(value, Team): - return value - return Team(value) - - def from_db_value(self, value, expression, connection, context): - return Team(value) - - def value_to_string(self, obj): - return self.value_from_object(obj).to_string() - - def deconstruct(self): - name, path, args, kwargs = super(TeamField, self).deconstruct() - del kwargs['max_length'] - return name, path, args, kwargs - - -@python_2_unicode_compatible -class Player(models.Model): - name = models.CharField(max_length=50) - rank = models.IntegerField() - team = TeamField() - - def __str__(self): - return '%s (%d) playing for %s' % (self.name, self.rank, self.team.to_string()) - - -class BaseModel(models.Model): - parent_data = models.IntegerField() - - -class ProxyBaseModel(BaseModel): - class Meta: - proxy = True - - -class ProxyProxyBaseModel(ProxyBaseModel): - class Meta: - proxy = True - - -class ComplexModel(models.Model): - field1 = models.CharField(max_length=10) - field2 = models.CharField(max_length=10) - field3 = models.CharField(max_length=10) - - -# ******** Models for test_natural.py *********** - -class NaturalKeyAnchorManager(models.Manager): - def get_by_natural_key(self, data): - return self.get(data=data) - - -class NaturalKeyAnchor(models.Model): - objects = NaturalKeyAnchorManager() - - data = models.CharField(max_length=100, unique=True) - title = models.CharField(max_length=100, null=True) - - def natural_key(self): - return (self.data,) - - -class FKDataNaturalKey(models.Model): - data = models.ForeignKey(NaturalKeyAnchor, models.SET_NULL, null=True) - - -# ******** Models for test_data.py *********** -# The following classes are for testing basic data marshalling, including -# NULL values, where allowed. -# The basic idea is to have a model for each Django data type. class BinaryData(models.Model): data = models.BinaryField(null=True) @@ -490,10 +296,6 @@ class InheritAbstractModel(AbstractBaseModel): child_data = models.IntegerField() -class BaseModel(models.Model): - parent_data = models.IntegerField() - - class InheritBaseModel(BaseModel): child_data = models.IntegerField() diff --git a/tests/serializers/models/natural.py b/tests/serializers/models/natural.py new file mode 100644 index 0000000000..b50956692e --- /dev/null +++ b/tests/serializers/models/natural.py @@ -0,0 +1,21 @@ +"""Models for test_natural.py""" +from django.db import models + + +class NaturalKeyAnchorManager(models.Manager): + def get_by_natural_key(self, data): + return self.get(data=data) + + +class NaturalKeyAnchor(models.Model): + objects = NaturalKeyAnchorManager() + + data = models.CharField(max_length=100, unique=True) + title = models.CharField(max_length=100, null=True) + + def natural_key(self): + return (self.data,) + + +class FKDataNaturalKey(models.Model): + data = models.ForeignKey(NaturalKeyAnchor, models.SET_NULL, null=True)