Add num query tests for codefix viewset

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

Author: TG1999

CHANGELOG.rst

@@ -4,11 +4,15 @@ Release notes
 Version v37.0.0
 ---------------------
 
-- This is a major version, this version introduces Advisory level details.
+- This is a major version, this version introduces Advisory level details
+  https://github.com/aboutcode-org/vulnerablecode/issues/1796
+  https://github.com/aboutcode-org/vulnerablecode/issues/1393
 - We have added new models AdvisoryV2, AdvisoryAlias, AdvisoryReference, AdvisorySeverity, AdvisoryWeakness, PackageV2 and CodeFixV2.
 - We are using ``avid`` as an internal advisory ID for uniquely identifying advisories.
 - We have a new route ``/v2`` which only support package search which has information on packages that are reported to be affected or fixing by advisories.
 - This version introduces ``/api/v2/advisories-packages`` which has information on packages that are reported to be affected or fixing by advisories.
+- Pipeline Dashboard improvements #1920.
+- Throttle API requests based on user permissions #1909.
 
 Version v36.1.3
 ---------------------

vulnerabilities/tests/test_api_v2.py

@@ -859,7 +859,7 @@ def setUp(self):
 
     def test_list_with_purl_filter(self):
         url = reverse("advisories-package-v2-list")
-        with self.assertNumQueries(16):
+        with self.assertNumQueries(18):
             response = self.client.get(url, {"purl": "pkg:pypi/sample@1.0.0"})
         assert response.status_code == 200
         assert "packages" in response.data["results"]
@@ -868,7 +868,7 @@ def test_list_with_purl_filter(self):
 
     def test_bulk_lookup(self):
         url = reverse("advisories-package-v2-bulk-lookup")
-        with self.assertNumQueries(11):
+        with self.assertNumQueries(13):
             response = self.client.post(url, {"purls": ["pkg:pypi/sample@1.0.0"]}, format="json")
         assert response.status_code == 200
         assert "packages" in response.data
@@ -878,7 +878,7 @@ def test_bulk_lookup(self):
     def test_bulk_search_plain(self):
         url = reverse("advisories-package-v2-bulk-search")
         payload = {"purls": ["pkg:pypi/sample@1.0.0"], "plain_purl": True, "purl_only": False}
-        with self.assertNumQueries(11):
+        with self.assertNumQueries(13):
             response = self.client.post(url, payload, format="json")
         assert response.status_code == 200
         assert "packages" in response.data
@@ -887,21 +887,21 @@ def test_bulk_search_plain(self):
     def test_bulk_search_purl_only(self):
         url = reverse("advisories-package-v2-bulk-search")
         payload = {"purls": ["pkg:pypi/sample@1.0.0"], "plain_purl": False, "purl_only": True}
-        with self.assertNumQueries(11):
+        with self.assertNumQueries(13):
             response = self.client.post(url, payload, format="json")
         assert response.status_code == 200
         assert "pkg:pypi/sample@1.0.0" in response.data
 
     def test_lookup_single_package(self):
         url = reverse("advisories-package-v2-lookup")
-        with self.assertNumQueries(9):
+        with self.assertNumQueries(11):
             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)
 
     def test_get_all_vulnerable_purls(self):
         url = reverse("advisories-package-v2-all")
-        with self.assertNumQueries(4):
+        with self.assertNumQueries(6):
             response = self.client.get(url)
         assert response.status_code == 200
         assert "pkg:pypi/sample@1.0.0" in response.data