Skip to content

Commit a831e85

Browse files
committed
Move example importer and improver to tests dir
It serves as a test data source and does not contribute any actual data. Signed-off-by: Hritik Vijay <hritikxx8@gmail.com>
1 parent ad3ff5a commit a831e85

2 files changed

Lines changed: 10 additions & 5 deletions

File tree

File renamed without changes.

vulnerabilities/tests/test_example.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,11 @@
1515
from vulnerabilities.importer import Reference
1616
from vulnerabilities.importer import ScoringSystem
1717
from vulnerabilities.importer import VulnerabilitySeverity
18-
from vulnerabilities.importers.example import ExampleAliasImprover
19-
from vulnerabilities.importers.example import ExampleImporter
20-
from vulnerabilities.importers.example import parse_advisory_data
2118
from vulnerabilities.improve_runner import ImproveRunner
2219
from vulnerabilities.improvers.default import DefaultImprover
20+
from vulnerabilities.tests.example_importer_improver import ExampleAliasImprover
21+
from vulnerabilities.tests.example_importer_improver import ExampleImporter
22+
from vulnerabilities.tests.example_importer_improver import parse_advisory_data
2323

2424

2525
def mock_fetch_advisory_data():
@@ -43,8 +43,13 @@ def mock_fetch_additional_aliases(alias):
4343
return alias_map.get(alias)
4444

4545

46-
@patch("vulnerabilities.importers.example.fetch_advisory_data", mock_fetch_advisory_data)
47-
@patch("vulnerabilities.importers.example.fetch_additional_aliases", mock_fetch_additional_aliases)
46+
@patch(
47+
"vulnerabilities.tests.example_importer_improver.fetch_advisory_data", mock_fetch_advisory_data
48+
)
49+
@patch(
50+
"vulnerabilities.tests.example_importer_improver.fetch_additional_aliases",
51+
mock_fetch_additional_aliases,
52+
)
4853
class TestExampleImporter(TestCase):
4954
def test_parse_advisory_data(self):
5055
raw_data = mock_fetch_advisory_data()[0]

0 commit comments

Comments
 (0)