summaryrefslogtreecommitdiff
path: root/tests/custom_pk
diff options
context:
space:
mode:
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],
)