Skip to content

Commit e5caa29

Browse files
committed
Test fixed for pagination
Signed-off-by: RISHI GARG <134256793+Rishi-source@users.noreply.github.com>
1 parent 7c2a138 commit e5caa29

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

vulnerabilities/tests/test_pagination.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def setUpTestData(cls):
2727
)
2828

2929
def test_default_pagination(self):
30-
response = self.client.get(reverse("package_search"))
30+
response = self.client.get(reverse("package_search") + "?search=test")
3131
self.assertEqual(response.status_code, 200)
3232
page_obj = response.context["page_obj"]
3333
self.assertIsNotNone(page_obj)
@@ -38,13 +38,13 @@ def test_default_pagination(self):
3838
def test_page_size_variations(self):
3939
valid_page_sizes = [20, 50, 100]
4040
for size in valid_page_sizes:
41-
url = f"{reverse('package_search')}?page_size={size}"
41+
url = f"{reverse('package_search')}?search=test&page_size={size}"
4242
response = self.client.get(url)
4343
self.assertEqual(response.status_code, 200)
44-
self.assertIn(response.context["current_page_size"], [20, size])
44+
self.assertEqual(response.context["current_page_size"], size)
4545

4646
def test_page_navigation(self):
47-
response = self.client.get(reverse("package_search"))
47+
response = self.client.get(reverse("package_search") + "?search=test")
4848
first_page = response.context["page_obj"]
4949
self.assertEqual(first_page.number, 1)
5050
self.assertTrue(first_page.has_next())
@@ -78,7 +78,7 @@ def test_invalid_page_size_inputs(self):
7878
"",
7979
]
8080
for input_value in malicious_inputs:
81-
url = f"{reverse('package_search')}?page_size={input_value}"
81+
url = f"{reverse('package_search')}?search=test&page_size={input_value}"
8282
response = self.client.get(url)
8383
self.assertEqual(response.status_code, 200)
8484
self.assertEqual(response.context["current_page_size"], 20)
@@ -93,8 +93,8 @@ def test_sql_injection_prevention(self):
9393
initial_package_count = Package.objects.count()
9494
for payload in sql_injection_payloads:
9595
urls = [
96-
f"{reverse('package_search')}?page={payload}",
97-
f"{reverse('package_search')}?page_size={payload}",
96+
f"{reverse('package_search')}?search=test&page={payload}",
97+
f"{reverse('package_search')}?search=test&page_size={payload}",
9898
]
9999
for url in urls:
100100
response = self.client.get(url)
@@ -114,15 +114,15 @@ def setUpTestData(cls):
114114
)
115115

116116
def test_small_dataset_pagination(self):
117-
response = self.client.get(reverse("package_search"))
117+
response = self.client.get(reverse("package_search") + "?search=test")
118118
self.assertEqual(response.status_code, 200)
119119
self.assertLessEqual(len(response.context["page_obj"].object_list), 20)
120120

121121
def test_out_of_range_page_number(self):
122122
out_of_range_urls = [
123-
f"{reverse('package_search')}?page=9999",
124-
f"{reverse('package_search')}?page=-5",
125-
f"{reverse('package_search')}?page=abc",
123+
f"{reverse('package_search')}?search=test&page=9999",
124+
f"{reverse('package_search')}?search=test&page=-5",
125+
f"{reverse('package_search')}?search=test&page=abc",
126126
]
127127
for url in out_of_range_urls:
128128
response = self.client.get(url)

0 commit comments

Comments
 (0)