Add tests for API and views

Signed-off-by: Tushar Goel <[email protected]>

Author: TG1999

vulnerabilities/tests/test_fix_api.py

@@ -281,6 +281,13 @@ def test_api_with_all_vulnerable_packages(self):
                 "pkg:generic/nginx/test@9",
             ]
 
+    def test_api_with_ignorning_qualifiers(self):
+        response = self.csrf_client.get(
+            f"/api/packages/?purl=pkg:generic/nginx/test@9?foo=bar", format="json"
+        ).data
+        assert response["count"] == 1
+        assert response["results"][0]["purl"] == "pkg:generic/nginx/test@9"
+
 
 class CPEApi(TestCase):
     def setUp(self):
@@ -353,7 +360,7 @@ def setUp(self):
             attrs = {k: v for k, v in purl.to_dict().items() if v}
             Package.objects.create(**attrs)
 
-    def test_api_response(self):
+    def test_bulk_api_response(self):
         request_body = {
             "purls": self.packages,
         }
@@ -364,6 +371,30 @@ def test_api_response(self):
         ).json()
         assert len(response) == 13
 
+    def test_bulk_api_response_with_ignoring_qualifiers(self):
+        request_body = {
+            "purls": ["pkg:nginx/[email protected]?qualifiers=dev"],
+        }
+        response = self.csrf_client.post(
+            "/api/packages/bulk_search",
+            data=json.dumps(request_body),
+            content_type="application/json",
+        ).json()
+        assert len(response) == 1
+        assert response[0]["purl"] == "pkg:nginx/[email protected]"
+
+    def test_bulk_api_response_with_ignoring_subpath(self):
+        request_body = {
+            "purls": ["pkg:nginx/[email protected]#dev/subpath"],
+        }
+        response = self.csrf_client.post(
+            "/api/packages/bulk_search",
+            data=json.dumps(request_body),
+            content_type="application/json",
+        ).json()
+        assert len(response) == 1
+        assert response[0]["purl"] == "pkg:nginx/[email protected]"
+
 
 class BulkSearchAPICPE(TestCase):
     def setUp(self):

vulnerabilities/tests/test_view.py

@@ -9,14 +9,37 @@
 
 from django.test import Client
 from django.test import TestCase
+from packageurl import PackageURL
 
 from vulnerabilities.models import Alias
+from vulnerabilities.models import Package
 from vulnerabilities.models import Vulnerability
+from vulnerabilities.views import PackageSearch
 
 
 class PackageSearchTestCase(TestCase):
     def setUp(self):
         self.client = Client()
+        packages = [
+            "pkg:nginx/[email protected]",
+            "pkg:nginx/[email protected]",
+            "pkg:nginx/[email protected]",
+            "pkg:nginx/[email protected]",
+            "pkg:nginx/[email protected]",
+            "pkg:nginx/[email protected]",
+            "pkg:nginx/[email protected]",
+            "pkg:nginx/[email protected]",
+            "pkg:nginx/[email protected]",
+            "pkg:nginx/[email protected]",
+            "pkg:nginx/[email protected]",
+            "pkg:nginx/[email protected]",
+            "pkg:nginx/[email protected]",
+        ]
+        self.packages = packages
+        for package in packages:
+            purl = PackageURL.from_string(package)
+            attrs = {k: v for k, v in purl.to_dict().items() if v}
+            Package.objects.create(**attrs)
 
     def test_packages_search_view_paginator(self):
         response = self.client.get("/packages/search?type=deb&name=&page=1")
@@ -28,6 +51,12 @@ def test_packages_search_view_paginator(self):
         response = self.client.get("/packages/search?type=&name=&page=")
         self.assertEqual(response.status_code, 200)
 
+    def test_package_view(self):
+        qs = PackageSearch().get_queryset(query="pkg:nginx/[email protected]?foo=bar")
+        pkgs = list(qs)
+        self.assertEqual(len(pkgs), 1)
+        self.assertEqual(pkgs[0].purl, "pkg:nginx/[email protected]")
+
 
 class VulnerabilitySearchTestCase(TestCase):
     def setUp(self):