Fix tests

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

Author: TG1999

vulnerabilities/tests/pipelines/v2_improvers/test_enhance_with_github_poc.py

@@ -41,6 +41,7 @@ def test_github_poc_db_improver(mock_fetch_via_vcs):
         unique_content_id="sgsdg45",
         url="https://test.com",
         date_collected=datetime.now(),
+        pipeline_id="ghsa_pipeline_v2",
     )
     adv2 = AdvisoryV2.objects.create(
         advisory_id="VCIO-123-1002",
@@ -49,6 +50,7 @@ def test_github_poc_db_improver(mock_fetch_via_vcs):
         unique_content_id="6hd4d6f",
         url="https://test.com",
         date_collected=datetime.now(),
+        pipeline_id="ghsa_pipeline_v2",
     )
     adv3 = AdvisoryV2.objects.create(
         advisory_id="VCIO-123-1003",
@@ -57,6 +59,7 @@ def test_github_poc_db_improver(mock_fetch_via_vcs):
         unique_content_id="sd6h4sh",
         url="https://test.com",
         date_collected=datetime.now(),
+        pipeline_id="ghsa_pipeline_v2",
     )
 
     alias1 = AdvisoryAlias.objects.create(alias="CVE-2022-0236")

vulnerabilities/tests/pipelines/v2_improvers/test_reference_collect_commits_v2.py

@@ -29,6 +29,7 @@ def test_collect_fix_commits_pipeline_creates_entry():
         url="https://example.com/advisory/CVE-2025-1000",
         unique_content_id="11111",
         date_collected=datetime.now(),
+        pipeline_id="test-pipeline",
     )
 
     reference = AdvisoryReference.objects.create(
@@ -60,6 +61,7 @@ def test_collect_fix_commits_pipeline_skips_non_commit_urls():
         url="https://example.com/advisory/CVE-2025-2000",
         unique_content_id="11111",
         date_collected=datetime.now(),
+        pipeline_id="test-pipeline",
     )
 
     reference = AdvisoryReference.objects.create(

vulnerabilities/tests/test_api_v3.py

@@ -32,7 +32,7 @@ def setUp(self):
                 advisory_id="GHSA-1234",
                 url="https://example.com/advisory",
             ),
-            pipeline_id="ghsa",
+            pipeline_id="ghsa_pipeline_v2",
             datasource_id="ghsa",
             logger=self.logger.write,
         )
@@ -95,7 +95,8 @@ def test_advisories_post(self):
         self.assertEqual(response.status_code, status.HTTP_200_OK)
 
         advisory = response.data["results"][0]
-        self.assertEqual(advisory["advisory_id"], "ghsa/GHSA-1234")
+        self.assertEqual(advisory["advisory_id"], "GHSA-1234")
+        self.assertEqual(advisory["advisory_uid"], "ghsa/GHSA-1234")
 
     def test_affected_by_advisories_list(self):
         url = reverse("affected-by-advisories-list")
@@ -110,7 +111,7 @@ def test_affected_by_advisories_list(self):
 
         results = response.data["results"]
         self.assertEqual(len(results), 1)
-        self.assertEqual(results[0]["advisory_id"], "ghsa/GHSA-1234")
+        self.assertEqual(results[0]["advisory_id"], "GHSA-1234")
 
     def test_fixing_advisories_list_empty(self):
         url = reverse("fixing-advisories-list")
@@ -198,7 +199,7 @@ def test_advisories_post(self):
         self.assertEqual(response.status_code, status.HTTP_200_OK)
         self.assertEqual(len(response.data["results"]), 100)
         advisory = response.data["results"][0]
-        self.assertEqual(advisory["advisory_id"], "ghsa/GHSA-12341")
+        self.assertEqual(advisory["advisory_id"], "GHSA-12341")
 
 
 class APIV3TestCaseOneAdvisoryMultiplePackages(APITestCase):