Adjust tests

Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>

Author: TG1999

vulnerabilities/api_v2.py

@@ -1059,13 +1059,13 @@ def list(self, request, *args, **kwargs):
             return self.get_paginated_response({"advisories": advisory_data, "packages": data})
 
         # If pagination is not applied, collect vulnerabilities for all packages
-        for package in filtered_queryset:
+        for package in queryset:
             advisories.update({impact.advisory for impact in package.affected_in_impacts.all()})
             advisories.update({impact.advisory for impact in package.fixed_in_impacts.all()})
 
         advisory_data = {f"{adv.avid}": AdvisoryV2Serializer(adv).data for adv in advisories}
 
-        serializer = self.get_serializer(filtered_queryset, many=True)
+        serializer = self.get_serializer(queryset, many=True)
         data = serializer.data
         return Response({"advisories": advisory_data, "packages": data})
 

vulnerabilities/tests/test_api_v2.py

@@ -892,7 +892,7 @@ def test_bulk_search_purl_only(self):
 
     def test_lookup_single_package(self):
         url = reverse("advisories-package-v2-lookup")
-        with self.assertNumQueries(16):
+        with self.assertNumQueries(12):
             response = self.client.post(url, {"purl": "pkg:pypi/sample@1.0.0"}, format="json")
         assert response.status_code == 200
         assert any(pkg["purl"] == "pkg:pypi/sample@1.0.0" for pkg in response.data)