Skip to content

Commit e6d652c

Browse files
committed
Rename DataSource -> Importer
Signed-off-by: Hritik Vijay <hritikxx8@gmail.com>
1 parent 62a2789 commit e6d652c

2 files changed

Lines changed: 12 additions & 12 deletions

File tree

vulnerabilities/importers/mozilla.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -7,21 +7,21 @@
77
from markdown import markdown
88
from packageurl import PackageURL
99

10-
from vulnerabilities.data_source import Advisory
11-
from vulnerabilities.data_source import GitDataSource
12-
from vulnerabilities.data_source import Reference
13-
from vulnerabilities.data_source import VulnerabilitySeverity
10+
from vulnerabilities.importer import Advisory
11+
from vulnerabilities.importer import GitImporter
12+
from vulnerabilities.importer import Reference
13+
from vulnerabilities.importer import VulnerabilitySeverity
1414
from vulnerabilities.helpers import is_cve
1515
from vulnerabilities.helpers import split_markdown_front_matter
16-
from vulnerabilities.severity_systems import scoring_systems
16+
from vulnerabilities.severity_systems import SCORING_SYSTEMS
1717

1818
REPOSITORY = "mozilla/foundation-security-advisories"
1919
MFSA_FILENAME_RE = re.compile(r"mfsa(\d{4}-\d{2,3})\.(md|yml)$")
2020

2121

22-
class MozillaDataSource(GitDataSource):
22+
class MozillaImporter(GitImporter):
2323
def __enter__(self):
24-
super(MozillaDataSource, self).__enter__()
24+
super(MozillaImporter, self).__enter__()
2525

2626
if not getattr(self, "_added_files", None):
2727
self._added_files, self._updated_files = self.file_changes(

vulnerabilities/tests/test_mozilla.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
TEST_DATA = os.path.join(BASE_DIR, "test_data/")
1515

1616

17-
@patch("vulnerabilities.importers.MozillaDataSource._update_from_remote")
17+
@patch("vulnerabilities.importers.MozillaImporter._update_from_remote")
1818
class MozillaImportTest(TestCase):
1919

2020
tempdir = None
@@ -31,7 +31,7 @@ def setUpClass(cls) -> None:
3131
name="mozilla_unittests",
3232
license="",
3333
last_run=None,
34-
data_source="MozillaDataSource",
34+
data_source="MozillaImporter",
3535
data_source_cfg={
3636
"repository_url": "https://example.git",
3737
"working_directory": os.path.join(cls.tempdir, "mozilla_test"),
@@ -48,9 +48,9 @@ def tearDownClass(cls) -> None:
4848
def test_import(self, _):
4949
runner = ImportRunner(self.importer, 100)
5050

51-
# Remove if we don't need set_api in MozillaDataSource
52-
# with patch("vulnerabilities.importers.MozillaDataSource.versions", new=MOCK_VERSION_API):
53-
# with patch("vulnerabilities.importers.MozillaDataSource.set_api"):
51+
# Remove if we don't need set_api in MozillaImporter
52+
# with patch("vulnerabilities.importers.MozillaImporter.versions", new=MOCK_VERSION_API):
53+
# with patch("vulnerabilities.importers.MozillaImporter.set_api"):
5454
# runner.run()
5555
runner.run()
5656

0 commit comments

Comments
 (0)