diff options
author | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2022-10-07 13:05:35 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-10-07 13:05:35 +0200 |
commit | d795259ea96004df0a2469246229a146307bcd2c (patch) | |
tree | 96d797dacdb7b4c68cd32afc1f2ed6f15e0a2bcd /tests/custom_pk | |
parent | 564b317fb53e9925b86ca5ef5d3bbcf99387602c (diff) | |
download | django-d795259ea96004df0a2469246229a146307bcd2c.tar.gz |
Replaced assertQuerysetEqual() to assertSequenceEqual()/assertCountEqual() where appropriate.
Follow up to 3f7b3275627385f8f7531fca01cdda50d4ec6b6e.
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], ) |