Briq
/
django-polymorphic
6
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Test missing items bug in qset on child of PolymorphicModel w custom pk 

refs #66
fix_request_path_info
Frankie Dintino 2015-01-31 15:59:28 -05:00
parent bf0d37a91c
commit ad63c6a442
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
polymorphic/tests.py
View File

@ -283,6 +283,12 @@ class RelatedNameClash(ShowFieldType, PolymorphicModel):
class TestParentLinkAndRelatedName(ModelShow1_plain): class TestParentLinkAndRelatedName(ModelShow1_plain):
superclass = models.OneToOneField(ModelShow1_plain, parent_link=True, related_name='related_name_subclass') 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 PolymorphicTests(TestCase): class PolymorphicTests(TestCase):
""" """
@ -304,7 +310,6 @@ class PolymorphicTests(TestCase):
print('DiamondXY fields 1: field_b "{0}", field_x "{1}", field_y "{2}"'.format(o1.field_b, o1.field_x, o1.field_y)) print('DiamondXY fields 1: field_b "{0}", field_x "{1}", field_y "{2}"'.format(o1.field_b, o1.field_x, o1.field_y))
print('DiamondXY fields 2: field_b "{0}", field_x "{1}", field_y "{2}"'.format(o2.field_b, o2.field_x, o2.field_y)) print('DiamondXY fields 2: field_b "{0}", field_x "{1}", field_y "{2}"'.format(o2.field_b, o2.field_x, o2.field_y))
def test_annotate_aggregate_order(self): def test_annotate_aggregate_order(self):
# create a blog of type BlogA # create a blog of type BlogA
# create two blog entries in BlogA # create two blog entries in BlogA
@ -819,6 +824,13 @@ class PolymorphicTests(TestCase):
self.assertIsInstance(objects[0], ProxyModelA) self.assertIsInstance(objects[0], ProxyModelA)
self.assertIsInstance(objects[1], ProxyModelB) 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): def test_fix_getattribute(self):
### fixed issue in PolymorphicModel.__getattribute__: field name same as model name ### fixed issue in PolymorphicModel.__getattribute__: field name same as model name