|
19 | 19 | from rest_framework.renderers import JSONRenderer |
20 | 20 |
|
21 | 21 | from .testapp.models import ( |
22 | | - User, Person, |
| 22 | + User, Person, Note, Post, Cover, |
23 | 23 | ) |
24 | 24 |
|
25 | 25 | from .testapp.filters import ( |
26 | 26 | UserFilter, |
27 | 27 | AllLookupsPersonDateFilter, |
28 | 28 | InSetLookupPersonIDFilter, |
29 | 29 | InSetLookupPersonNameFilter, |
| 30 | + PostFilter, |
| 31 | + CoverFilterWithRelatedMethodFilter, |
30 | 32 | ) |
31 | 33 |
|
32 | 34 |
|
@@ -284,3 +286,44 @@ def test_isnull_override(self): |
284 | 286 | results = list(filterset.qs) |
285 | 287 | self.assertEqual(len(results), 1) |
286 | 288 | self.assertEqual(results[0].username, 'user2') |
| 289 | + |
| 290 | + |
| 291 | +class FilterMethodTests(TestCase): |
| 292 | + """ |
| 293 | + Old test case for MethodFilter. Ensure that the new Filter.method remains compatible. |
| 294 | + """ |
| 295 | + |
| 296 | + @classmethod |
| 297 | + def setUpTestData(cls): |
| 298 | + user = User.objects.create(username="user1", email="user1@example.org") |
| 299 | + |
| 300 | + note1 = Note.objects.create(title="Test 1", content="Test content 1", author=user) |
| 301 | + note2 = Note.objects.create(title="Test 2", content="Test content 2", author=user) |
| 302 | + |
| 303 | + post1 = Post.objects.create(note=note1, content="Test content in post 1") |
| 304 | + post2 = Post.objects.create(note=note2, content="Test content in post 2", date_published=datetime.date.today()) |
| 305 | + |
| 306 | + Cover.objects.create(post=post1, comment="Cover 1") |
| 307 | + Cover.objects.create(post=post2, comment="Cover 2") |
| 308 | + |
| 309 | + def test_method_filter(self): |
| 310 | + GET = { |
| 311 | + 'is_published': 'true' |
| 312 | + } |
| 313 | + filterset = PostFilter(GET, queryset=Post.objects.all()) |
| 314 | + results = list(filterset.qs) |
| 315 | + self.assertEqual(len(results), 1) |
| 316 | + self.assertEqual(results[0].content, "Test content in post 2") |
| 317 | + |
| 318 | + def test_related_method_filter(self): |
| 319 | + """ |
| 320 | + Missing MethodFilter filter methods are silently ignored, returning |
| 321 | + the unfiltered queryset. |
| 322 | + """ |
| 323 | + GET = { |
| 324 | + 'post__is_published': 'true' |
| 325 | + } |
| 326 | + filterset = CoverFilterWithRelatedMethodFilter(GET, queryset=Cover.objects.all()) |
| 327 | + results = list(filterset.qs) |
| 328 | + self.assertEqual(len(results), 1) |
| 329 | + self.assertEqual(results[0].comment, "Cover 2") |
0 commit comments