diff --git a/swh/loader/git/from_disk.py b/swh/loader/git/from_disk.py --- a/swh/loader/git/from_disk.py +++ b/swh/loader/git/from_disk.py @@ -8,7 +8,7 @@ import datetime import os import shutil -from typing import Dict, Optional +from typing import Any, Dict, Optional from dulwich.errors import ObjectFormatException @@ -35,8 +35,14 @@ visit_type = "git" - def __init__(self, url, visit_date=None, directory=None): - super().__init__(logging_class="swh.loader.git.Loader") + def __init__( + self, + url, + visit_date=None, + directory=None, + config: Optional[Dict[str, Any]] = None, + ): + super().__init__(logging_class="swh.loader.git.Loader", config=config) self.origin_url = url self.visit_date = visit_date self.directory = directory diff --git a/swh/loader/git/loader.py b/swh/loader/git/loader.py --- a/swh/loader/git/loader.py +++ b/swh/loader/git/loader.py @@ -112,6 +112,7 @@ base_url: Optional[str] = None, ignore_history: bool = False, repo_representation: Type[RepoRepresentation] = RepoRepresentation, + config: Optional[Dict[str, Any]] = None, ): """Initialize the bulk updater. @@ -121,7 +122,7 @@ data. """ - super().__init__(logging_class="swh.loader.git.BulkLoader") + super().__init__(logging_class="swh.loader.git.BulkLoader", config=config) self.config = merge_configs(DEFAULT_CONFIG, self.config) self.origin_url = url self.base_url = base_url diff --git a/swh/loader/git/tests/test_loader.py b/swh/loader/git/tests/test_loader.py --- a/swh/loader/git/tests/test_loader.py +++ b/swh/loader/git/tests/test_loader.py @@ -41,7 +41,7 @@ """ @pytest.fixture(autouse=True) - def init(self, swh_config, datadir, tmp_path): + def init(self, swh_loader_config, datadir, tmp_path): super().setUp() archive_name = "testrepo" archive_path = os.path.join(datadir, f"{archive_name}.tgz") @@ -51,5 +51,7 @@ ) self.destination_path = os.path.join(tmp_path, archive_name) base_url = f"base://{self.repo_url}" - self.loader = GitLoader(self.repo_url, base_url=base_url) + self.loader = GitLoader( + self.repo_url, base_url=base_url, config=swh_loader_config + ) self.repo = dulwich.repo.Repo(self.destination_path)