Skip to content

Commit 2e21596

Browse files
committed
Rewrite bad comparison of instances against strings
- DRY on none files - no implicit casting of instances - support Django 4.1
1 parent 16066f7 commit 2e21596

File tree

1 file changed

+22
-29
lines changed

1 file changed

+22
-29
lines changed

testproj/testproj/testapp/tests.py

Lines changed: 22 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,12 @@ class IndexTest(TestCase):
99
def setUp(self):
1010
self.url = reverse("secret_list")
1111

12-
models.SecretFile.objects.create(filename="foo.txt", order=1, size=1024)
13-
models.SecretFile.objects.create(filename="bar.txt", order=2, size=512)
12+
self.foo_file = models.SecretFile.objects.create(
13+
filename="foo.txt", order=1, size=1024
14+
)
15+
self.bar_file = models.SecretFile.objects.create(
16+
filename="bar.txt", order=2, size=512
17+
)
1418

1519
def test_list(self):
1620
response = self.client.get(self.url)
@@ -19,16 +23,16 @@ def test_list(self):
1923

2024
def test_sorting_direction(self):
2125
response = self.client.get(self.url, {"sort": "order", "dir": "asc"})
22-
values = ["<SecretFile: #1 foo.txt>", "<SecretFile: #2 bar.txt>"]
26+
values = [self.foo_file, self.bar_file]
2327
self.assertQuerysetEqual(response.context["secret_files"], values)
2428

2529
response = self.client.get(self.url, {"sort": "order", "dir": "desc"})
26-
values = ["<SecretFile: #2 bar.txt>", "<SecretFile: #1 foo.txt>"]
30+
values = [self.bar_file, self.foo_file]
2731
self.assertQuerysetEqual(response.context["secret_files"], values)
2832

2933
# Nothing wrong happens with an invalid direction and ASC is used
3034
response = self.client.get(self.url, {"sort": "order", "dir": "NOT"})
31-
values = ["<SecretFile: #1 foo.txt>", "<SecretFile: #2 bar.txt>"]
35+
values = [self.foo_file, self.bar_file]
3236
self.assertQuerysetEqual(response.context["secret_files"], values)
3337

3438
def test_sorting_argument(self):
@@ -42,30 +46,28 @@ def setUp(self):
4246
self.nulls_first_url = reverse("nulls_first")
4347
self.nulls_last_url = reverse("nulls_last")
4448

45-
models.SecretFile.objects.create(filename="foo.txt", order=1, size=1024)
46-
models.SecretFile.objects.create(filename="bar.txt", order=2, size=512)
49+
self.foo_file = models.SecretFile.objects.create(
50+
filename="foo.txt", order=1, size=1024
51+
)
52+
self.bar_file = models.SecretFile.objects.create(
53+
filename="bar.txt", order=2, size=512
54+
)
55+
self.none_file = models.SecretFile.objects.create(
56+
filename=None, order=3, size=512
57+
)
4758

4859
def test_sorting_nulls_first(self):
4960
"""Verify None sorted field_name is in first places when sorting in asc and desc order"""
5061

51-
models.SecretFile.objects.create(filename=None, order=3, size=512)
5262
# asc order
53-
values = [
54-
"<SecretFile: #3 None>",
55-
"<SecretFile: #2 bar.txt>",
56-
"<SecretFile: #1 foo.txt>",
57-
]
63+
values = [self.none_file, self.bar_file, self.foo_file]
5864
response = self.client.get(
5965
self.nulls_first_url, {"sort": "filename", "nulls": "first", "dir": "asc"}
6066
)
6167
self.assertQuerysetEqual(list(response.context["secret_files"]), values)
6268

6369
# desc order
64-
values = [
65-
"<SecretFile: #3 None>",
66-
"<SecretFile: #1 foo.txt>",
67-
"<SecretFile: #2 bar.txt>",
68-
]
70+
values = [self.none_file, self.foo_file, self.bar_file]
6971
response = self.client.get(
7072
self.nulls_first_url,
7173
{"sort": "filename", "nulls": "first", "dir": "desc"},
@@ -75,24 +77,15 @@ def test_sorting_nulls_first(self):
7577
def test_sorting_nulls_last(self):
7678
"""Verify None sorted field_name is in last places when sorting in asc and desc order."""
7779

78-
models.SecretFile.objects.create(filename=None, order=3, size=512)
7980
# asc order
80-
values = [
81-
"<SecretFile: #2 bar.txt>",
82-
"<SecretFile: #1 foo.txt>",
83-
"<SecretFile: #3 None>",
84-
]
81+
values = [self.bar_file, self.foo_file, self.none_file]
8582
response = self.client.get(
8683
self.nulls_last_url, {"sort": "filename", "nulls": "last", "dir": "asc"}
8784
)
8885
self.assertQuerysetEqual(list(response.context["secret_files"]), values)
8986

9087
# desc order
91-
values = [
92-
"<SecretFile: #1 foo.txt>",
93-
"<SecretFile: #2 bar.txt>",
94-
"<SecretFile: #3 None>",
95-
]
88+
values = [self.foo_file, self.bar_file, self.none_file]
9689
response = self.client.get(
9790
self.nulls_last_url, {"sort": "filename", "nulls": "last", "dir": "desc"}
9891
)

0 commit comments

Comments
 (0)