vulnerablecode/vulnerabilities/tests/test_almalinux.py at e82f47ee26c3b6fe9b2491c44f997a8719b6cffc · aboutcode-org/vulnerablecode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
#
# 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 unittest import TestCase

from vulnerabilities.importers.almalinux import get_affected_purl
from vulnerabilities.importers.almalinux import get_fixed_versions
from vulnerabilities.importers.almalinux import parse_advisory_data
from vulnerabilities.tests import util_tests

BASE_DIR = os.path.dirname(os.path.abspath(__file__))
TEST_DATA = os.path.join(BASE_DIR, "test_data/almalinux")


class AlmaImporter(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(
            mock_response,
            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(
            mock_response,
            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_github_osv_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(
            mock_response,
            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)