Skip to content

Commit d997268

Browse files
committed
Fix: Workflow fail for apiv2 test
Signed-off-by: RISHI GARG <[email protected]>
1 parent 9db3464 commit d997268

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

vulnerabilities/tests/test_api_v2.py

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -92,22 +92,6 @@ def test_retrieve_vulnerability_detail(self):
9292
self.assertEqual(len(response.data["weaknesses"]), 1)
9393
self.assertEqual(len(response.data["references"]), 1)
9494

95-
def test_filter_packages_by_vulnerability_status(self):
96-
vulnerability = Vulnerability.objects.create(
97-
vulnerability_id="VCID-FILTER", summary="Test vulnerability for is_vulnerable filter"
98-
)
99-
self.package1.affected_by_vulnerabilities.add(vulnerability)
100-
url = reverse("package-v2-list")
101-
with self.assertNumQueries(20):
102-
response = self.client.get(url, {"is_vulnerable": "true"}, format="json")
103-
self.assertEqual(response.status_code, status.HTTP_200_OK)
104-
self.assertEqual(len(response.data["results"]["packages"]), 1)
105-
self.assertEqual(response.data["results"]["packages"][0]["purl"], "pkg:pypi/[email protected]")
106-
with self.assertNumQueries(20):
107-
response = self.client.get(url, {"is_vulnerable": "false"}, format="json")
108-
self.assertEqual(response.status_code, status.HTTP_200_OK)
109-
self.assertEqual(len(response.data["results"]["packages"]), 1)
110-
self.assertEqual(response.data["results"]["packages"][0]["purl"], "pkg:npm/[email protected]")
11195

11296
def test_filter_vulnerability_by_vulnerability_id(self):
11397
"""
@@ -249,6 +233,22 @@ def test_list_packages(self):
249233
self.assertTrue(
250234
all(vuln_id in response.data["results"]["vulnerabilities"] for vuln_id in package_vulns)
251235
)
236+
def test_filter_packages_by_vulnerability_status(self):
237+
vulnerability = Vulnerability.objects.create(
238+
vulnerability_id="VCID-FILTER", summary="Test vulnerability for is_vulnerable filter"
239+
)
240+
self.package1.affected_by_vulnerabilities.add(vulnerability)
241+
url = reverse("package-v2-list")
242+
with self.assertNumQueries(20):
243+
response = self.client.get(url, {"is_vulnerable": "true"}, format="json")
244+
self.assertEqual(response.status_code, status.HTTP_200_OK)
245+
self.assertEqual(len(response.data["results"]["packages"]), 1)
246+
self.assertEqual(response.data["results"]["packages"][0]["purl"], "pkg:pypi/[email protected]")
247+
with self.assertNumQueries(20):
248+
response = self.client.get(url, {"is_vulnerable": "false"}, format="json")
249+
self.assertEqual(response.status_code, status.HTTP_200_OK)
250+
self.assertEqual(len(response.data["results"]["packages"]), 1)
251+
self.assertEqual(response.data["results"]["packages"][0]["purl"], "pkg:npm/[email protected]")
252252

253253
def test_filter_packages_by_purl(self):
254254
"""

0 commit comments

Comments
 (0)