Changeset View
Changeset View
Standalone View
Standalone View
swh/loader/core/tests/test_loader.py
# Copyright (C) 2018-2020 The Software Heritage developers | # Copyright (C) 2018-2020 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 | ||||
import datetime | import datetime | ||||
import hashlib | import hashlib | ||||
import logging | import logging | ||||
from swh.loader.core.loader import BaseLoader, DVCSLoader | from swh.loader.core.loader import DEFAULT_CONFIG, BaseLoader, DVCSLoader | ||||
from swh.loader.tests import assert_last_visit_matches | from swh.loader.tests import assert_last_visit_matches | ||||
from swh.model.model import Origin, OriginVisit, Snapshot | from swh.model.model import Origin, OriginVisit, Snapshot | ||||
ORIGIN = Origin(url="some-url") | ORIGIN = Origin(url="some-url") | ||||
class DummyLoader: | class DummyLoader: | ||||
"""Base Loader to overload and simplify the base class (technical: to avoid repetition | """Base Loader to overload and simplify the base class (technical: to avoid repetition | ||||
▲ Show 20 Lines • Show All 57 Lines • ▼ Show 20 Lines | |||||
def test_base_loader(swh_config): | def test_base_loader(swh_config): | ||||
loader = DummyBaseLoader() | loader = DummyBaseLoader() | ||||
result = loader.load() | result = loader.load() | ||||
assert result == {"status": "eventful"} | assert result == {"status": "eventful"} | ||||
def test_base_loader_with_config(swh_config): | |||||
loader = DummyBaseLoader("logger-name", DEFAULT_CONFIG) | |||||
result = loader.load() | |||||
assert result == {"status": "eventful"} | |||||
def test_dvcs_loader(swh_config): | def test_dvcs_loader(swh_config): | ||||
loader = DummyDVCSLoader() | loader = DummyDVCSLoader() | ||||
result = loader.load() | result = loader.load() | ||||
assert result == {"status": "eventful"} | assert result == {"status": "eventful"} | ||||
def test_loader_logger_default_name(swh_config): | def test_loader_logger_default_name(swh_config): | ||||
loader = DummyBaseLoader() | loader = DummyBaseLoader() | ||||
▲ Show 20 Lines • Show All 101 Lines • Show Last 20 Lines |