Fix errors

Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>

Author: TG1999

vulnerabilities/pipelines/exporters/federate_vulnerabilities.py

@@ -14,11 +14,11 @@
 from operator import attrgetter
 from pathlib import Path
 
+from aboutcode.federated import DataFederation
 from aboutcode.pipeline import LoopProgress
 from django.conf import settings
 from django.utils import timezone
 
-from aboutcode.federated import DataFederation
 from vulnerabilities.pipelines import VulnerableCodePipeline
 from vulnerabilities.pipes import export
 from vulnerabilities.pipes import federatedcode

vulnerabilities/pipelines/v2_improvers/reference_collect_commits.py

@@ -8,12 +8,12 @@
 #
 from collections import defaultdict
 
+from aboutcode.federated import get_core_purl
 from aboutcode.pipeline import LoopProgress
 from django.db.models import Prefetch
 from packageurl.contrib.purl2url import purl2url
 from packageurl.contrib.url2purl import url2purl
 
-from aboutcode.federated import get_core_purl
 from vulnerabilities.models import AdvisoryReference
 from vulnerabilities.models import AdvisoryV2
 from vulnerabilities.models import ImpactedPackage

vulnerabilities/tests/pipelines/v2_importers/test_apache_httpd_importer_pipeline_v2.py

@@ -90,6 +90,8 @@ def test_get_weaknesses_with_data_meta_structure():
 
 # Tests for ApacheHTTPDImporterPipeline
 class DummyPipeline(ApacheHTTPDImporterPipeline):
+    datasource_id = "dummy"
+    pipeline_id = "dummy_pipeline"
     # Expose protected methods for testing
     pass
 

vulnerabilities/tests/test_tasks.py

@@ -20,6 +20,7 @@
 
 class OneTimePipeline(VulnerableCodeBaseImporterPipelineV2):
     pipeline_id = "one_time_pipeline_test"
+    datasource_id = "one_time_pipeline_test"
     run_once = True
 
     def collect_advisories(self):
@@ -31,6 +32,7 @@ def advisories_count(self):
 
 class NotOneTimePipeline(VulnerableCodeBaseImporterPipelineV2):
     pipeline_id = "not_one_time_pipeline_test"
+    datasource_id = "not_one_time_pipeline_test"
     run_once = False
 
     def collect_advisories(self):