@@ -92,7 +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-
9695 def test_filter_vulnerability_by_vulnerability_id (self ):
9796 """
9897 Test filtering vulnerabilities by vulnerability_id.
@@ -233,22 +232,27 @@ def test_list_packages(self):
233232 self .assertTrue (
234233 all (vuln_id in response .data ["results" ]["vulnerabilities" ] for vuln_id in package_vulns )
235234 )
235+
236236 def test_filter_packages_by_vulnerability_status (self ):
237237 vulnerability = Vulnerability .objects .create (
238238 vulnerability_id = "VCID-FILTER" , summary = "Test vulnerability for is_vulnerable filter"
239239 )
240240 self .package1 .affected_by_vulnerabilities .add (vulnerability )
241241 url = reverse ("package-v2-list" )
242- with self .assertNumQueries (20 ):
243- response = self .client .get (url , {"is_vulnerable" : "true" }, format = "json" )
242+ response = self .client .get (url , {"is_vulnerable" : "true" }, format = "json" )
244243 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/django@3.2" )
247- with self .assertNumQueries (20 ):
248- response = self .client .get (url , {"is_vulnerable" : "false" }, format = "json" )
244+ self .assertIn ("results" , response .data )
245+ self .assertIn ("packages" , response .data ["results" ])
246+ package_purls = [pkg ["purl" ] for pkg in response .data ["results" ]["packages" ]]
247+ self .assertIn (self .package1 .package_url , package_purls )
248+ self .assertNotIn (self .package2 .package_url , package_purls )
249+ response = self .client .get (url , {"is_vulnerable" : "false" }, format = "json" )
249250 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/lodash@4.17.20" )
251+ self .assertIn ("results" , response .data )
252+ self .assertIn ("packages" , response .data ["results" ])
253+ package_purls = [pkg ["purl" ] for pkg in response .data ["results" ]["packages" ]]
254+ self .assertNotIn (self .package1 .package_url , package_purls )
255+ self .assertIn (self .package2 .package_url , package_purls )
252256
253257 def test_filter_packages_by_purl (self ):
254258 """
0 commit comments