1414TEST_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" )
1818class 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