1304 lines
57 KiB
Python
1304 lines
57 KiB
Python
# -*- coding: utf-8 -*-
|
|
""" Test Cases
|
|
Please see README.rst or DOCS.rst or http://chrisglass.github.com/django_polymorphic/
|
|
"""
|
|
from __future__ import print_function
|
|
import uuid
|
|
import re
|
|
import django
|
|
try:
|
|
from unittest import skipIf
|
|
except ImportError:
|
|
# python<2.7
|
|
from django.utils.unittest import skipIf
|
|
from django.db.models.query import QuerySet
|
|
|
|
from django.test import TestCase
|
|
from django.db.models import Q, Count
|
|
if django.VERSION >= (1, 8):
|
|
from django.db.models import Case, When
|
|
from django.db import models
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.utils import six
|
|
|
|
from polymorphic.models import PolymorphicModel
|
|
from polymorphic.managers import PolymorphicManager
|
|
from polymorphic.query import PolymorphicQuerySet
|
|
from polymorphic.showfields import ShowFieldContent, ShowFieldType, ShowFieldTypeAndContent
|
|
try:
|
|
from django.db.models import UUIDField
|
|
except ImportError:
|
|
# django<1.8
|
|
from polymorphic.tools_for_tests import UUIDField
|
|
|
|
|
|
class PlainA(models.Model):
|
|
field1 = models.CharField(max_length=10)
|
|
|
|
|
|
class PlainB(PlainA):
|
|
field2 = models.CharField(max_length=10)
|
|
|
|
|
|
class PlainC(PlainB):
|
|
field3 = models.CharField(max_length=10)
|
|
|
|
|
|
class Model2A(ShowFieldType, PolymorphicModel):
|
|
field1 = models.CharField(max_length=10)
|
|
polymorphic_showfield_deferred = True
|
|
|
|
|
|
class Model2B(Model2A):
|
|
field2 = models.CharField(max_length=10)
|
|
|
|
|
|
class Model2C(Model2B):
|
|
field3 = models.CharField(max_length=10)
|
|
|
|
|
|
class Model2D(Model2C):
|
|
field4 = models.CharField(max_length=10)
|
|
|
|
|
|
class ModelExtraA(ShowFieldTypeAndContent, PolymorphicModel):
|
|
field1 = models.CharField(max_length=10)
|
|
|
|
|
|
class ModelExtraB(ModelExtraA):
|
|
field2 = models.CharField(max_length=10)
|
|
|
|
|
|
class ModelExtraC(ModelExtraB):
|
|
field3 = models.CharField(max_length=10)
|
|
|
|
|
|
class ModelExtraExternal(models.Model):
|
|
topic = models.CharField(max_length=10)
|
|
|
|
|
|
class ModelShow1(ShowFieldType, PolymorphicModel):
|
|
field1 = models.CharField(max_length=10)
|
|
m2m = models.ManyToManyField('self')
|
|
|
|
|
|
class ModelShow2(ShowFieldContent, PolymorphicModel):
|
|
field1 = models.CharField(max_length=10)
|
|
m2m = models.ManyToManyField('self')
|
|
|
|
|
|
class ModelShow3(ShowFieldTypeAndContent, PolymorphicModel):
|
|
field1 = models.CharField(max_length=10)
|
|
m2m = models.ManyToManyField('self')
|
|
|
|
|
|
class ModelShow1_plain(PolymorphicModel):
|
|
field1 = models.CharField(max_length=10)
|
|
|
|
|
|
class ModelShow2_plain(ModelShow1_plain):
|
|
field2 = models.CharField(max_length=10)
|
|
|
|
|
|
class Base(ShowFieldType, PolymorphicModel):
|
|
field_b = models.CharField(max_length=10)
|
|
|
|
|
|
class ModelX(Base):
|
|
field_x = models.CharField(max_length=10)
|
|
|
|
|
|
class ModelY(Base):
|
|
field_y = models.CharField(max_length=10)
|
|
|
|
|
|
class Enhance_Plain(models.Model):
|
|
field_p = models.CharField(max_length=10)
|
|
|
|
|
|
class Enhance_Base(ShowFieldTypeAndContent, PolymorphicModel):
|
|
base_id = models.AutoField(primary_key=True)
|
|
field_b = models.CharField(max_length=10)
|
|
|
|
|
|
class Enhance_Inherit(Enhance_Base, Enhance_Plain):
|
|
field_i = models.CharField(max_length=10)
|
|
|
|
|
|
class RelationBase(ShowFieldTypeAndContent, PolymorphicModel):
|
|
field_base = models.CharField(max_length=10)
|
|
fk = models.ForeignKey('self', null=True, related_name='relationbase_set')
|
|
m2m = models.ManyToManyField('self')
|
|
|
|
|
|
class RelationA(RelationBase):
|
|
field_a = models.CharField(max_length=10)
|
|
|
|
|
|
class RelationB(RelationBase):
|
|
field_b = models.CharField(max_length=10)
|
|
|
|
|
|
class RelationBC(RelationB):
|
|
field_c = models.CharField(max_length=10)
|
|
|
|
|
|
class RelatingModel(models.Model):
|
|
many2many = models.ManyToManyField(Model2A)
|
|
|
|
|
|
class One2OneRelatingModel(PolymorphicModel):
|
|
one2one = models.OneToOneField(Model2A)
|
|
field1 = models.CharField(max_length=10)
|
|
|
|
|
|
class One2OneRelatingModelDerived(One2OneRelatingModel):
|
|
field2 = models.CharField(max_length=10)
|
|
|
|
|
|
class ModelUnderRelParent(PolymorphicModel):
|
|
field1 = models.CharField(max_length=10)
|
|
_private = models.CharField(max_length=10)
|
|
|
|
|
|
class ModelUnderRelChild(PolymorphicModel):
|
|
parent = models.ForeignKey(ModelUnderRelParent, related_name='children')
|
|
_private2 = models.CharField(max_length=10)
|
|
|
|
|
|
class MyManagerQuerySet(PolymorphicQuerySet):
|
|
|
|
def my_queryset_foo(self):
|
|
return self.all() # Just a method to prove the existance of the custom queryset.
|
|
|
|
|
|
class MyManager(PolymorphicManager):
|
|
queryset_class = MyManagerQuerySet
|
|
|
|
def get_queryset(self):
|
|
return super(MyManager, self).get_queryset().order_by('-field1')
|
|
|
|
def my_queryset_foo(self):
|
|
return self.all().my_queryset_foo()
|
|
|
|
# Django <= 1.5 compatibility
|
|
get_query_set = get_queryset
|
|
|
|
|
|
class ModelWithMyManager(ShowFieldTypeAndContent, Model2A):
|
|
objects = MyManager()
|
|
field4 = models.CharField(max_length=10)
|
|
|
|
|
|
class ModelWithMyManagerNoDefault(ShowFieldTypeAndContent, Model2A):
|
|
objects = PolymorphicManager()
|
|
my_objects = MyManager()
|
|
field4 = models.CharField(max_length=10)
|
|
|
|
class ModelWithMyManagerDefault(ShowFieldTypeAndContent, Model2A):
|
|
my_objects = MyManager()
|
|
objects = PolymorphicManager()
|
|
field4 = models.CharField(max_length=10)
|
|
|
|
|
|
if django.VERSION >= (1, 7):
|
|
class ModelWithMyManager2(ShowFieldTypeAndContent, Model2A):
|
|
objects = MyManagerQuerySet.as_manager()
|
|
field4 = models.CharField(max_length=10)
|
|
|
|
|
|
class MROBase1(ShowFieldType, PolymorphicModel):
|
|
objects = MyManager()
|
|
field1 = models.CharField(max_length=10) # needed as MyManager uses it
|
|
|
|
|
|
class MROBase2(MROBase1):
|
|
pass # Django vanilla inheritance does not inherit MyManager as _default_manager here
|
|
|
|
|
|
class MROBase3(models.Model):
|
|
objects = PolymorphicManager()
|
|
|
|
|
|
class MRODerived(MROBase2, MROBase3):
|
|
pass
|
|
|
|
|
|
class ParentModelWithManager(PolymorphicModel):
|
|
pass
|
|
|
|
|
|
class ChildModelWithManager(PolymorphicModel):
|
|
# Also test whether foreign keys receive the manager:
|
|
fk = models.ForeignKey(ParentModelWithManager, related_name='childmodel_set')
|
|
objects = MyManager()
|
|
|
|
|
|
class PlainMyManagerQuerySet(QuerySet):
|
|
|
|
def my_queryset_foo(self):
|
|
return self.all() # Just a method to prove the existance of the custom queryset.
|
|
|
|
|
|
class PlainMyManager(models.Manager):
|
|
|
|
def my_queryset_foo(self):
|
|
return self.get_queryset().my_queryset_foo()
|
|
|
|
def get_queryset(self):
|
|
return PlainMyManagerQuerySet(self.model, using=self._db)
|
|
|
|
# Django <= 1.5 compatibility
|
|
get_query_set = get_queryset
|
|
|
|
|
|
class PlainParentModelWithManager(models.Model):
|
|
pass
|
|
|
|
|
|
class PlainChildModelWithManager(models.Model):
|
|
fk = models.ForeignKey(PlainParentModelWithManager, related_name='childmodel_set')
|
|
objects = PlainMyManager()
|
|
|
|
|
|
class MgrInheritA(models.Model):
|
|
mgrA = models.Manager()
|
|
mgrA2 = models.Manager()
|
|
field1 = models.CharField(max_length=10)
|
|
|
|
|
|
class MgrInheritB(MgrInheritA):
|
|
mgrB = models.Manager()
|
|
field2 = models.CharField(max_length=10)
|
|
|
|
|
|
class MgrInheritC(ShowFieldTypeAndContent, MgrInheritB):
|
|
pass
|
|
|
|
|
|
class BlogBase(ShowFieldTypeAndContent, PolymorphicModel):
|
|
name = models.CharField(max_length=10)
|
|
|
|
|
|
class BlogA(BlogBase):
|
|
info = models.CharField(max_length=10)
|
|
|
|
|
|
class BlogB(BlogBase):
|
|
pass
|
|
|
|
|
|
class BlogEntry(ShowFieldTypeAndContent, PolymorphicModel):
|
|
blog = models.ForeignKey(BlogA)
|
|
text = models.CharField(max_length=10)
|
|
|
|
|
|
class BlogEntry_limit_choices_to(ShowFieldTypeAndContent, PolymorphicModel):
|
|
blog = models.ForeignKey(BlogBase)
|
|
text = models.CharField(max_length=10)
|
|
|
|
|
|
class ModelFieldNameTest(ShowFieldType, PolymorphicModel):
|
|
modelfieldnametest = models.CharField(max_length=10)
|
|
|
|
|
|
class InitTestModel(ShowFieldType, PolymorphicModel):
|
|
bar = models.CharField(max_length=100)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
kwargs['bar'] = self.x()
|
|
super(InitTestModel, self).__init__(*args, **kwargs)
|
|
|
|
|
|
class InitTestModelSubclass(InitTestModel):
|
|
|
|
def x(self):
|
|
return 'XYZ'
|
|
|
|
# models from github issue
|
|
|
|
|
|
class Top(PolymorphicModel):
|
|
name = models.CharField(max_length=50)
|
|
|
|
class Meta:
|
|
ordering = ('pk',)
|
|
|
|
|
|
class Middle(Top):
|
|
description = models.TextField()
|
|
|
|
|
|
class Bottom(Middle):
|
|
author = models.CharField(max_length=50)
|
|
|
|
|
|
class UUIDProject(ShowFieldTypeAndContent, PolymorphicModel):
|
|
uuid_primary_key = UUIDField(primary_key=True, default=uuid.uuid1)
|
|
topic = models.CharField(max_length=30)
|
|
|
|
|
|
class UUIDArtProject(UUIDProject):
|
|
artist = models.CharField(max_length=30)
|
|
|
|
|
|
class UUIDResearchProject(UUIDProject):
|
|
supervisor = models.CharField(max_length=30)
|
|
|
|
|
|
class UUIDPlainA(models.Model):
|
|
uuid_primary_key = UUIDField(primary_key=True, default=uuid.uuid1)
|
|
field1 = models.CharField(max_length=10)
|
|
|
|
|
|
class UUIDPlainB(UUIDPlainA):
|
|
field2 = models.CharField(max_length=10)
|
|
|
|
|
|
class UUIDPlainC(UUIDPlainB):
|
|
field3 = models.CharField(max_length=10)
|
|
|
|
# base -> proxy
|
|
|
|
|
|
class ProxyBase(PolymorphicModel):
|
|
some_data = models.CharField(max_length=128)
|
|
|
|
|
|
class ProxyChild(ProxyBase):
|
|
|
|
class Meta:
|
|
proxy = True
|
|
|
|
|
|
class NonProxyChild(ProxyBase):
|
|
name = models.CharField(max_length=10)
|
|
|
|
# base -> proxy -> real models
|
|
|
|
|
|
class ProxiedBase(ShowFieldTypeAndContent, PolymorphicModel):
|
|
name = models.CharField(max_length=10)
|
|
|
|
|
|
class ProxyModelBase(ProxiedBase):
|
|
|
|
class Meta:
|
|
proxy = True
|
|
|
|
|
|
class ProxyModelA(ProxyModelBase):
|
|
field1 = models.CharField(max_length=10)
|
|
|
|
|
|
class ProxyModelB(ProxyModelBase):
|
|
field2 = models.CharField(max_length=10)
|
|
|
|
|
|
# test bad field name
|
|
# class TestBadFieldModel(ShowFieldType, PolymorphicModel):
|
|
# instance_of = models.CharField(max_length=10)
|
|
|
|
# validation error: "polymorphic.relatednameclash: Accessor for field 'polymorphic_ctype' clashes
|
|
# with related field 'ContentType.relatednameclash_set'." (reported by Andrew Ingram)
|
|
# fixed with related_name
|
|
class RelatedNameClash(ShowFieldType, PolymorphicModel):
|
|
ctype = models.ForeignKey(ContentType, null=True, editable=False)
|
|
|
|
# class with a parent_link to superclass, and a related_name back to subclass
|
|
|
|
|
|
class TestParentLinkAndRelatedName(ModelShow1_plain):
|
|
superclass = models.OneToOneField(ModelShow1_plain, parent_link=True, related_name='related_name_subclass')
|
|
|
|
|
|
class CustomPkBase(ShowFieldTypeAndContent, PolymorphicModel):
|
|
b = models.CharField(max_length=1)
|
|
|
|
|
|
class CustomPkInherit(CustomPkBase):
|
|
custom_id = models.AutoField(primary_key=True)
|
|
i = models.CharField(max_length=1)
|
|
|
|
|
|
class DateModel(PolymorphicModel):
|
|
|
|
date = models.DateTimeField()
|
|
|
|
|
|
class PolymorphicTests(TestCase):
|
|
"""
|
|
The test suite
|
|
"""
|
|
def test_annotate_aggregate_order(self):
|
|
# create a blog of type BlogA
|
|
# create two blog entries in BlogA
|
|
# create some blogs of type BlogB to make the BlogBase table data really polymorphic
|
|
blog = BlogA.objects.create(name='B1', info='i1')
|
|
blog.blogentry_set.create(text='bla')
|
|
BlogEntry.objects.create(blog=blog, text='bla2')
|
|
BlogB.objects.create(name='Bb1')
|
|
BlogB.objects.create(name='Bb2')
|
|
BlogB.objects.create(name='Bb3')
|
|
|
|
qs = BlogBase.objects.annotate(entrycount=Count('BlogA___blogentry'))
|
|
self.assertEqual(len(qs), 4)
|
|
|
|
for o in qs:
|
|
if o.name == 'B1':
|
|
self.assertEqual(o.entrycount, 2)
|
|
else:
|
|
self.assertEqual(o.entrycount, 0)
|
|
|
|
x = BlogBase.objects.aggregate(entrycount=Count('BlogA___blogentry'))
|
|
self.assertEqual(x['entrycount'], 2)
|
|
|
|
# create some more blogs for next test
|
|
BlogA.objects.create(name='B2', info='i2')
|
|
BlogA.objects.create(name='B3', info='i3')
|
|
BlogA.objects.create(name='B4', info='i4')
|
|
BlogA.objects.create(name='B5', info='i5')
|
|
|
|
# test ordering for field in all entries
|
|
expected = '''
|
|
[ <BlogB: id 4, name (CharField) "Bb3">,
|
|
<BlogB: id 3, name (CharField) "Bb2">,
|
|
<BlogB: id 2, name (CharField) "Bb1">,
|
|
<BlogA: id 8, name (CharField) "B5", info (CharField) "i5">,
|
|
<BlogA: id 7, name (CharField) "B4", info (CharField) "i4">,
|
|
<BlogA: id 6, name (CharField) "B3", info (CharField) "i3">,
|
|
<BlogA: id 5, name (CharField) "B2", info (CharField) "i2">,
|
|
<BlogA: id 1, name (CharField) "B1", info (CharField) "i1"> ]'''
|
|
x = '\n' + repr(BlogBase.objects.order_by('-name'))
|
|
self.assertEqual(x, expected)
|
|
|
|
# test ordering for field in one subclass only
|
|
# MySQL and SQLite return this order
|
|
expected1 = '''
|
|
[ <BlogA: id 8, name (CharField) "B5", info (CharField) "i5">,
|
|
<BlogA: id 7, name (CharField) "B4", info (CharField) "i4">,
|
|
<BlogA: id 6, name (CharField) "B3", info (CharField) "i3">,
|
|
<BlogA: id 5, name (CharField) "B2", info (CharField) "i2">,
|
|
<BlogA: id 1, name (CharField) "B1", info (CharField) "i1">,
|
|
<BlogB: id 2, name (CharField) "Bb1">,
|
|
<BlogB: id 3, name (CharField) "Bb2">,
|
|
<BlogB: id 4, name (CharField) "Bb3"> ]'''
|
|
|
|
# PostgreSQL returns this order
|
|
expected2 = '''
|
|
[ <BlogB: id 2, name (CharField) "Bb1">,
|
|
<BlogB: id 3, name (CharField) "Bb2">,
|
|
<BlogB: id 4, name (CharField) "Bb3">,
|
|
<BlogA: id 8, name (CharField) "B5", info (CharField) "i5">,
|
|
<BlogA: id 7, name (CharField) "B4", info (CharField) "i4">,
|
|
<BlogA: id 6, name (CharField) "B3", info (CharField) "i3">,
|
|
<BlogA: id 5, name (CharField) "B2", info (CharField) "i2">,
|
|
<BlogA: id 1, name (CharField) "B1", info (CharField) "i1"> ]'''
|
|
|
|
x = '\n' + repr(BlogBase.objects.order_by('-BlogA___info'))
|
|
self.assertTrue(x == expected1 or x == expected2)
|
|
|
|
def test_limit_choices_to(self):
|
|
"""
|
|
this is not really a testcase, as limit_choices_to only affects the Django admin
|
|
"""
|
|
# create a blog of type BlogA
|
|
blog_a = BlogA.objects.create(name='aa', info='aa')
|
|
blog_b = BlogB.objects.create(name='bb')
|
|
# create two blog entries
|
|
entry1 = BlogEntry_limit_choices_to.objects.create(blog=blog_b, text='bla2')
|
|
entry2 = BlogEntry_limit_choices_to.objects.create(blog=blog_b, text='bla2')
|
|
|
|
def test_primary_key_custom_field_problem(self):
|
|
"""
|
|
object retrieval problem occuring with some custom primary key fields (UUIDField as test case)
|
|
"""
|
|
UUIDProject.objects.create(topic="John's gathering")
|
|
UUIDArtProject.objects.create(topic="Sculpting with Tim", artist="T. Turner")
|
|
UUIDResearchProject.objects.create(topic="Swallow Aerodynamics", supervisor="Dr. Winter")
|
|
|
|
qs = UUIDProject.objects.all()
|
|
ol = list(qs)
|
|
a = qs[0]
|
|
b = qs[1]
|
|
c = qs[2]
|
|
self.assertEqual(len(qs), 3)
|
|
self.assertIsInstance(a.uuid_primary_key, uuid.UUID)
|
|
self.assertIsInstance(a.pk, uuid.UUID)
|
|
|
|
res = re.sub(' "(.*?)..", topic', ', topic', repr(qs))
|
|
res_exp = """[ <UUIDProject: uuid_primary_key (UUIDField/pk), topic (CharField) "John's gathering">,
|
|
<UUIDArtProject: uuid_primary_key (UUIDField/pk), topic (CharField) "Sculpting with Tim", artist (CharField) "T. Turner">,
|
|
<UUIDResearchProject: uuid_primary_key (UUIDField/pk), topic (CharField) "Swallow Aerodynamics", supervisor (CharField) "Dr. Winter"> ]"""
|
|
self.assertEqual(res, res_exp)
|
|
|
|
a = UUIDPlainA.objects.create(field1='A1')
|
|
b = UUIDPlainB.objects.create(field1='B1', field2='B2')
|
|
c = UUIDPlainC.objects.create(field1='C1', field2='C2', field3='C3')
|
|
qs = UUIDPlainA.objects.all()
|
|
# Test that primary key values are valid UUIDs
|
|
self.assertEqual(uuid.UUID("urn:uuid:%s" % a.pk, version=1), a.pk)
|
|
self.assertEqual(uuid.UUID("urn:uuid:%s" % c.pk, version=1), c.pk)
|
|
|
|
def create_model2abcd(self):
|
|
"""
|
|
Create the chain of objects of Model2,
|
|
this is reused in various tests.
|
|
"""
|
|
Model2A.objects.create(field1='A1')
|
|
Model2B.objects.create(field1='B1', field2='B2')
|
|
Model2C.objects.create(field1='C1', field2='C2', field3='C3')
|
|
Model2D.objects.create(field1='D1', field2='D2', field3='D3', field4='D4')
|
|
|
|
def test_simple_inheritance(self):
|
|
self.create_model2abcd()
|
|
|
|
objects = list(Model2A.objects.all())
|
|
self.assertEqual(repr(objects[0]), '<Model2A: id 1, field1 (CharField)>')
|
|
self.assertEqual(repr(objects[1]), '<Model2B: id 2, field1 (CharField), field2 (CharField)>')
|
|
self.assertEqual(repr(objects[2]), '<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
self.assertEqual(repr(objects[3]), '<Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)>')
|
|
|
|
def test_defer_fields(self):
|
|
self.create_model2abcd()
|
|
|
|
objects_deferred = Model2A.objects.defer('field1')
|
|
|
|
self.assertNotIn('field1', objects_deferred[0].__dict__, 'field1 was not deferred (using defer())')
|
|
self.assertEqual(repr(objects_deferred[0]),
|
|
'<Model2A: id 1, field1 (CharField), deferred[field1]>')
|
|
self.assertEqual(repr(objects_deferred[1]),
|
|
'<Model2B: id 2, field1 (CharField), field2 (CharField), deferred[field1]>')
|
|
self.assertEqual(repr(objects_deferred[2]),
|
|
'<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField), deferred[field1]>')
|
|
self.assertEqual(repr(objects_deferred[3]),
|
|
'<Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField), deferred[field1]>')
|
|
|
|
objects_only = Model2A.objects.only('pk', 'polymorphic_ctype', 'field1')
|
|
|
|
self.assertIn('field1', objects_only[0].__dict__,
|
|
'qs.only("field1") was used, but field1 was incorrectly deferred')
|
|
self.assertIn('field1', objects_only[3].__dict__,
|
|
'qs.only("field1") was used, but field1 was incorrectly deferred'
|
|
' on a child model')
|
|
self.assertNotIn('field4', objects_only[3].__dict__,
|
|
'field4 was not deferred (using only())')
|
|
self.assertEqual(repr(objects_only[0]),
|
|
'<Model2A: id 1, field1 (CharField)>')
|
|
self.assertEqual(repr(objects_only[1]),
|
|
'<Model2B: id 2, field1 (CharField), field2 (CharField), deferred[field2]>')
|
|
self.assertEqual(repr(objects_only[2]),
|
|
'<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField), '
|
|
'deferred[field2,field3,model2a_ptr_id]>')
|
|
self.assertEqual(repr(objects_only[3]),
|
|
'<Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField), '
|
|
'deferred[field2,field3,field4,model2a_ptr_id,model2b_ptr_id]>')
|
|
|
|
# A bug in Django 1.4 prevents using defer across reverse relations
|
|
# <https://code.djangoproject.com/ticket/14694>. Since polymorphic
|
|
# uses reverse relations to traverse down model inheritance, deferring
|
|
# fields in child models will not work in Django 1.4.
|
|
@skipIf(django.VERSION < (1, 5), "Django 1.4 does not support defer on related fields")
|
|
def test_defer_related_fields(self):
|
|
self.create_model2abcd()
|
|
|
|
objects_deferred_field4 = Model2A.objects.defer('Model2D___field4')
|
|
self.assertNotIn('field4', objects_deferred_field4[3].__dict__,
|
|
'field4 was not deferred (using defer(), traversing inheritance)')
|
|
self.assertEqual(repr(objects_deferred_field4[0]),
|
|
'<Model2A: id 1, field1 (CharField)>')
|
|
self.assertEqual(repr(objects_deferred_field4[1]),
|
|
'<Model2B: id 2, field1 (CharField), field2 (CharField)>')
|
|
self.assertEqual(repr(objects_deferred_field4[2]),
|
|
'<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
self.assertEqual(repr(objects_deferred_field4[3]),
|
|
'<Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField), deferred[field4]>')
|
|
|
|
objects_only_field4 = Model2A.objects.only(
|
|
'polymorphic_ctype', 'field1',
|
|
'Model2B___id', 'Model2B___field2', 'Model2B___model2a_ptr',
|
|
'Model2C___id', 'Model2C___field3', 'Model2C___model2b_ptr',
|
|
'Model2D___id', 'Model2D___model2c_ptr')
|
|
self.assertEqual(repr(objects_only_field4[0]),
|
|
'<Model2A: id 1, field1 (CharField)>')
|
|
self.assertEqual(repr(objects_only_field4[1]),
|
|
'<Model2B: id 2, field1 (CharField), field2 (CharField)>')
|
|
self.assertEqual(repr(objects_only_field4[2]),
|
|
'<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
self.assertEqual(repr(objects_only_field4[3]),
|
|
'<Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField), deferred[field4]>')
|
|
|
|
|
|
def test_manual_get_real_instance(self):
|
|
self.create_model2abcd()
|
|
|
|
o = Model2A.objects.non_polymorphic().get(field1='C1')
|
|
self.assertEqual(repr(o.get_real_instance()), '<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
|
|
def test_non_polymorphic(self):
|
|
self.create_model2abcd()
|
|
|
|
objects = list(Model2A.objects.all().non_polymorphic())
|
|
self.assertEqual(repr(objects[0]), '<Model2A: id 1, field1 (CharField)>')
|
|
self.assertEqual(repr(objects[1]), '<Model2A: id 2, field1 (CharField)>')
|
|
self.assertEqual(repr(objects[2]), '<Model2A: id 3, field1 (CharField)>')
|
|
self.assertEqual(repr(objects[3]), '<Model2A: id 4, field1 (CharField)>')
|
|
|
|
def test_get_real_instances(self):
|
|
self.create_model2abcd()
|
|
qs = Model2A.objects.all().non_polymorphic()
|
|
|
|
# from queryset
|
|
objects = qs.get_real_instances()
|
|
self.assertEqual(repr(objects[0]), '<Model2A: id 1, field1 (CharField)>')
|
|
self.assertEqual(repr(objects[1]), '<Model2B: id 2, field1 (CharField), field2 (CharField)>')
|
|
self.assertEqual(repr(objects[2]), '<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
self.assertEqual(repr(objects[3]), '<Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)>')
|
|
|
|
# from a manual list
|
|
objects = Model2A.objects.get_real_instances(list(qs))
|
|
self.assertEqual(repr(objects[0]), '<Model2A: id 1, field1 (CharField)>')
|
|
self.assertEqual(repr(objects[1]), '<Model2B: id 2, field1 (CharField), field2 (CharField)>')
|
|
self.assertEqual(repr(objects[2]), '<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
self.assertEqual(repr(objects[3]), '<Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)>')
|
|
|
|
def test_translate_polymorphic_q_object(self):
|
|
self.create_model2abcd()
|
|
|
|
q = Model2A.translate_polymorphic_Q_object(Q(instance_of=Model2C))
|
|
objects = Model2A.objects.filter(q)
|
|
self.assertEqual(repr(objects[0]), '<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
self.assertEqual(repr(objects[1]), '<Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)>')
|
|
|
|
def test_base_manager(self):
|
|
def show_base_manager(model):
|
|
return "{0} {1}".format(
|
|
repr(type(model._base_manager)),
|
|
repr(model._base_manager.model)
|
|
)
|
|
|
|
self.assertEqual(show_base_manager(PlainA), "<class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.PlainA'>")
|
|
self.assertEqual(show_base_manager(PlainB), "<class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.PlainB'>")
|
|
self.assertEqual(show_base_manager(PlainC), "<class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.PlainC'>")
|
|
|
|
self.assertEqual(show_base_manager(Model2A), "<class 'polymorphic.managers.PolymorphicManager'> <class 'polymorphic.tests.Model2A'>")
|
|
self.assertEqual(show_base_manager(Model2B), "<class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.Model2B'>")
|
|
self.assertEqual(show_base_manager(Model2C), "<class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.Model2C'>")
|
|
|
|
self.assertEqual(show_base_manager(One2OneRelatingModel), "<class 'polymorphic.managers.PolymorphicManager'> <class 'polymorphic.tests.One2OneRelatingModel'>")
|
|
self.assertEqual(show_base_manager(One2OneRelatingModelDerived), "<class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.One2OneRelatingModelDerived'>")
|
|
|
|
def test_instance_default_manager(self):
|
|
def show_default_manager(instance):
|
|
return "{0} {1}".format(
|
|
repr(type(instance.__class__._default_manager)),
|
|
repr(instance.__class__._default_manager.model)
|
|
)
|
|
|
|
plain_a = PlainA(field1='C1')
|
|
plain_b = PlainB(field2='C1')
|
|
plain_c = PlainC(field3='C1')
|
|
|
|
model_2a = Model2A(field1='C1')
|
|
model_2b = Model2B(field2='C1')
|
|
model_2c = Model2C(field3='C1')
|
|
|
|
self.assertEqual(show_default_manager(plain_a), "<class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.PlainA'>")
|
|
self.assertEqual(show_default_manager(plain_b), "<class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.PlainB'>")
|
|
self.assertEqual(show_default_manager(plain_c), "<class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.PlainC'>")
|
|
|
|
self.assertEqual(show_default_manager(model_2a), "<class 'polymorphic.managers.PolymorphicManager'> <class 'polymorphic.tests.Model2A'>")
|
|
self.assertEqual(show_default_manager(model_2b), "<class 'polymorphic.managers.PolymorphicManager'> <class 'polymorphic.tests.Model2B'>")
|
|
self.assertEqual(show_default_manager(model_2c), "<class 'polymorphic.managers.PolymorphicManager'> <class 'polymorphic.tests.Model2C'>")
|
|
|
|
def test_foreignkey_field(self):
|
|
self.create_model2abcd()
|
|
|
|
object2a = Model2A.base_objects.get(field1='C1')
|
|
self.assertEqual(repr(object2a.model2b), '<Model2B: id 3, field1 (CharField), field2 (CharField)>')
|
|
|
|
object2b = Model2B.base_objects.get(field1='C1')
|
|
self.assertEqual(repr(object2b.model2c), '<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
|
|
def test_onetoone_field(self):
|
|
self.create_model2abcd()
|
|
|
|
a = Model2A.base_objects.get(field1='C1')
|
|
b = One2OneRelatingModelDerived.objects.create(one2one=a, field1='f1', field2='f2')
|
|
|
|
# this result is basically wrong, probably due to Django cacheing (we used base_objects), but should not be a problem
|
|
self.assertEqual(repr(b.one2one), '<Model2A: id 3, field1 (CharField)>')
|
|
|
|
c = One2OneRelatingModelDerived.objects.get(field1='f1')
|
|
self.assertEqual(repr(c.one2one), '<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
self.assertEqual(repr(a.one2onerelatingmodel), '<One2OneRelatingModelDerived: One2OneRelatingModelDerived object>')
|
|
|
|
def test_manytomany_field(self):
|
|
# Model 1
|
|
o = ModelShow1.objects.create(field1='abc')
|
|
o.m2m.add(o)
|
|
o.save()
|
|
self.assertEqual(repr(ModelShow1.objects.all()), '[ <ModelShow1: id 1, field1 (CharField), m2m (ManyToManyField)> ]')
|
|
|
|
# Model 2
|
|
o = ModelShow2.objects.create(field1='abc')
|
|
o.m2m.add(o)
|
|
o.save()
|
|
self.assertEqual(repr(ModelShow2.objects.all()), '[ <ModelShow2: id 1, field1 "abc", m2m 1> ]')
|
|
|
|
# Model 3
|
|
o = ModelShow3.objects.create(field1='abc')
|
|
o.m2m.add(o)
|
|
o.save()
|
|
self.assertEqual(repr(ModelShow3.objects.all()), '[ <ModelShow3: id 1, field1 (CharField) "abc", m2m (ManyToManyField) 1> ]')
|
|
self.assertEqual(repr(ModelShow1.objects.all().annotate(Count('m2m'))), '[ <ModelShow1: id 1, field1 (CharField), m2m (ManyToManyField) - Ann: m2m__count (int)> ]')
|
|
self.assertEqual(repr(ModelShow2.objects.all().annotate(Count('m2m'))), '[ <ModelShow2: id 1, field1 "abc", m2m 1 - Ann: m2m__count 1> ]')
|
|
self.assertEqual(repr(ModelShow3.objects.all().annotate(Count('m2m'))), '[ <ModelShow3: id 1, field1 (CharField) "abc", m2m (ManyToManyField) 1 - Ann: m2m__count (int) 1> ]')
|
|
|
|
# no pretty printing
|
|
ModelShow1_plain.objects.create(field1='abc')
|
|
ModelShow2_plain.objects.create(field1='abc', field2='def')
|
|
self.assertEqual(qrepr(ModelShow1_plain.objects.all()), '<QuerySet [<ModelShow1_plain: ModelShow1_plain object>, <ModelShow2_plain: ModelShow2_plain object>]>')
|
|
|
|
def test_extra_method(self):
|
|
self.create_model2abcd()
|
|
|
|
objects = list(Model2A.objects.extra(where=['id IN (2, 3)']))
|
|
self.assertEqual(repr(objects[0]), '<Model2B: id 2, field1 (CharField), field2 (CharField)>')
|
|
self.assertEqual(repr(objects[1]), '<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
|
|
objects = Model2A.objects.extra(select={"select_test": "field1 = 'A1'"}, where=["field1 = 'A1' OR field1 = 'B1'"], order_by=['-id'])
|
|
self.assertEqual(repr(objects[0]), '<Model2B: id 2, field1 (CharField), field2 (CharField) - Extra: select_test (int)>')
|
|
self.assertEqual(repr(objects[1]), '<Model2A: id 1, field1 (CharField) - Extra: select_test (int)>')
|
|
self.assertEqual(len(objects), 2) # Placed after the other tests, only verifying whether there are no more additional objects.
|
|
|
|
ModelExtraA.objects.create(field1='A1')
|
|
ModelExtraB.objects.create(field1='B1', field2='B2')
|
|
ModelExtraC.objects.create(field1='C1', field2='C2', field3='C3')
|
|
ModelExtraExternal.objects.create(topic='extra1')
|
|
ModelExtraExternal.objects.create(topic='extra2')
|
|
ModelExtraExternal.objects.create(topic='extra3')
|
|
objects = ModelExtraA.objects.extra(tables=["polymorphic_modelextraexternal"], select={"topic": "polymorphic_modelextraexternal.topic"}, where=["polymorphic_modelextraa.id = polymorphic_modelextraexternal.id"])
|
|
if six.PY3:
|
|
self.assertEqual(repr(objects[0]), '<ModelExtraA: id 1, field1 (CharField) "A1" - Extra: topic (str) "extra1">')
|
|
self.assertEqual(repr(objects[1]), '<ModelExtraB: id 2, field1 (CharField) "B1", field2 (CharField) "B2" - Extra: topic (str) "extra2">')
|
|
self.assertEqual(repr(objects[2]), '<ModelExtraC: id 3, field1 (CharField) "C1", field2 (CharField) "C2", field3 (CharField) "C3" - Extra: topic (str) "extra3">')
|
|
else:
|
|
self.assertEqual(repr(objects[0]), '<ModelExtraA: id 1, field1 (CharField) "A1" - Extra: topic (unicode) "extra1">')
|
|
self.assertEqual(repr(objects[1]), '<ModelExtraB: id 2, field1 (CharField) "B1", field2 (CharField) "B2" - Extra: topic (unicode) "extra2">')
|
|
self.assertEqual(repr(objects[2]), '<ModelExtraC: id 3, field1 (CharField) "C1", field2 (CharField) "C2", field3 (CharField) "C3" - Extra: topic (unicode) "extra3">')
|
|
self.assertEqual(len(objects), 3)
|
|
|
|
def test_instance_of_filter(self):
|
|
self.create_model2abcd()
|
|
|
|
objects = Model2A.objects.instance_of(Model2B)
|
|
self.assertEqual(repr(objects[0]), '<Model2B: id 2, field1 (CharField), field2 (CharField)>')
|
|
self.assertEqual(repr(objects[1]), '<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
self.assertEqual(repr(objects[2]), '<Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)>')
|
|
self.assertEqual(len(objects), 3)
|
|
|
|
objects = Model2A.objects.filter(instance_of=Model2B)
|
|
self.assertEqual(repr(objects[0]), '<Model2B: id 2, field1 (CharField), field2 (CharField)>')
|
|
self.assertEqual(repr(objects[1]), '<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
self.assertEqual(repr(objects[2]), '<Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)>')
|
|
self.assertEqual(len(objects), 3)
|
|
|
|
objects = Model2A.objects.filter(Q(instance_of=Model2B))
|
|
self.assertEqual(repr(objects[0]), '<Model2B: id 2, field1 (CharField), field2 (CharField)>')
|
|
self.assertEqual(repr(objects[1]), '<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
self.assertEqual(repr(objects[2]), '<Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)>')
|
|
self.assertEqual(len(objects), 3)
|
|
|
|
objects = Model2A.objects.not_instance_of(Model2B)
|
|
self.assertEqual(repr(objects[0]), '<Model2A: id 1, field1 (CharField)>')
|
|
self.assertEqual(len(objects), 1)
|
|
|
|
def test_polymorphic___filter(self):
|
|
self.create_model2abcd()
|
|
|
|
objects = Model2A.objects.filter(Q(Model2B___field2='B2') | Q(Model2C___field3='C3'))
|
|
self.assertEqual(len(objects), 2)
|
|
self.assertEqual(repr(objects[0]), '<Model2B: id 2, field1 (CharField), field2 (CharField)>')
|
|
self.assertEqual(repr(objects[1]), '<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
|
|
@skipIf(django.VERSION < (1, 6), "Django 1.4 and 1.5 don't support q.clone()")
|
|
def test_query_filter_exclude_is_immutable(self):
|
|
# given
|
|
q_to_reuse = Q(Model2B___field2='something')
|
|
untouched_q_object = Q(Model2B___field2='something')
|
|
# when
|
|
Model2A.objects.filter(q_to_reuse).all()
|
|
# then
|
|
self.assertEquals(q_to_reuse.children, untouched_q_object.children)
|
|
|
|
# given
|
|
q_to_reuse = Q(Model2B___field2='something')
|
|
untouched_q_object = Q(Model2B___field2='something')
|
|
# when
|
|
Model2B.objects.filter(q_to_reuse).all()
|
|
# then
|
|
self.assertEquals(q_to_reuse.children, untouched_q_object.children)
|
|
|
|
def test_polymorphic___filter_field(self):
|
|
p = ModelUnderRelParent.objects.create(_private=True, field1='AA')
|
|
ModelUnderRelChild.objects.create(parent=p, _private2=True)
|
|
|
|
# The "___" filter should also parse to "parent" -> "_private" as fallback.
|
|
objects = ModelUnderRelChild.objects.filter(parent___private=True)
|
|
self.assertEqual(len(objects), 1)
|
|
|
|
def test_polymorphic___filter_reverse_field(self):
|
|
p = ModelUnderRelParent.objects.create(_private=True, field1='BB')
|
|
ModelUnderRelChild.objects.create(parent=p, _private2=True)
|
|
|
|
# Also test for reverse relations
|
|
objects = ModelUnderRelParent.objects.filter(children___private2=True)
|
|
self.assertEqual(len(objects), 1)
|
|
|
|
def test_delete(self):
|
|
self.create_model2abcd()
|
|
|
|
oa = Model2A.objects.get(id=2)
|
|
self.assertEqual(repr(oa), '<Model2B: id 2, field1 (CharField), field2 (CharField)>')
|
|
self.assertEqual(Model2A.objects.count(), 4)
|
|
|
|
oa.delete()
|
|
objects = Model2A.objects.all()
|
|
self.assertEqual(repr(objects[0]), '<Model2A: id 1, field1 (CharField)>')
|
|
self.assertEqual(repr(objects[1]), '<Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
self.assertEqual(repr(objects[2]), '<Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)>')
|
|
self.assertEqual(len(objects), 3)
|
|
|
|
def test_combine_querysets(self):
|
|
ModelX.objects.create(field_x='x')
|
|
ModelY.objects.create(field_y='y')
|
|
|
|
qs = Base.objects.instance_of(ModelX) | Base.objects.instance_of(ModelY)
|
|
self.assertEqual(repr(qs[0]), '<ModelX: id 1, field_b (CharField), field_x (CharField)>')
|
|
self.assertEqual(repr(qs[1]), '<ModelY: id 2, field_b (CharField), field_y (CharField)>')
|
|
self.assertEqual(len(qs), 2)
|
|
|
|
def test_multiple_inheritance(self):
|
|
# multiple inheritance, subclassing third party models (mix PolymorphicModel with models.Model)
|
|
|
|
Enhance_Base.objects.create(field_b='b-base')
|
|
Enhance_Inherit.objects.create(field_b='b-inherit', field_p='p', field_i='i')
|
|
|
|
qs = Enhance_Base.objects.all()
|
|
self.assertEqual(len(qs), 2)
|
|
self.assertEqual(repr(qs[0]), '<Enhance_Base: base_id (AutoField/pk) 1, field_b (CharField) "b-base">')
|
|
self.assertEqual(repr(qs[1]), '<Enhance_Inherit: base_id (AutoField/pk) 2, field_b (CharField) "b-inherit", id 1, field_p (CharField) "p", field_i (CharField) "i">')
|
|
|
|
def test_relation_base(self):
|
|
# ForeignKey, ManyToManyField
|
|
obase = RelationBase.objects.create(field_base='base')
|
|
oa = RelationA.objects.create(field_base='A1', field_a='A2', fk=obase)
|
|
ob = RelationB.objects.create(field_base='B1', field_b='B2', fk=oa)
|
|
oc = RelationBC.objects.create(field_base='C1', field_b='C2', field_c='C3', fk=oa)
|
|
oa.m2m.add(oa)
|
|
oa.m2m.add(ob)
|
|
|
|
objects = RelationBase.objects.all()
|
|
self.assertEqual(repr(objects[0]), '<RelationBase: id 1, field_base (CharField) "base", fk (ForeignKey) None, m2m (ManyToManyField) 0>')
|
|
self.assertEqual(repr(objects[1]), '<RelationA: id 2, field_base (CharField) "A1", fk (ForeignKey) RelationBase, field_a (CharField) "A2", m2m (ManyToManyField) 2>')
|
|
self.assertEqual(repr(objects[2]), '<RelationB: id 3, field_base (CharField) "B1", fk (ForeignKey) RelationA, field_b (CharField) "B2", m2m (ManyToManyField) 1>')
|
|
self.assertEqual(repr(objects[3]), '<RelationBC: id 4, field_base (CharField) "C1", fk (ForeignKey) RelationA, field_b (CharField) "C2", field_c (CharField) "C3", m2m (ManyToManyField) 0>')
|
|
self.assertEqual(len(objects), 4)
|
|
|
|
oa = RelationBase.objects.get(id=2)
|
|
self.assertEqual(repr(oa.fk), '<RelationBase: id 1, field_base (CharField) "base", fk (ForeignKey) None, m2m (ManyToManyField) 0>')
|
|
|
|
objects = oa.relationbase_set.all()
|
|
self.assertEqual(repr(objects[0]), '<RelationB: id 3, field_base (CharField) "B1", fk (ForeignKey) RelationA, field_b (CharField) "B2", m2m (ManyToManyField) 1>')
|
|
self.assertEqual(repr(objects[1]), '<RelationBC: id 4, field_base (CharField) "C1", fk (ForeignKey) RelationA, field_b (CharField) "C2", field_c (CharField) "C3", m2m (ManyToManyField) 0>')
|
|
self.assertEqual(len(objects), 2)
|
|
|
|
ob = RelationBase.objects.get(id=3)
|
|
self.assertEqual(repr(ob.fk), '<RelationA: id 2, field_base (CharField) "A1", fk (ForeignKey) RelationBase, field_a (CharField) "A2", m2m (ManyToManyField) 2>')
|
|
|
|
oa = RelationA.objects.get()
|
|
objects = oa.m2m.all()
|
|
self.assertEqual(repr(objects[0]), '<RelationA: id 2, field_base (CharField) "A1", fk (ForeignKey) RelationBase, field_a (CharField) "A2", m2m (ManyToManyField) 2>')
|
|
self.assertEqual(repr(objects[1]), '<RelationB: id 3, field_base (CharField) "B1", fk (ForeignKey) RelationA, field_b (CharField) "B2", m2m (ManyToManyField) 1>')
|
|
self.assertEqual(len(objects), 2)
|
|
|
|
def test_user_defined_manager(self):
|
|
self.create_model2abcd()
|
|
ModelWithMyManager.objects.create(field1='D1a', field4='D4a')
|
|
ModelWithMyManager.objects.create(field1='D1b', field4='D4b')
|
|
|
|
objects = ModelWithMyManager.objects.all() # MyManager should reverse the sorting of field1
|
|
self.assertEqual(repr(objects[0]), '<ModelWithMyManager: id 6, field1 (CharField) "D1b", field4 (CharField) "D4b">')
|
|
self.assertEqual(repr(objects[1]), '<ModelWithMyManager: id 5, field1 (CharField) "D1a", field4 (CharField) "D4a">')
|
|
self.assertEqual(len(objects), 2)
|
|
|
|
self.assertIs(type(ModelWithMyManager.objects), MyManager)
|
|
self.assertIs(type(ModelWithMyManager._default_manager), MyManager)
|
|
self.assertIs(type(ModelWithMyManager.base_objects), models.Manager)
|
|
|
|
def test_user_defined_manager_as_secondary(self):
|
|
self.create_model2abcd()
|
|
ModelWithMyManagerNoDefault.objects.create(field1='D1a', field4='D4a')
|
|
ModelWithMyManagerNoDefault.objects.create(field1='D1b', field4='D4b')
|
|
|
|
objects = ModelWithMyManagerNoDefault.my_objects.all() # MyManager should reverse the sorting of field1
|
|
self.assertEqual(repr(objects[0]), '<ModelWithMyManagerNoDefault: id 6, field1 (CharField) "D1b", field4 (CharField) "D4b">')
|
|
self.assertEqual(repr(objects[1]), '<ModelWithMyManagerNoDefault: id 5, field1 (CharField) "D1a", field4 (CharField) "D4a">')
|
|
self.assertEqual(len(objects), 2)
|
|
|
|
self.assertIs(type(ModelWithMyManagerNoDefault.my_objects), MyManager)
|
|
self.assertIs(type(ModelWithMyManagerNoDefault.objects), PolymorphicManager)
|
|
self.assertIs(type(ModelWithMyManagerNoDefault._default_manager), PolymorphicManager)
|
|
self.assertIs(type(ModelWithMyManagerNoDefault.base_objects), models.Manager)
|
|
|
|
def test_user_objects_manager_as_secondary(self):
|
|
self.create_model2abcd()
|
|
ModelWithMyManagerDefault.objects.create(field1='D1a', field4='D4a')
|
|
ModelWithMyManagerDefault.objects.create(field1='D1b', field4='D4b')
|
|
|
|
self.assertIs(type(ModelWithMyManagerDefault.my_objects), MyManager)
|
|
self.assertIs(type(ModelWithMyManagerDefault.objects), PolymorphicManager)
|
|
self.assertIs(type(ModelWithMyManagerDefault._default_manager), MyManager)
|
|
self.assertIs(type(ModelWithMyManagerDefault.base_objects), models.Manager)
|
|
|
|
@skipIf(django.VERSION < (1, 7), "This test needs Django 1.7+")
|
|
def test_user_defined_queryset_as_manager(self):
|
|
self.create_model2abcd()
|
|
ModelWithMyManager2.objects.create(field1='D1a', field4='D4a')
|
|
ModelWithMyManager2.objects.create(field1='D1b', field4='D4b')
|
|
|
|
objects = ModelWithMyManager2.objects.all()
|
|
self.assertEqual(repr(objects[0]), '<ModelWithMyManager2: id 5, field1 (CharField) "D1a", field4 (CharField) "D4a">')
|
|
self.assertEqual(repr(objects[1]), '<ModelWithMyManager2: id 6, field1 (CharField) "D1b", field4 (CharField) "D4b">')
|
|
self.assertEqual(len(objects), 2)
|
|
|
|
self.assertEqual(type(ModelWithMyManager2.objects).__name__, 'PolymorphicManagerFromMyManagerQuerySet')
|
|
self.assertEqual(type(ModelWithMyManager2._default_manager).__name__, 'PolymorphicManagerFromMyManagerQuerySet')
|
|
self.assertIs(type(ModelWithMyManager2.base_objects), models.Manager)
|
|
|
|
def test_manager_inheritance(self):
|
|
# by choice of MRO, should be MyManager from MROBase1.
|
|
self.assertIs(type(MRODerived.objects), MyManager)
|
|
|
|
# check for correct default manager
|
|
self.assertIs(type(MROBase1._default_manager), MyManager)
|
|
|
|
# Django vanilla inheritance does not inherit MyManager as _default_manager here
|
|
self.assertIs(type(MROBase2._default_manager), MyManager)
|
|
|
|
def test_queryset_assignment(self):
|
|
# This is just a consistency check for now, testing standard Django behavior.
|
|
parent = PlainParentModelWithManager.objects.create()
|
|
child = PlainChildModelWithManager.objects.create(fk=parent)
|
|
self.assertIs(type(PlainParentModelWithManager._default_manager), models.Manager)
|
|
self.assertIs(type(PlainChildModelWithManager._default_manager), PlainMyManager)
|
|
self.assertIs(type(PlainChildModelWithManager.objects), PlainMyManager)
|
|
self.assertIs(type(PlainChildModelWithManager.objects.all()), PlainMyManagerQuerySet)
|
|
|
|
# A related set is created using the model's _default_manager, so does gain extra methods.
|
|
self.assertIs(type(parent.childmodel_set.my_queryset_foo()), PlainMyManagerQuerySet)
|
|
|
|
# For polymorphic models, the same should happen.
|
|
parent = ParentModelWithManager.objects.create()
|
|
child = ChildModelWithManager.objects.create(fk=parent)
|
|
self.assertIs(type(ParentModelWithManager._default_manager), PolymorphicManager)
|
|
self.assertIs(type(ChildModelWithManager._default_manager), MyManager)
|
|
self.assertIs(type(ChildModelWithManager.objects), MyManager)
|
|
self.assertIs(type(ChildModelWithManager.objects.my_queryset_foo()), MyManagerQuerySet)
|
|
|
|
# A related set is created using the model's _default_manager, so does gain extra methods.
|
|
self.assertIs(type(parent.childmodel_set.my_queryset_foo()), MyManagerQuerySet)
|
|
|
|
def test_proxy_models(self):
|
|
# prepare some data
|
|
for data in ('bleep bloop', 'I am a', 'computer'):
|
|
ProxyChild.objects.create(some_data=data)
|
|
|
|
# this caches ContentType queries so they don't interfere with our query counts later
|
|
list(ProxyBase.objects.all())
|
|
|
|
# one query per concrete class
|
|
with self.assertNumQueries(1):
|
|
items = list(ProxyBase.objects.all())
|
|
|
|
self.assertIsInstance(items[0], ProxyChild)
|
|
|
|
def test_queryset_on_proxy_model_does_not_return_superclasses(self):
|
|
ProxyBase.objects.create(some_data='Base1')
|
|
ProxyBase.objects.create(some_data='Base2')
|
|
ProxyChild.objects.create(some_data='Child1')
|
|
ProxyChild.objects.create(some_data='Child2')
|
|
ProxyChild.objects.create(some_data='Child3')
|
|
|
|
self.assertEquals(5, ProxyBase.objects.count())
|
|
self.assertEquals(3, ProxyChild.objects.count())
|
|
|
|
def test_proxy_get_real_instance_class(self):
|
|
"""
|
|
The call to ``get_real_instance()`` also checks whether the returned model is of the correct type.
|
|
This unit test guards that this check is working properly. For instance,
|
|
proxy child models need to be handled separately.
|
|
"""
|
|
name = "Item1"
|
|
nonproxychild = NonProxyChild.objects.create(name=name)
|
|
|
|
pb = ProxyBase.objects.get(id=1)
|
|
self.assertEqual(pb.get_real_instance_class(), NonProxyChild)
|
|
self.assertEqual(pb.get_real_instance(), nonproxychild)
|
|
self.assertEqual(pb.name, name)
|
|
|
|
pbm = NonProxyChild.objects.get(id=1)
|
|
self.assertEqual(pbm.get_real_instance_class(), NonProxyChild)
|
|
self.assertEqual(pbm.get_real_instance(), nonproxychild)
|
|
self.assertEqual(pbm.name, name)
|
|
|
|
def test_content_types_for_proxy_models(self):
|
|
"""Checks if ContentType is capable of returning proxy models."""
|
|
from django.db.models import Model
|
|
from django.contrib.contenttypes.models import ContentType
|
|
|
|
ct = ContentType.objects.get_for_model(ProxyChild, for_concrete_model=False)
|
|
self.assertEqual(ProxyChild, ct.model_class())
|
|
|
|
def test_proxy_model_inheritance(self):
|
|
"""
|
|
Polymorphic abilities should also work when the base model is a proxy object.
|
|
"""
|
|
# The managers should point to the proper objects.
|
|
# otherwise, the whole excersise is pointless.
|
|
self.assertEqual(ProxiedBase.objects.model, ProxiedBase)
|
|
self.assertEqual(ProxyModelBase.objects.model, ProxyModelBase)
|
|
self.assertEqual(ProxyModelA.objects.model, ProxyModelA)
|
|
self.assertEqual(ProxyModelB.objects.model, ProxyModelB)
|
|
|
|
# Create objects
|
|
ProxyModelA.objects.create(name="object1")
|
|
ProxyModelB.objects.create(name="object2", field2="bb")
|
|
|
|
# Getting single objects
|
|
object1 = ProxyModelBase.objects.get(name='object1')
|
|
object2 = ProxyModelBase.objects.get(name='object2')
|
|
self.assertEqual(repr(object1), '<ProxyModelA: id 1, name (CharField) "object1", field1 (CharField) "">')
|
|
self.assertEqual(repr(object2), '<ProxyModelB: id 2, name (CharField) "object2", field2 (CharField) "bb">')
|
|
self.assertIsInstance(object1, ProxyModelA)
|
|
self.assertIsInstance(object2, ProxyModelB)
|
|
|
|
# Same for lists
|
|
objects = list(ProxyModelBase.objects.all().order_by('name'))
|
|
self.assertEqual(repr(objects[0]), '<ProxyModelA: id 1, name (CharField) "object1", field1 (CharField) "">')
|
|
self.assertEqual(repr(objects[1]), '<ProxyModelB: id 2, name (CharField) "object2", field2 (CharField) "bb">')
|
|
self.assertIsInstance(objects[0], ProxyModelA)
|
|
self.assertIsInstance(objects[1], ProxyModelB)
|
|
|
|
def test_custom_pk(self):
|
|
CustomPkBase.objects.create(b='b')
|
|
CustomPkInherit.objects.create(b='b', i='i')
|
|
qs = CustomPkBase.objects.all()
|
|
self.assertEqual(len(qs), 2)
|
|
self.assertEqual(repr(qs[0]), '<CustomPkBase: id 1, b (CharField) "b">')
|
|
self.assertEqual(repr(qs[1]), '<CustomPkInherit: id 2, b (CharField) "b", custom_id (AutoField/pk) 1, i (CharField) "i">')
|
|
|
|
def test_fix_getattribute(self):
|
|
# fixed issue in PolymorphicModel.__getattribute__: field name same as model name
|
|
o = ModelFieldNameTest.objects.create(modelfieldnametest='1')
|
|
self.assertEqual(repr(o), '<ModelFieldNameTest: id 1, modelfieldnametest (CharField)>')
|
|
|
|
# if subclass defined __init__ and accessed class members,
|
|
# __getattribute__ had a problem: "...has no attribute 'sub_and_superclass_dict'"
|
|
o = InitTestModelSubclass.objects.create()
|
|
self.assertEqual(o.bar, 'XYZ')
|
|
|
|
def test_parent_link_and_related_name(self):
|
|
t = TestParentLinkAndRelatedName(field1="TestParentLinkAndRelatedName")
|
|
t.save()
|
|
p = ModelShow1_plain.objects.get(field1="TestParentLinkAndRelatedName")
|
|
|
|
# check that p is equal to the
|
|
self.assertIsInstance(p, TestParentLinkAndRelatedName)
|
|
self.assertEqual(p, t)
|
|
|
|
# check that the accessors to parent and sublass work correctly and return the right object
|
|
p = ModelShow1_plain.objects.non_polymorphic().get(field1="TestParentLinkAndRelatedName")
|
|
self.assertNotEqual(p, t) # p should be Plain1 and t TestParentLinkAndRelatedName, so not equal
|
|
self.assertEqual(p, t.superclass)
|
|
self.assertEqual(p.related_name_subclass, t)
|
|
|
|
# test that we can delete the object
|
|
t.delete()
|
|
|
|
def test_polymorphic__aggregate(self):
|
|
""" test ModelX___field syntax on aggregate (should work for annotate either) """
|
|
|
|
Model2A.objects.create(field1='A1')
|
|
Model2B.objects.create(field1='A1', field2='B2')
|
|
Model2B.objects.create(field1='A1', field2='B2')
|
|
|
|
# aggregate using **kwargs
|
|
result = Model2A.objects.aggregate(cnt=Count('Model2B___field2'))
|
|
self.assertEqual(result, {'cnt': 2})
|
|
|
|
# aggregate using **args
|
|
with self.assertRaisesMessage(AssertionError, 'PolymorphicModel: annotate()/aggregate(): ___ model lookup supported for keyword arguments only'):
|
|
Model2A.objects.aggregate(Count('Model2B___field2'))
|
|
|
|
|
|
|
|
@skipIf(django.VERSION < (1,8,), "This test needs Django >=1.8")
|
|
def test_polymorphic__complex_aggregate(self):
|
|
""" test (complex expression on) aggregate (should work for annotate either) """
|
|
|
|
Model2A.objects.create(field1='A1')
|
|
Model2B.objects.create(field1='A1', field2='B2')
|
|
Model2B.objects.create(field1='A1', field2='B2')
|
|
|
|
# aggregate using **kwargs
|
|
result = Model2A.objects.aggregate(
|
|
cnt_a1=Count(Case(When(field1='A1', then=1))),
|
|
cnt_b2=Count(Case(When(Model2B___field2='B2', then=1))),
|
|
)
|
|
self.assertEqual(result, {'cnt_b2': 2, 'cnt_a1': 3})
|
|
|
|
# aggregate using **args
|
|
# we have to set the defaul alias or django won't except a complex expression
|
|
# on aggregate/annotate
|
|
def ComplexAgg(expression):
|
|
complexagg = Count(expression)*10
|
|
complexagg.default_alias = 'complexagg'
|
|
return complexagg
|
|
|
|
with self.assertRaisesMessage(AssertionError, 'PolymorphicModel: annotate()/aggregate(): ___ model lookup supported for keyword arguments only'):
|
|
Model2A.objects.aggregate(ComplexAgg('Model2B___field2'))
|
|
|
|
@skipIf(django.VERSION < (1,8,), "This test needs Django >=1.8")
|
|
def test_polymorphic__expressions(self):
|
|
|
|
from django.db.models.functions import Concat
|
|
|
|
# no exception raised
|
|
result = Model2B.objects.annotate(val=Concat('field1', 'field2'))
|
|
self.assertEqual(list(result), [])
|
|
|
|
class RegressionTests(TestCase):
|
|
|
|
def test_for_query_result_incomplete_with_inheritance(self):
|
|
""" https://github.com/bconstantin/django_polymorphic/issues/15 """
|
|
|
|
top = Top()
|
|
top.save()
|
|
middle = Middle()
|
|
middle.save()
|
|
bottom = Bottom()
|
|
bottom.save()
|
|
|
|
expected_queryset = [top, middle, bottom]
|
|
self.assertQuerysetEqual(Top.objects.all(), [repr(r) for r in expected_queryset])
|
|
|
|
expected_queryset = [middle, bottom]
|
|
self.assertQuerysetEqual(Middle.objects.all(), [repr(r) for r in expected_queryset])
|
|
|
|
expected_queryset = [bottom]
|
|
self.assertQuerysetEqual(Bottom.objects.all(), [repr(r) for r in expected_queryset])
|
|
|
|
class MultipleDatabasesTests(TestCase):
|
|
multi_db = True
|
|
|
|
@skipIf(django.VERSION < (1,5,), "This test needs Django >=1.5")
|
|
def test_save_to_non_default_database(self):
|
|
Model2A.objects.db_manager('secondary').create(field1='A1')
|
|
Model2C(field1='C1', field2='C2', field3='C3').save(using='secondary')
|
|
Model2B.objects.create(field1='B1', field2='B2')
|
|
Model2D(field1='D1', field2='D2', field3='D3', field4='D4').save()
|
|
|
|
default_objects = list(Model2A.objects.order_by('id'))
|
|
self.assertEqual(len(default_objects), 2)
|
|
self.assertEqual(repr(default_objects[0]), '<Model2B: id 1, field1 (CharField), field2 (CharField)>')
|
|
self.assertEqual(repr(default_objects[1]), '<Model2D: id 2, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)>')
|
|
|
|
secondary_objects = list(Model2A.objects.db_manager('secondary').order_by('id'))
|
|
self.assertEqual(len(secondary_objects), 2)
|
|
self.assertEqual(repr(secondary_objects[0]), '<Model2A: id 1, field1 (CharField)>')
|
|
self.assertEqual(repr(secondary_objects[1]), '<Model2C: id 2, field1 (CharField), field2 (CharField), field3 (CharField)>')
|
|
|
|
def test_instance_of_filter_on_non_default_database(self):
|
|
Base.objects.db_manager('secondary').create(field_b='B1')
|
|
ModelX.objects.db_manager('secondary').create(field_b='B', field_x='X')
|
|
ModelY.objects.db_manager('secondary').create(field_b='Y', field_y='Y')
|
|
|
|
objects = Base.objects.db_manager('secondary').filter(instance_of=Base)
|
|
self.assertEqual(len(objects), 3)
|
|
self.assertEqual(repr(objects[0]), '<Base: id 1, field_b (CharField)>')
|
|
self.assertEqual(repr(objects[1]), '<ModelX: id 2, field_b (CharField), field_x (CharField)>')
|
|
self.assertEqual(repr(objects[2]), '<ModelY: id 3, field_b (CharField), field_y (CharField)>')
|
|
|
|
objects = Base.objects.db_manager('secondary').filter(instance_of=ModelX)
|
|
self.assertEqual(len(objects), 1)
|
|
self.assertEqual(repr(objects[0]), '<ModelX: id 2, field_b (CharField), field_x (CharField)>')
|
|
|
|
objects = Base.objects.db_manager('secondary').filter(instance_of=ModelY)
|
|
self.assertEqual(len(objects), 1)
|
|
self.assertEqual(repr(objects[0]), '<ModelY: id 3, field_b (CharField), field_y (CharField)>')
|
|
|
|
objects = Base.objects.db_manager('secondary').filter(Q(instance_of=ModelX) | Q(instance_of=ModelY))
|
|
self.assertEqual(len(objects), 2)
|
|
self.assertEqual(repr(objects[0]), '<ModelX: id 2, field_b (CharField), field_x (CharField)>')
|
|
self.assertEqual(repr(objects[1]), '<ModelY: id 3, field_b (CharField), field_y (CharField)>')
|
|
|
|
def test_forward_many_to_one_descriptor_on_non_default_database(self):
|
|
def func():
|
|
blog = BlogA.objects.db_manager('secondary').create(name='Blog', info='Info')
|
|
entry = BlogEntry.objects.db_manager('secondary').create(blog=blog, text='Text')
|
|
ContentType.objects.clear_cache()
|
|
entry = BlogEntry.objects.db_manager('secondary').get(pk=entry.id)
|
|
self.assertEqual(blog, entry.blog)
|
|
|
|
# Ensure no queries are made using the default database.
|
|
self.assertNumQueries(0, func)
|
|
|
|
def test_reverse_many_to_one_descriptor_on_non_default_database(self):
|
|
def func():
|
|
blog = BlogA.objects.db_manager('secondary').create(name='Blog', info='Info')
|
|
entry = BlogEntry.objects.db_manager('secondary').create(blog=blog, text='Text')
|
|
ContentType.objects.clear_cache()
|
|
blog = BlogA.objects.db_manager('secondary').get(pk=blog.id)
|
|
self.assertEqual(entry, blog.blogentry_set.using('secondary').get())
|
|
|
|
# Ensure no queries are made using the default database.
|
|
self.assertNumQueries(0, func)
|
|
|
|
def test_reverse_one_to_one_descriptor_on_non_default_database(self):
|
|
def func():
|
|
m2a = Model2A.objects.db_manager('secondary').create(field1='A1')
|
|
one2one = One2OneRelatingModel.objects.db_manager('secondary').create(one2one=m2a, field1='121')
|
|
ContentType.objects.clear_cache()
|
|
m2a = Model2A.objects.db_manager('secondary').get(pk=m2a.id)
|
|
self.assertEqual(one2one, m2a.one2onerelatingmodel)
|
|
|
|
# Ensure no queries are made using the default database.
|
|
self.assertNumQueries(0, func)
|
|
|
|
def test_many_to_many_descriptor_on_non_default_database(self):
|
|
def func():
|
|
m2a = Model2A.objects.db_manager('secondary').create(field1='A1')
|
|
rm = RelatingModel.objects.db_manager('secondary').create()
|
|
rm.many2many.add(m2a)
|
|
ContentType.objects.clear_cache()
|
|
m2a = Model2A.objects.db_manager('secondary').get(pk=m2a.id)
|
|
self.assertEqual(rm, m2a.relatingmodel_set.using('secondary').get())
|
|
|
|
# Ensure no queries are made using the default database.
|
|
self.assertNumQueries(0, func)
|
|
|
|
|
|
def qrepr(data):
|
|
"""
|
|
Ensure consistent repr() output for the QuerySet object.
|
|
"""
|
|
if isinstance(data, models.QuerySet):
|
|
if django.VERSION >= (1, 10):
|
|
return repr(data)
|
|
else:
|
|
return '<QuerySet %r>' % data
|
|
|
|
return repr(data)
|