test_almalinux_importer_pipeline.py

#
# 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/nexB/vulnerablecode for support or download.
# See https://aboutcode.org for more information about nexB OSS projects.
#
import json
import os
from pathlib import Path
from unittest import TestCase

from vulnerabilities.importers.osv import parse_advisory_data
from vulnerabilities.tests import util_tests

TEST_DATA = Path(__file__).parent.parent / "test_data" / "almalinux"


class TestAlmalinuxImporterPipelin(TestCase):
    def test_almalinux_importer1(self):
        with open(os.path.join(TEST_DATA, "almalinux_test_1.json")) as f:
            mock_response = json.load(f)
        expected_file = os.path.join(TEST_DATA, "almalinux_expected_1.json")
        imported_data = parse_advisory_data(
            raw_data=mock_response,
            supported_ecosystems="rpm",
            advisory_url="https://github.com/AlmaLinux/osv-database"
            "/blob/master/advisories/almalinux8/almalinux_test_1.json",
        )
        result = imported_data.to_dict()
        util_tests.check_results_against_json(result, expected_file)

    def test_almalinux_importer2(self):
        with open(os.path.join(TEST_DATA, "almalinux_test_2.json")) as f:
            mock_response = json.load(f)
        expected_file = os.path.join(TEST_DATA, "almalinux_expected_2.json")
        imported_data = parse_advisory_data(
            raw_data=mock_response,
            supported_ecosystems="rpm",
            advisory_url="https://github.com/AlmaLinux/osv-database"
            "/blob/master/advisories/almalinux8/almalinux_test_2.json",
        )
        result = imported_data.to_dict()
        util_tests.check_results_against_json(result, expected_file)

    def test_almalinux_importer3(self):
        with open(os.path.join(TEST_DATA, "almalinux_test_3.json")) as f:
            mock_response = json.load(f)
        expected_file = os.path.join(TEST_DATA, "almalinux_expected_3.json")
        imported_data = parse_advisory_data(
            raw_data=mock_response,
            supported_ecosystems="rpm",
            advisory_url="https://github.com/AlmaLinux/osv-database"
            "/blob/master/advisories/almalinux8/almalinux_test_3.json",
        )
        result = imported_data.to_dict()
        util_tests.check_results_against_json(result, expected_file)