diff options
Diffstat (limited to 'tests/custom_pk')
-rw-r--r-- | tests/custom_pk/tests.py | 68 |
1 files changed, 13 insertions, 55 deletions
diff --git a/tests/custom_pk/tests.py b/tests/custom_pk/tests.py index ee2d56efdd..cb9fb02133 100644 --- a/tests/custom_pk/tests.py +++ b/tests/custom_pk/tests.py @@ -25,39 +25,13 @@ class BasicCustomPKTests(TestCase): """ Both pk and custom attribute_name can be used in filter and friends """ - self.assertQuerysetEqual( - Employee.objects.filter(pk=123), - [ - "Dan Jones", - ], - str, - ) - - self.assertQuerysetEqual( - Employee.objects.filter(employee_code=123), - [ - "Dan Jones", - ], - str, - ) - - self.assertQuerysetEqual( + self.assertSequenceEqual(Employee.objects.filter(pk=123), [self.dan]) + self.assertSequenceEqual(Employee.objects.filter(employee_code=123), [self.dan]) + self.assertSequenceEqual( Employee.objects.filter(pk__in=[123, 456]), - [ - "Fran Bones", - "Dan Jones", - ], - str, - ) - - self.assertQuerysetEqual( - Employee.objects.all(), - [ - "Fran Bones", - "Dan Jones", - ], - str, + [self.fran, self.dan], ) + self.assertSequenceEqual(Employee.objects.all(), [self.fran, self.dan]) self.assertQuerysetEqual( Business.objects.filter(name="Sears"), ["Sears"], lambda b: b.name @@ -74,13 +48,9 @@ class BasicCustomPKTests(TestCase): """ Custom pk doesn't affect related_name based lookups """ - self.assertQuerysetEqual( + self.assertSequenceEqual( self.business.employees.all(), - [ - "Fran Bones", - "Dan Jones", - ], - str, + [self.fran, self.dan], ) self.assertQuerysetEqual( self.fran.business_set.all(), @@ -94,21 +64,13 @@ class BasicCustomPKTests(TestCase): """ Queries across tables, involving primary key """ - self.assertQuerysetEqual( + self.assertSequenceEqual( Employee.objects.filter(business__name="Sears"), - [ - "Fran Bones", - "Dan Jones", - ], - str, + [self.fran, self.dan], ) - self.assertQuerysetEqual( + self.assertSequenceEqual( Employee.objects.filter(business__pk="Sears"), - [ - "Fran Bones", - "Dan Jones", - ], - str, + [self.fran, self.dan], ) self.assertQuerysetEqual( @@ -186,13 +148,9 @@ class BasicCustomPKTests(TestCase): fran.last_name = "Jones" fran.save() - self.assertQuerysetEqual( + self.assertSequenceEqual( Employee.objects.filter(last_name="Jones"), - [ - "Dan Jones", - "Fran Jones", - ], - str, + [self.dan, fran], ) |