summaryrefslogtreecommitdiff
path: root/tests/custom_pk
diff options
context:
space:
mode:
authorMariusz Felisiak <felisiak.mariusz@gmail.com>2022-10-07 13:05:35 +0200
committerGitHub <noreply@github.com>2022-10-07 13:05:35 +0200
commitd795259ea96004df0a2469246229a146307bcd2c (patch)
tree96d797dacdb7b4c68cd32afc1f2ed6f15e0a2bcd /tests/custom_pk
parent564b317fb53e9925b86ca5ef5d3bbcf99387602c (diff)
downloaddjango-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.py68
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],
)