Skip to content

Commit e4849ff

Browse files
authored
Merge pull request #1158 from TG1999/fix_project_kbmsr2019
Fix kbmsr2019 importer
2 parents 3aff154 + 9c53921 commit e4849ff

File tree

2 files changed

+9
-0
lines changed

2 files changed

+9
-0
lines changed

vulnerabilities/importers/project_kb_msr2019.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from vulnerabilities.importer import AdvisoryData
1111
from vulnerabilities.importer import Importer
1212
from vulnerabilities.importer import Reference
13+
from vulnerabilities.utils import fetch_and_read_from_csv
1314
from vulnerabilities.utils import is_cve
1415

1516
# Reading CSV file from a url using `requests` is bit too complicated.

vulnerabilities/tests/test_msr2019.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99

1010
import csv
1111
import os
12+
from unittest import mock
1213

1314
from vulnerabilities.importers.project_kb_msr2019 import ProjectKBMSRImporter
1415
from vulnerabilities.tests import util_tests
@@ -17,6 +18,13 @@
1718
TEST_DIR = os.path.join(BASE_DIR, "test_data/kbmsr2019")
1819

1920

21+
@mock.patch("vulnerabilities.importers.project_kb_msr2019.fetch_and_read_from_csv")
22+
def test_data_fetch(mock_value):
23+
importer = ProjectKBMSRImporter()
24+
mock_value.return_value = []
25+
importer.advisory_data()
26+
27+
2028
def test_kbmsr_to_advisories():
2129
TEST_DATA = os.path.join(TEST_DIR, "test_msr_data.csv")
2230
with open(TEST_DATA) as f:

0 commit comments

Comments
 (0)