Add pipeline_id to nvd pipeline

Signed-off-by: Keshav Priyadarshi <[email protected]>

Author: keshav-space

vulnerabilities/improvers/vulnerability_status.py

@@ -38,7 +38,7 @@ class VulnerabilityStatusImprover(Improver):
     @property
     def interesting_advisories(self) -> QuerySet:
         return (
-            Advisory.objects.filter(Q(created_by=NVDImporterPipeline.qualified_name))
+            Advisory.objects.filter(Q(created_by=NVDImporterPipeline.pipeline_id))
             .distinct("aliases")
             .paginated()
         )

vulnerabilities/pipelines/nvd_importer.py

@@ -30,6 +30,8 @@
 class NVDImporterPipeline(VulnerableCodeBaseImporterPipeline):
     """Collect advisories from NVD."""
 
+    pipeline_id = "nvd_importer"
+
     # See https://github.com/nexB/vulnerablecode/issues/665 for follow up
     spdx_license_expression = (
         "LicenseRef-scancode-us-govt-public-domain  AND LicenseRef-scancode-cve-tou"

vulnerabilities/tests/test_vulnerability_status_improver.py

@@ -33,13 +33,13 @@
 def test_interesting_advisories():
     Advisory.objects.create(
         aliases=["CVE-1"],
-        created_by=NVDImporterPipeline.qualified_name,
+        created_by=NVDImporterPipeline.pipeline_id,
         summary="1",
         date_collected=datetime.now(),
     )
     Advisory.objects.create(
         aliases=["CVE-1"],
-        created_by=NVDImporterPipeline.qualified_name,
+        created_by=NVDImporterPipeline.pipeline_id,
         summary="2",
         date_collected=datetime.now(),
     )
@@ -54,7 +54,7 @@ def test_improver_end_to_end(mock_response):
     mock_response.return_value = response
     adv = Advisory.objects.create(
         aliases=["CVE-2023-35866"],
-        created_by=NVDImporterPipeline.qualified_name,
+        created_by=NVDImporterPipeline.pipeline_id,
         summary="1",
         date_collected=datetime.now(),
     )