Merge pull request #765 from TG1999/search_from_alias

Search from alias in UI

Author: pombredanne

vulnerabilities/tests/test_view.py

@@ -1,6 +1,7 @@
 from django.test import Client
 from django.test import TestCase
 
+from vulnerabilities.models import Alias
 from vulnerabilities.models import Vulnerability
 
 
@@ -26,6 +27,8 @@ class VulnerabilitySearchTestCase(TestCase):
     def setUp(self):
         vulnerability = Vulnerability(summary="test")
         vulnerability.save()
+        alias = Alias(alias="TEST-2022", vulnerability=vulnerability)
+        alias.save()
         self.id = vulnerability.id
         self.client = Client()
 
@@ -36,9 +39,16 @@ def test_vulnerabilties(self):
         response = self.client.get(f"/vulnerabilities/{self.id}")
         self.assertEqual(response.status_code, 200)
 
-    def test_vulnerabilties(self):
+    def test_vulnerabilties_search(self):
         """
         Test Vulnerability Search View
         """
         response = self.client.get(f"/vulnerabilities/search")
         self.assertEqual(response.status_code, 200)
+
+    def test_alias(self):
+        """
+        Test Vulnerability Search View
+        """
+        response = self.client.get(f"/vulnerabilities/search?vuln_id=TEST-2022")
+        self.assertEqual(response.status_code, 200)

vulnerabilities/views.py

@@ -110,7 +110,9 @@ def get(self, request):
     def request_to_vulnerabilities(request):
         vuln_id = request.GET["vuln_id"]
         return list(
-            models.Vulnerability.objects.filter(vulnerability_id=vuln_id).annotate(
+            models.Vulnerability.objects.filter(
+                Q(vulnerability_id=vuln_id) | Q(aliases__alias__icontains=vuln_id)
+            ).annotate(
                 vulnerable_package_count=Count(
                     "packages", filter=Q(packagerelatedvulnerability__fix=False)
                 ),