File tree Expand file tree Collapse file tree 2 files changed +9
-0
lines changed Expand file tree Collapse file tree 2 files changed +9
-0
lines changed Original file line number Diff line number Diff line change 1010from vulnerabilities .importer import AdvisoryData
1111from vulnerabilities .importer import Importer
1212from vulnerabilities .importer import Reference
13+ from vulnerabilities .utils import fetch_and_read_from_csv
1314from vulnerabilities .utils import is_cve
1415
1516# Reading CSV file from a url using `requests` is bit too complicated.
Original file line number Diff line number Diff line change 99
1010import csv
1111import os
12+ from unittest import mock
1213
1314from vulnerabilities .importers .project_kb_msr2019 import ProjectKBMSRImporter
1415from vulnerabilities .tests import util_tests
1718TEST_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+
2028def test_kbmsr_to_advisories ():
2129 TEST_DATA = os .path .join (TEST_DIR , "test_msr_data.csv" )
2230 with open (TEST_DATA ) as f :
You can’t perform that action at this time.
0 commit comments