File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree 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