From 9c539212448acb2236c0976a15f53f676c161fc6 Mon Sep 17 00:00:00 2001 From: Tushar Goel Date: Mon, 27 Mar 2023 21:54:18 +0530 Subject: [PATCH] Fix kbmsr2019 importer Signed-off-by: Tushar Goel --- vulnerabilities/importers/project_kb_msr2019.py | 1 + vulnerabilities/tests/test_msr2019.py | 8 ++++++++ 2 files changed, 9 insertions(+) diff --git a/vulnerabilities/importers/project_kb_msr2019.py b/vulnerabilities/importers/project_kb_msr2019.py index 9d281de72..9fcfb7cc6 100644 --- a/vulnerabilities/importers/project_kb_msr2019.py +++ b/vulnerabilities/importers/project_kb_msr2019.py @@ -10,6 +10,7 @@ from vulnerabilities.importer import AdvisoryData from vulnerabilities.importer import Importer from vulnerabilities.importer import Reference +from vulnerabilities.utils import fetch_and_read_from_csv from vulnerabilities.utils import is_cve # Reading CSV file from a url using `requests` is bit too complicated. diff --git a/vulnerabilities/tests/test_msr2019.py b/vulnerabilities/tests/test_msr2019.py index c921de0b2..4af8334e3 100644 --- a/vulnerabilities/tests/test_msr2019.py +++ b/vulnerabilities/tests/test_msr2019.py @@ -9,6 +9,7 @@ import csv import os +from unittest import mock from vulnerabilities.importers.project_kb_msr2019 import ProjectKBMSRImporter from vulnerabilities.tests import util_tests @@ -17,6 +18,13 @@ TEST_DIR = os.path.join(BASE_DIR, "test_data/kbmsr2019") +@mock.patch("vulnerabilities.importers.project_kb_msr2019.fetch_and_read_from_csv") +def test_data_fetch(mock_value): + importer = ProjectKBMSRImporter() + mock_value.return_value = [] + importer.advisory_data() + + def test_kbmsr_to_advisories(): TEST_DATA = os.path.join(TEST_DIR, "test_msr_data.csv") with open(TEST_DATA) as f: