Add aliases at package endpoint

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

Author: TG1999

vulnerabilities/api.py

@@ -53,22 +53,6 @@ class Meta:
         fields = ["url", "purl", "is_vulnerable"]
 
 
-class VulnSerializerRefsAndSummary(serializers.HyperlinkedModelSerializer):
-    """
-    Lookup vulnerabilities references by aliases (such as a CVE).
-    """
-
-    fixed_packages = MinimalPackageSerializer(
-        many=True, source="filtered_fixed_packages", read_only=True
-    )
-
-    references = VulnerabilityReferenceSerializer(many=True, source="vulnerabilityreference_set")
-
-    class Meta:
-        model = Vulnerability
-        fields = ["url", "vulnerability_id", "summary", "references", "fixed_packages"]
-
-
 class MinimalVulnerabilitySerializer(serializers.HyperlinkedModelSerializer):
     """
     Lookup vulnerabilities by aliases (such as a CVE).
@@ -89,6 +73,23 @@ class Meta:
         fields = ["alias"]
 
 
+class VulnSerializerRefsAndSummary(serializers.HyperlinkedModelSerializer):
+    """
+    Lookup vulnerabilities references by aliases (such as a CVE).
+    """
+
+    fixed_packages = MinimalPackageSerializer(
+        many=True, source="filtered_fixed_packages", read_only=True
+    )
+
+    references = VulnerabilityReferenceSerializer(many=True, source="vulnerabilityreference_set")
+    aliases = AliasSerializer(many=True, source="alias")
+
+    class Meta:
+        model = Vulnerability
+        fields = ["url", "vulnerability_id", "summary", "references", "fixed_packages", "aliases"]
+
+
 class VulnerabilitySerializer(serializers.HyperlinkedModelSerializer):
 
     fixed_packages = MinimalPackageSerializer(

vulnerabilities/tests/test_api.py

@@ -296,6 +296,7 @@ def setUp(self):
         vuln1 = Vulnerability.objects.create(
             summary="test-vuln1",
         )
+        alias = Alias.objects.create(alias="CVE-2019-1234", vulnerability=vuln1)
         self.vuln1 = vuln1
         PackageRelatedVulnerability.objects.create(
             package=self.package,
@@ -340,6 +341,7 @@ def test_api_with_single_vulnerability_and_fixed_package(self):
                     "summary": "test-vuln1",
                     "references": [],
                     "fixed_packages": [],
+                    "aliases": [{"alias": "CVE-2019-1234"}],
                 }
             ],
             "fixing_vulnerabilities": [
@@ -355,6 +357,7 @@ def test_api_with_single_vulnerability_and_fixed_package(self):
                             "is_vulnerable": True,
                         }
                     ],
+                    "aliases": [],
                 },
             ],
             "unresolved_vulnerabilities": [
@@ -364,6 +367,7 @@ def test_api_with_single_vulnerability_and_fixed_package(self):
                     "summary": "test-vuln1",
                     "references": [],
                     "fixed_packages": [],
+                    "aliases": [{"alias": "CVE-2019-1234"}],
                 }
             ],
         }
@@ -392,6 +396,7 @@ def test_api_with_single_vulnerability_and_vulnerable_package(self):
                             "is_vulnerable": True,
                         }
                     ],
+                    "aliases": [],
                 }
             ],
             "fixing_vulnerabilities": [],
@@ -408,6 +413,7 @@ def test_api_with_single_vulnerability_and_vulnerable_package(self):
                             "is_vulnerable": True,
                         }
                     ],
+                    "aliases": [],
                 }
             ],
         }