/polymorphic/tests.py

https://bitbucket.org/bconstantin/django_polymorphic/ · Python · 639 lines · 557 code · 56 blank · 26 comment · 13 complexity · e21ee72a316fae161b6e6de0466ad7f6 MD5 · raw file

  1. # -*- coding: utf-8 -*-
  2. """ Test Cases
  3. Please see README.rst or DOCS.rst or http://bserve.webhop.org/wiki/django_polymorphic
  4. """
  5. import settings
  6. import sys
  7. from pprint import pprint
  8. from django import VERSION as django_VERSION
  9. from django.test import TestCase
  10. from django.db.models.query import QuerySet
  11. from django.db.models import Q,Count
  12. from django.db import models
  13. from django.contrib.contenttypes.models import ContentType
  14. from polymorphic import PolymorphicModel, PolymorphicManager, PolymorphicQuerySet
  15. from polymorphic import ShowFieldContent, ShowFieldType, ShowFieldTypeAndContent, get_version
  16. from polymorphic import translate_polymorphic_Q_object
  17. class PlainA(models.Model):
  18. field1 = models.CharField(max_length=10)
  19. class PlainB(PlainA):
  20. field2 = models.CharField(max_length=10)
  21. class PlainC(PlainB):
  22. field3 = models.CharField(max_length=10)
  23. class Model2A(ShowFieldType, PolymorphicModel):
  24. field1 = models.CharField(max_length=10)
  25. class Model2B(Model2A):
  26. field2 = models.CharField(max_length=10)
  27. class Model2C(Model2B):
  28. field3 = models.CharField(max_length=10)
  29. class Model2D(Model2C):
  30. field4 = models.CharField(max_length=10)
  31. class ModelExtraA(ShowFieldTypeAndContent, PolymorphicModel):
  32. field1 = models.CharField(max_length=10)
  33. class ModelExtraB(ModelExtraA):
  34. field2 = models.CharField(max_length=10)
  35. class ModelExtraC(ModelExtraB):
  36. field3 = models.CharField(max_length=10)
  37. class ModelExtraExternal(models.Model):
  38. topic = models.CharField(max_length=10)
  39. class ModelShow1(ShowFieldType,PolymorphicModel):
  40. field1 = models.CharField(max_length=10)
  41. m2m = models.ManyToManyField('self')
  42. class ModelShow2(ShowFieldContent, PolymorphicModel):
  43. field1 = models.CharField(max_length=10)
  44. m2m = models.ManyToManyField('self')
  45. class ModelShow3(ShowFieldTypeAndContent, PolymorphicModel):
  46. field1 = models.CharField(max_length=10)
  47. m2m = models.ManyToManyField('self')
  48. class ModelShow1_plain(PolymorphicModel):
  49. field1 = models.CharField(max_length=10)
  50. class ModelShow2_plain(ModelShow1_plain):
  51. field2 = models.CharField(max_length=10)
  52. class Base(ShowFieldType, PolymorphicModel):
  53. field_b = models.CharField(max_length=10)
  54. class ModelX(Base):
  55. field_x = models.CharField(max_length=10)
  56. class ModelY(Base):
  57. field_y = models.CharField(max_length=10)
  58. class Enhance_Plain(models.Model):
  59. field_p = models.CharField(max_length=10)
  60. class Enhance_Base(ShowFieldTypeAndContent, PolymorphicModel):
  61. field_b = models.CharField(max_length=10)
  62. class Enhance_Inherit(Enhance_Base, Enhance_Plain):
  63. field_i = models.CharField(max_length=10)
  64. class DiamondBase(models.Model):
  65. field_b = models.CharField(max_length=10)
  66. class DiamondX(DiamondBase):
  67. field_x = models.CharField(max_length=10)
  68. class DiamondY(DiamondBase):
  69. field_y = models.CharField(max_length=10)
  70. class DiamondXY(DiamondX, DiamondY):
  71. pass
  72. class RelationBase(ShowFieldTypeAndContent, PolymorphicModel):
  73. field_base = models.CharField(max_length=10)
  74. fk = models.ForeignKey('self', null=True)
  75. m2m = models.ManyToManyField('self')
  76. class RelationA(RelationBase):
  77. field_a = models.CharField(max_length=10)
  78. class RelationB(RelationBase):
  79. field_b = models.CharField(max_length=10)
  80. class RelationBC(RelationB):
  81. field_c = models.CharField(max_length=10)
  82. class RelatingModel(models.Model):
  83. many2many = models.ManyToManyField(Model2A)
  84. class One2OneRelatingModel(PolymorphicModel):
  85. one2one = models.OneToOneField(Model2A)
  86. field1 = models.CharField(max_length=10)
  87. class One2OneRelatingModelDerived(One2OneRelatingModel):
  88. field2 = models.CharField(max_length=10)
  89. class MyManager(PolymorphicManager):
  90. def get_query_set(self):
  91. return super(MyManager, self).get_query_set().order_by('-field1')
  92. class ModelWithMyManager(ShowFieldTypeAndContent, Model2A):
  93. objects = MyManager()
  94. field4 = models.CharField(max_length=10)
  95. class MROBase1(ShowFieldType, PolymorphicModel):
  96. objects = MyManager()
  97. field1 = models.CharField(max_length=10) # needed as MyManager uses it
  98. class MROBase2(MROBase1):
  99. pass # Django vanilla inheritance does not inherit MyManager as _default_manager here
  100. class MROBase3(models.Model):
  101. objects = PolymorphicManager()
  102. class MRODerived(MROBase2, MROBase3):
  103. pass
  104. class MgrInheritA(models.Model):
  105. mgrA = models.Manager()
  106. mgrA2 = models.Manager()
  107. field1 = models.CharField(max_length=10)
  108. class MgrInheritB(MgrInheritA):
  109. mgrB = models.Manager()
  110. field2 = models.CharField(max_length=10)
  111. class MgrInheritC(ShowFieldTypeAndContent, MgrInheritB):
  112. pass
  113. class BlogBase(ShowFieldTypeAndContent, PolymorphicModel):
  114. name = models.CharField(max_length=10)
  115. class BlogA(BlogBase):
  116. info = models.CharField(max_length=10)
  117. class BlogB(BlogBase):
  118. pass
  119. class BlogEntry(ShowFieldTypeAndContent, PolymorphicModel):
  120. blog = models.ForeignKey(BlogA)
  121. text = models.CharField(max_length=10)
  122. class BlogEntry_limit_choices_to(ShowFieldTypeAndContent, PolymorphicModel):
  123. blog = models.ForeignKey(BlogBase)
  124. text = models.CharField(max_length=10)
  125. class ModelFieldNameTest(ShowFieldType, PolymorphicModel):
  126. modelfieldnametest = models.CharField(max_length=10)
  127. class InitTestModel(ShowFieldType, PolymorphicModel):
  128. bar = models.CharField(max_length=100)
  129. def __init__(self, *args, **kwargs):
  130. kwargs['bar'] = self.x()
  131. super(InitTestModel, self).__init__(*args, **kwargs)
  132. class InitTestModelSubclass(InitTestModel):
  133. def x(self):
  134. return 'XYZ'
  135. # models from github issue
  136. class Top(PolymorphicModel):
  137. name = models.CharField(max_length=50)
  138. class Middle(Top):
  139. description = models.TextField()
  140. class Bottom(Middle):
  141. author = models.CharField(max_length=50)
  142. # UUID tests won't work with Django 1.1
  143. if not (django_VERSION[0] <= 1 and django_VERSION[1] <= 1):
  144. try: from polymorphic.tools_for_tests import UUIDField
  145. except: pass
  146. if 'UUIDField' in globals():
  147. import uuid
  148. class UUIDProject(ShowFieldTypeAndContent, PolymorphicModel):
  149. uuid_primary_key = UUIDField(primary_key = True)
  150. topic = models.CharField(max_length = 30)
  151. class UUIDArtProject(UUIDProject):
  152. artist = models.CharField(max_length = 30)
  153. class UUIDResearchProject(UUIDProject):
  154. supervisor = models.CharField(max_length = 30)
  155. class UUIDPlainA(models.Model):
  156. uuid_primary_key = UUIDField(primary_key = True)
  157. field1 = models.CharField(max_length=10)
  158. class UUIDPlainB(UUIDPlainA):
  159. field2 = models.CharField(max_length=10)
  160. class UUIDPlainC(UUIDPlainB):
  161. field3 = models.CharField(max_length=10)
  162. # test bad field name
  163. #class TestBadFieldModel(ShowFieldType, PolymorphicModel):
  164. # instance_of = models.CharField(max_length=10)
  165. # validation error: "polymorphic.relatednameclash: Accessor for field 'polymorphic_ctype' clashes
  166. # with related field 'ContentType.relatednameclash_set'." (reported by Andrew Ingram)
  167. # fixed with related_name
  168. class RelatedNameClash(ShowFieldType, PolymorphicModel):
  169. ctype = models.ForeignKey(ContentType, null=True, editable=False)
  170. class testclass(TestCase):
  171. def test_diamond_inheritance(self):
  172. # Django diamond problem
  173. o = DiamondXY.objects.create(field_b='b', field_x='x', field_y='y')
  174. print 'DiamondXY fields 1: field_b "%s", field_x "%s", field_y "%s"' % (o.field_b, o.field_x, o.field_y)
  175. o = DiamondXY.objects.get()
  176. print 'DiamondXY fields 2: field_b "%s", field_x "%s", field_y "%s"' % (o.field_b, o.field_x, o.field_y)
  177. if o.field_b != 'b':
  178. print
  179. print '# known django model inheritance diamond problem detected'
  180. def test_annotate_aggregate_order(self):
  181. # create a blog of type BlogA
  182. blog = BlogA.objects.create(name='B1', info='i1')
  183. # create two blog entries in BlogA
  184. entry1 = blog.blogentry_set.create(text='bla')
  185. entry2 = BlogEntry.objects.create(blog=blog, text='bla2')
  186. # create some blogs of type BlogB to make the BlogBase table data really polymorphic
  187. o = BlogB.objects.create(name='Bb1')
  188. o = BlogB.objects.create(name='Bb2')
  189. o = BlogB.objects.create(name='Bb3')
  190. qs = BlogBase.objects.annotate(entrycount=Count('BlogA___blogentry'))
  191. assert len(qs)==4
  192. for o in qs:
  193. if o.name=='B1':
  194. assert o.entrycount == 2
  195. else:
  196. assert o.entrycount == 0
  197. x = BlogBase.objects.aggregate(entrycount=Count('BlogA___blogentry'))
  198. assert x['entrycount'] == 2
  199. # create some more blogs for next test
  200. b2 = BlogA.objects.create(name='B2', info='i2')
  201. b2 = BlogA.objects.create(name='B3', info='i3')
  202. b2 = BlogA.objects.create(name='B4', info='i4')
  203. b2 = BlogA.objects.create(name='B5', info='i5')
  204. ### test ordering for field in all entries
  205. expected = '''
  206. [ <BlogB: id 4, name (CharField) "Bb3">,
  207. <BlogB: id 3, name (CharField) "Bb2">,
  208. <BlogB: id 2, name (CharField) "Bb1">,
  209. <BlogA: id 8, name (CharField) "B5", info (CharField) "i5">,
  210. <BlogA: id 7, name (CharField) "B4", info (CharField) "i4">,
  211. <BlogA: id 6, name (CharField) "B3", info (CharField) "i3">,
  212. <BlogA: id 5, name (CharField) "B2", info (CharField) "i2">,
  213. <BlogA: id 1, name (CharField) "B1", info (CharField) "i1"> ]'''
  214. x = '\n' + repr(BlogBase.objects.order_by('-name'))
  215. assert x == expected
  216. ### test ordering for field in one subclass only
  217. # MySQL and SQLite return this order
  218. expected1='''
  219. [ <BlogA: id 8, name (CharField) "B5", info (CharField) "i5">,
  220. <BlogA: id 7, name (CharField) "B4", info (CharField) "i4">,
  221. <BlogA: id 6, name (CharField) "B3", info (CharField) "i3">,
  222. <BlogA: id 5, name (CharField) "B2", info (CharField) "i2">,
  223. <BlogA: id 1, name (CharField) "B1", info (CharField) "i1">,
  224. <BlogB: id 2, name (CharField) "Bb1">,
  225. <BlogB: id 3, name (CharField) "Bb2">,
  226. <BlogB: id 4, name (CharField) "Bb3"> ]'''
  227. # PostgreSQL returns this order
  228. expected2='''
  229. [ <BlogB: id 2, name (CharField) "Bb1">,
  230. <BlogB: id 3, name (CharField) "Bb2">,
  231. <BlogB: id 4, name (CharField) "Bb3">,
  232. <BlogA: id 8, name (CharField) "B5", info (CharField) "i5">,
  233. <BlogA: id 7, name (CharField) "B4", info (CharField) "i4">,
  234. <BlogA: id 6, name (CharField) "B3", info (CharField) "i3">,
  235. <BlogA: id 5, name (CharField) "B2", info (CharField) "i2">,
  236. <BlogA: id 1, name (CharField) "B1", info (CharField) "i1"> ]'''
  237. x = '\n' + repr(BlogBase.objects.order_by('-BlogA___info'))
  238. assert x == expected1 or x == expected2
  239. def test_limit_choices_to(self):
  240. "this is not really a testcase, as limit_choices_to only affects the Django admin"
  241. # create a blog of type BlogA
  242. blog_a = BlogA.objects.create(name='aa', info='aa')
  243. blog_b = BlogB.objects.create(name='bb')
  244. # create two blog entries
  245. entry1 = BlogEntry_limit_choices_to.objects.create(blog=blog_b, text='bla2')
  246. entry2 = BlogEntry_limit_choices_to.objects.create(blog=blog_b, text='bla2')
  247. def test_primary_key_custom_field_problem(self):
  248. "object retrieval problem occuring with some custom primary key fields (UUIDField as test case)"
  249. if not 'UUIDField' in globals(): return
  250. a=UUIDProject.objects.create(topic="John's gathering")
  251. b=UUIDArtProject.objects.create(topic="Sculpting with Tim", artist="T. Turner")
  252. c=UUIDResearchProject.objects.create(topic="Swallow Aerodynamics", supervisor="Dr. Winter")
  253. qs=UUIDProject.objects.all()
  254. ol=list(qs)
  255. a=qs[0]
  256. b=qs[1]
  257. c=qs[2]
  258. assert len(qs)==3
  259. assert type(a.uuid_primary_key)==uuid.UUID and type(a.pk)==uuid.UUID
  260. res=repr(qs)
  261. import re
  262. res=re.sub(' "(.*?)..", topic',', topic',res)
  263. res_exp="""[ <UUIDProject: uuid_primary_key (UUIDField/pk), topic (CharField) "John's gathering">,
  264. <UUIDArtProject: uuid_primary_key (UUIDField/pk), topic (CharField) "Sculpting with Tim", artist (CharField) "T. Turner">,
  265. <UUIDResearchProject: uuid_primary_key (UUIDField/pk), topic (CharField) "Swallow Aerodynamics", supervisor (CharField) "Dr. Winter"> ]"""
  266. assert res==res_exp, res
  267. #if (a.pk!= uuid.UUID or c.pk!= uuid.UUID):
  268. # print
  269. # print '# known inconstency with custom primary key field detected (django problem?)'
  270. a=UUIDPlainA.objects.create(field1='A1')
  271. b=UUIDPlainB.objects.create(field1='B1', field2='B2')
  272. c=UUIDPlainC.objects.create(field1='C1', field2='C2', field3='C3')
  273. qs=UUIDPlainA.objects.all()
  274. if (a.pk!= uuid.UUID or c.pk!= uuid.UUID):
  275. print
  276. print '# known type inconstency with custom primary key field detected (django problem?)'
  277. def show_base_manager(model):
  278. print type(model._base_manager),model._base_manager.model
  279. __test__ = {"doctest": """
  280. #######################################################
  281. ### Tests
  282. >>> settings.DEBUG=True
  283. ### simple inheritance
  284. >>> o=Model2A.objects.create(field1='A1')
  285. >>> o=Model2B.objects.create(field1='B1', field2='B2')
  286. >>> o=Model2C.objects.create(field1='C1', field2='C2', field3='C3')
  287. >>> o=Model2D.objects.create(field1='D1', field2='D2', field3='D3', field4='D4')
  288. >>> Model2A.objects.all()
  289. [ <Model2A: id 1, field1 (CharField)>,
  290. <Model2B: id 2, field1 (CharField), field2 (CharField)>,
  291. <Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>,
  292. <Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)> ]
  293. # manual get_real_instance()
  294. >>> o=Model2A.objects.non_polymorphic().get(field1='C1')
  295. >>> o.get_real_instance()
  296. <Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>
  297. # non_polymorphic()
  298. >>> qs=Model2A.objects.all().non_polymorphic()
  299. >>> qs
  300. [ <Model2A: id 1, field1 (CharField)>,
  301. <Model2A: id 2, field1 (CharField)>,
  302. <Model2A: id 3, field1 (CharField)>,
  303. <Model2A: id 4, field1 (CharField)> ]
  304. # get_real_instances()
  305. >>> qs.get_real_instances()
  306. [ <Model2A: id 1, field1 (CharField)>,
  307. <Model2B: id 2, field1 (CharField), field2 (CharField)>,
  308. <Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>,
  309. <Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)> ]
  310. >>> l=list(qs)
  311. >>> Model2A.objects.get_real_instances(l)
  312. [ <Model2A: id 1, field1 (CharField)>,
  313. <Model2B: id 2, field1 (CharField), field2 (CharField)>,
  314. <Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>,
  315. <Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)> ]
  316. # translate_polymorphic_Q_object
  317. >>> q=Model2A.translate_polymorphic_Q_object( Q(instance_of=Model2C) )
  318. >>> Model2A.objects.filter(q)
  319. [ <Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>,
  320. <Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)> ]
  321. ### test inheritance pointers & _base_managers
  322. >>> show_base_manager(PlainA)
  323. <class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.PlainA'>
  324. >>> show_base_manager(PlainB)
  325. <class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.PlainB'>
  326. >>> show_base_manager(PlainC)
  327. <class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.PlainC'>
  328. >>> show_base_manager(Model2A)
  329. <class 'polymorphic.manager.PolymorphicManager'> <class 'polymorphic.tests.Model2A'>
  330. >>> show_base_manager(Model2B)
  331. <class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.Model2B'>
  332. >>> show_base_manager(Model2C)
  333. <class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.Model2C'>
  334. >>> show_base_manager(One2OneRelatingModel)
  335. <class 'polymorphic.manager.PolymorphicManager'> <class 'polymorphic.tests.One2OneRelatingModel'>
  336. >>> show_base_manager(One2OneRelatingModelDerived)
  337. <class 'django.db.models.manager.Manager'> <class 'polymorphic.tests.One2OneRelatingModelDerived'>
  338. >>> o=Model2A.base_objects.get(field1='C1')
  339. >>> o.model2b
  340. <Model2B: id 3, field1 (CharField), field2 (CharField)>
  341. >>> o=Model2B.base_objects.get(field1='C1')
  342. >>> o.model2c
  343. <Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>
  344. ### OneToOneField, test both directions for polymorphism
  345. >>> a=Model2A.base_objects.get(field1='C1')
  346. >>> b=One2OneRelatingModelDerived.objects.create(one2one=a, field1='f1', field2='f2')
  347. >>> b.one2one # this result is basically wrong, probably due to Django cacheing (we used base_objects), but should not be a problem
  348. <Model2A: id 3, field1 (CharField)>
  349. >>> c=One2OneRelatingModelDerived.objects.get(field1='f1')
  350. >>> c.one2one
  351. <Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>
  352. >>> a.one2onerelatingmodel
  353. <One2OneRelatingModelDerived: One2OneRelatingModelDerived object>
  354. ### ShowFieldContent, ShowFieldType, ShowFieldTypeAndContent, also with annotate()
  355. >>> o=ModelShow1.objects.create(field1='abc')
  356. >>> o.m2m.add(o) ; o.save()
  357. >>> ModelShow1.objects.all()
  358. [ <ModelShow1: id 1, field1 (CharField), m2m (ManyToManyField)> ]
  359. >>> o=ModelShow2.objects.create(field1='abc')
  360. >>> o.m2m.add(o) ; o.save()
  361. >>> ModelShow2.objects.all()
  362. [ <ModelShow2: id 1, field1 "abc", m2m 1> ]
  363. >>> o=ModelShow3.objects.create(field1='abc')
  364. >>> o.m2m.add(o) ; o.save()
  365. >>> ModelShow3.objects.all()
  366. [ <ModelShow3: id 1, field1 (CharField) "abc", m2m (ManyToManyField) 1> ]
  367. >>> ModelShow1.objects.all().annotate(Count('m2m'))
  368. [ <ModelShow1: id 1, field1 (CharField), m2m (ManyToManyField) - Ann: m2m__count (int)> ]
  369. >>> ModelShow2.objects.all().annotate(Count('m2m'))
  370. [ <ModelShow2: id 1, field1 "abc", m2m 1 - Ann: m2m__count 1> ]
  371. >>> ModelShow3.objects.all().annotate(Count('m2m'))
  372. [ <ModelShow3: id 1, field1 (CharField) "abc", m2m (ManyToManyField) 1 - Ann: m2m__count (int) 1> ]
  373. # no pretty printing
  374. >>> o=ModelShow1_plain.objects.create(field1='abc')
  375. >>> o=ModelShow2_plain.objects.create(field1='abc', field2='def')
  376. >>> ModelShow1_plain.objects.all()
  377. [<ModelShow1_plain: ModelShow1_plain object>, <ModelShow2_plain: ModelShow2_plain object>]
  378. ### extra() method
  379. >>> Model2A.objects.extra(where=['id IN (2, 3)'])
  380. [ <Model2B: id 2, field1 (CharField), field2 (CharField)>,
  381. <Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)> ]
  382. >>> Model2A.objects.extra(select={"select_test": "field1 = 'A1'"}, where=["field1 = 'A1' OR field1 = 'B1'"], order_by = ['-id'] )
  383. [ <Model2B: id 2, field1 (CharField), field2 (CharField) - Extra: select_test (int)>,
  384. <Model2A: id 1, field1 (CharField) - Extra: select_test (int)> ]
  385. >>> o=ModelExtraA.objects.create(field1='A1')
  386. >>> o=ModelExtraB.objects.create(field1='B1', field2='B2')
  387. >>> o=ModelExtraC.objects.create(field1='C1', field2='C2', field3='C3')
  388. >>> o=ModelExtraExternal.objects.create(topic='extra1')
  389. >>> o=ModelExtraExternal.objects.create(topic='extra2')
  390. >>> o=ModelExtraExternal.objects.create(topic='extra3')
  391. >>> ModelExtraA.objects.extra(tables=["polymorphic_modelextraexternal"], select={"topic":"polymorphic_modelextraexternal.topic"}, where=["polymorphic_modelextraa.id = polymorphic_modelextraexternal.id"] )
  392. [ <ModelExtraA: id 1, field1 (CharField) "A1" - Extra: topic (unicode) "extra1">,
  393. <ModelExtraB: id 2, field1 (CharField) "B1", field2 (CharField) "B2" - Extra: topic (unicode) "extra2">,
  394. <ModelExtraC: id 3, field1 (CharField) "C1", field2 (CharField) "C2", field3 (CharField) "C3" - Extra: topic (unicode) "extra3"> ]
  395. ### class filtering, instance_of, not_instance_of
  396. >>> Model2A.objects.instance_of(Model2B)
  397. [ <Model2B: id 2, field1 (CharField), field2 (CharField)>,
  398. <Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>,
  399. <Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)> ]
  400. >>> Model2A.objects.filter(instance_of=Model2B)
  401. [ <Model2B: id 2, field1 (CharField), field2 (CharField)>,
  402. <Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>,
  403. <Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)> ]
  404. >>> Model2A.objects.filter(Q(instance_of=Model2B))
  405. [ <Model2B: id 2, field1 (CharField), field2 (CharField)>,
  406. <Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>,
  407. <Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)> ]
  408. >>> Model2A.objects.not_instance_of(Model2B)
  409. [ <Model2A: id 1, field1 (CharField)> ]
  410. ### polymorphic filtering
  411. >>> Model2A.objects.filter( Q( Model2B___field2 = 'B2' ) | Q( Model2C___field3 = 'C3' ) )
  412. [ <Model2B: id 2, field1 (CharField), field2 (CharField)>,
  413. <Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)> ]
  414. ### get & delete
  415. >>> oa=Model2A.objects.get(id=2)
  416. >>> oa
  417. <Model2B: id 2, field1 (CharField), field2 (CharField)>
  418. >>> oa.delete()
  419. >>> Model2A.objects.all()
  420. [ <Model2A: id 1, field1 (CharField)>,
  421. <Model2C: id 3, field1 (CharField), field2 (CharField), field3 (CharField)>,
  422. <Model2D: id 4, field1 (CharField), field2 (CharField), field3 (CharField), field4 (CharField)> ]
  423. ### queryset combining
  424. >>> o=ModelX.objects.create(field_x='x')
  425. >>> o=ModelY.objects.create(field_y='y')
  426. >>> Base.objects.instance_of(ModelX) | Base.objects.instance_of(ModelY)
  427. [ <ModelX: id 1, field_b (CharField), field_x (CharField)>,
  428. <ModelY: id 2, field_b (CharField), field_y (CharField)> ]
  429. ### multiple inheritance, subclassing third party models (mix PolymorphicModel with models.Model)
  430. >>> o = Enhance_Base.objects.create(field_b='b-base')
  431. >>> o = Enhance_Inherit.objects.create(field_b='b-inherit', field_p='p', field_i='i')
  432. >>> Enhance_Base.objects.all()
  433. [ <Enhance_Base: id 1, field_b (CharField) "b-base">,
  434. <Enhance_Inherit: id 2, field_b (CharField) "b-inherit", field_p (CharField) "p", field_i (CharField) "i"> ]
  435. ### ForeignKey, ManyToManyField
  436. >>> obase=RelationBase.objects.create(field_base='base')
  437. >>> oa=RelationA.objects.create(field_base='A1', field_a='A2', fk=obase)
  438. >>> ob=RelationB.objects.create(field_base='B1', field_b='B2', fk=oa)
  439. >>> oc=RelationBC.objects.create(field_base='C1', field_b='C2', field_c='C3', fk=oa)
  440. >>> oa.m2m.add(oa); oa.m2m.add(ob)
  441. >>> RelationBase.objects.all()
  442. [ <RelationBase: id 1, field_base (CharField) "base", fk (ForeignKey) None, m2m (ManyToManyField) 0>,
  443. <RelationA: id 2, field_base (CharField) "A1", fk (ForeignKey) RelationBase, field_a (CharField) "A2", m2m (ManyToManyField) 2>,
  444. <RelationB: id 3, field_base (CharField) "B1", fk (ForeignKey) RelationA, field_b (CharField) "B2", m2m (ManyToManyField) 1>,
  445. <RelationBC: id 4, field_base (CharField) "C1", fk (ForeignKey) RelationA, field_b (CharField) "C2", field_c (CharField) "C3", m2m (ManyToManyField) 0> ]
  446. >>> oa=RelationBase.objects.get(id=2)
  447. >>> oa.fk
  448. <RelationBase: id 1, field_base (CharField) "base", fk (ForeignKey) None, m2m (ManyToManyField) 0>
  449. >>> oa.relationbase_set.all()
  450. [ <RelationB: id 3, field_base (CharField) "B1", fk (ForeignKey) RelationA, field_b (CharField) "B2", m2m (ManyToManyField) 1>,
  451. <RelationBC: id 4, field_base (CharField) "C1", fk (ForeignKey) RelationA, field_b (CharField) "C2", field_c (CharField) "C3", m2m (ManyToManyField) 0> ]
  452. >>> ob=RelationBase.objects.get(id=3)
  453. >>> ob.fk
  454. <RelationA: id 2, field_base (CharField) "A1", fk (ForeignKey) RelationBase, field_a (CharField) "A2", m2m (ManyToManyField) 2>
  455. >>> oa=RelationA.objects.get()
  456. >>> oa.m2m.all()
  457. [ <RelationA: id 2, field_base (CharField) "A1", fk (ForeignKey) RelationBase, field_a (CharField) "A2", m2m (ManyToManyField) 2>,
  458. <RelationB: id 3, field_base (CharField) "B1", fk (ForeignKey) RelationA, field_b (CharField) "B2", m2m (ManyToManyField) 1> ]
  459. ### user-defined manager
  460. >>> o=ModelWithMyManager.objects.create(field1='D1a', field4='D4a')
  461. >>> o=ModelWithMyManager.objects.create(field1='D1b', field4='D4b')
  462. >>> ModelWithMyManager.objects.all()
  463. [ <ModelWithMyManager: id 6, field1 (CharField) "D1b", field4 (CharField) "D4b">,
  464. <ModelWithMyManager: id 5, field1 (CharField) "D1a", field4 (CharField) "D4a"> ]
  465. >>> type(ModelWithMyManager.objects)
  466. <class 'polymorphic.tests.MyManager'>
  467. >>> type(ModelWithMyManager._default_manager)
  468. <class 'polymorphic.tests.MyManager'>
  469. ### Manager Inheritance
  470. >>> type(MRODerived.objects) # MRO
  471. <class 'polymorphic.tests.MyManager'>
  472. # check for correct default manager
  473. >>> type(MROBase1._default_manager)
  474. <class 'polymorphic.tests.MyManager'>
  475. # Django vanilla inheritance does not inherit MyManager as _default_manager here
  476. >>> type(MROBase2._default_manager)
  477. <class 'polymorphic.tests.MyManager'>
  478. ### fixed issue in PolymorphicModel.__getattribute__: field name same as model name
  479. >>> ModelFieldNameTest.objects.create(modelfieldnametest='1')
  480. <ModelFieldNameTest: id 1, modelfieldnametest (CharField)>
  481. ### fixed issue in PolymorphicModel.__getattribute__:
  482. # if subclass defined __init__ and accessed class members, __getattribute__ had a problem: "...has no attribute 'sub_and_superclass_dict'"
  483. #>>> o
  484. >>> o = InitTestModelSubclass.objects.create()
  485. >>> o.bar
  486. 'XYZ'
  487. ### Django model inheritance diamond problem, fails for Django 1.1
  488. #>>> o=DiamondXY.objects.create(field_b='b', field_x='x', field_y='y')
  489. #>>> print 'DiamondXY fields 1: field_b "%s", field_x "%s", field_y "%s"' % (o.field_b, o.field_x, o.field_y)
  490. #DiamondXY fields 1: field_b "a", field_x "x", field_y "y"
  491. # test for github issue
  492. >>> t = Top()
  493. >>> t.save()
  494. >>> m = Middle()
  495. >>> m.save()
  496. >>> b = Bottom()
  497. >>> b.save()
  498. >>> Top.objects.all()
  499. [<Top: Top object>, <Middle: Middle object>, <Bottom: Bottom object>]
  500. >>> Middle.objects.all()
  501. [<Middle: Middle object>, <Bottom: Bottom object>]
  502. >>> Bottom.objects.all()
  503. [<Bottom: Bottom object>]
  504. >>> settings.DEBUG=False
  505. """}