Add fixed packages in packages endpoint

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

Author: TG1999

vulnerabilities/api.py

@@ -106,6 +106,7 @@ def to_representation(self, instance):
     fixing_vulnerabilities = MinimalVulnerabilitySerializer(
         many=True, source="resolved_to", read_only=True
     )
+    fixed_packages = MinimalPackageSerializer(many=True, read_only=True)
 
     class Meta:
         model = Package
@@ -119,6 +120,7 @@ class Meta:
             "qualifiers",
             "subpath",
             "affected_by_vulnerabilities",
+            "fixed_packages",
             "fixing_vulnerabilities",
         ]
 

vulnerabilities/models.py

@@ -193,6 +193,23 @@ def resolved_to(self):
         """
         return self.vulnerabilities.filter(packagerelatedvulnerability__fix=True)
 
+    @property
+    def fixed_packages(self):
+        """
+        Returns vulnerabilities which are affecting this package.
+        """
+        return (
+            Package.objects.filter(
+                name=self.name,
+                namespace=self.namespace,
+                type=self.type,
+                qualifiers=self.qualifiers,
+                subpath=self.subpath,
+            )
+            .filter(packagerelatedvulnerability__fix=True)
+            .distinct()
+        )
+
     def set_package_url(self, package_url):
         """
         Set each field values to the values of the provided `package_url` string

vulnerabilities/tests/test_fix_api.py

@@ -91,6 +91,7 @@ def test_api_with_single_vulnerability(self):
             "unresolved_vulnerabilities": [],
             "qualifiers": {},
             "subpath": "",
+            "fixed_packages": [],
             "affected_by_vulnerabilities": [],
             "fixing_vulnerabilities": [],
         }