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

Fix QuerySte repr output for Django 1.10 in unit tests

fix_request_path_info
Diederik van der Boor 2016-08-05 17:21:53 +02:00
parent a5e348ffb9
commit 1a998733e1
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
polymorphic/tests.py
View File

@ -757,7 +757,7 @@ class PolymorphicTests(TestCase):
# no pretty printing # no pretty printing
ModelShow1_plain.objects.create(field1='abc') ModelShow1_plain.objects.create(field1='abc')
ModelShow2_plain.objects.create(field1='abc', field2='def') ModelShow2_plain.objects.create(field1='abc', field2='def')
self.assertEqual(repr(ModelShow1_plain.objects.all()), '[<ModelShow1_plain: ModelShow1_plain object>, <ModelShow2_plain: ModelShow2_plain object>]') self.assertEqual(qrepr(ModelShow1_plain.objects.all()), '<QuerySet [<ModelShow1_plain: ModelShow1_plain object>, <ModelShow2_plain: ModelShow2_plain object>]>')
def test_extra_method(self): def test_extra_method(self):
self.create_model2abcd() self.create_model2abcd()
@ -1288,3 +1288,16 @@ class MultipleDatabasesTests(TestCase):
# Ensure no queries are made using the default database. # Ensure no queries are made using the default database.
self.assertNumQueries(0, func) 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)