Test v2 ubuntu importer pipeline

Signed-off-by: Keshav Priyadarshi <git@keshav.space>

Author: keshav-space

vulnerabilities/tests/pipelines/v2_importers/test_ubuntu_osv_importer.py

@@ -0,0 +1,43 @@
+#
+# Copyright (c) nexB Inc. and others. All rights reserved.
+# VulnerableCode is a trademark of nexB Inc.
+# SPDX-License-Identifier: Apache-2.0
+# See http://www.apache.org/licenses/LICENSE-2.0 for the license text.
+# See https://github.com/aboutcode-org/vulnerablecode for support or download.
+# See https://aboutcode.org for more information about nexB OSS projects.
+#
+
+
+from pathlib import Path
+from unittest.mock import patch
+
+from django.test import TestCase
+
+from vulnerabilities.models import AdvisoryV2
+from vulnerabilities.pipelines.v2_importers.ubuntu_osv_importer import UbuntuOSVImporterPipeline
+from vulnerabilities.tests import util_tests
+from vulnerabilities.tests.pipelines import TestLogger
+
+TEST_DATA = Path(__file__).parent.parent.parent / "test_data" / "ubuntu"
+
+
+class TestUbuntuOSVImporterPipeline(TestCase):
+    def setUp(self):
+        self.logger = TestLogger()
+
+    @patch(
+        "vulnerabilities.pipelines.v2_importers.ubuntu_osv_importer.UbuntuOSVImporterPipeline.clone"
+    )
+    def test_redhat_advisories_v2(self, mock_clone):
+        mock_clone.__name__ = "clone"
+        pipeline = UbuntuOSVImporterPipeline()
+        pipeline.advisories_path = TEST_DATA / "ubuntu_security_notices"
+        pipeline.vcs_response = None
+        pipeline.log = self.logger.write
+        pipeline.execute()
+
+        self.assertEqual(AdvisoryV2.objects.count(), 22)
+
+        expected_file = TEST_DATA / "ubuntu_osv_advisoryv2-expected.json"
+        result = [adv.to_advisory_data().to_dict() for adv in AdvisoryV2.objects.all()]
+        util_tests.check_results_against_json(result, expected_file, regen=True)