Changeset View
Changeset View
Standalone View
Standalone View
swh/loader/git/tests/test_loader.py
# Copyright (C) 2018-2019 The Software Heritage developers | # Copyright (C) 2018-2019 The Software Heritage developers | ||||
# See the AUTHORS file at the top-level directory of this distribution | # See the AUTHORS file at the top-level directory of this distribution | ||||
# License: GNU General Public License version 3, or any later version | # License: GNU General Public License version 3, or any later version | ||||
# See top-level LICENSE file for more information | # See top-level LICENSE file for more information | ||||
from swh.loader.git.loader import GitLoader | from swh.loader.git.loader import GitLoader | ||||
from swh.loader.git.tests.test_from_disk import DirGitLoaderTest | from swh.loader.git.tests.test_from_disk import DirGitLoaderTest | ||||
from . import TEST_LOADER_CONFIG | |||||
class GitLoaderTest(GitLoader): | class GitLoaderTest(GitLoader): | ||||
def parse_config_file(self, *args, **kwargs): | def parse_config_file(self, *args, **kwargs): | ||||
return { | return { | ||||
**super().parse_config_file(*args, **kwargs), | **super().parse_config_file(*args, **kwargs), | ||||
'storage': {'cls': 'memory', 'args': {}} | **TEST_LOADER_CONFIG | ||||
} | } | ||||
class TestGitLoader(DirGitLoaderTest): | class TestGitLoader(DirGitLoaderTest): | ||||
"""Same tests as for the GitLoaderFromDisk, but running on GitLoader.""" | """Same tests as for the GitLoaderFromDisk, but running on GitLoader.""" | ||||
def setUp(self): | def setUp(self): | ||||
super().setUp() | super().setUp() | ||||
self.loader = GitLoaderTest(self.repo_url) | self.loader = GitLoaderTest(self.repo_url) | ||||
self.storage = self.loader.storage | self.storage = self.loader.storage | ||||
def load(self): | def load(self): | ||||
return self.loader.load() | return self.loader.load() |