Changeset View
Changeset View
Standalone View
Standalone View
swh/deposit/tests/loader/conftest.py
# Copyright (C) 2019 The Software Heritage developers | # Copyright (C) 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 | ||||
import re | import re | ||||
import os | import os | ||||
import pytest | import pytest | ||||
import yaml | import yaml | ||||
from functools import partial | from functools import partial | ||||
from swh.core.pytest_plugin import get_response_cb | from swh.core.pytest_plugin import get_response_cb | ||||
from swh.scheduler.tests.conftest import * # noqa | from swh.scheduler.tests.conftest import * # noqa | ||||
from swh.storage.tests.conftest import * # noqa | from swh.storage.tests.conftest import * # noqa | ||||
from swh.deposit.loader.checker import DepositChecker | from swh.deposit.loader.checker import DepositChecker | ||||
from swh.deposit.loader.loader import DepositLoader | |||||
@pytest.fixture(scope='session') # type: ignore # expected redefinition | @pytest.fixture(scope='session') # type: ignore # expected redefinition | ||||
def celery_includes(): | def celery_includes(): | ||||
return [ | return [ | ||||
'swh.deposit.loader.tasks', | 'swh.deposit.loader.tasks', | ||||
] | ] | ||||
Show All 22 Lines | |||||
@pytest.fixture | @pytest.fixture | ||||
def deposit_checker(swh_config): | def deposit_checker(swh_config): | ||||
return DepositChecker() | return DepositChecker() | ||||
@pytest.fixture | @pytest.fixture | ||||
def deposit_loader(swh_config): | |||||
return DepositLoader() | |||||
@pytest.fixture | |||||
def requests_mock_datadir(datadir, requests_mock_datadir): | def requests_mock_datadir(datadir, requests_mock_datadir): | ||||
"""Override default behavior to deal with put method | """Override default behavior to deal with put method | ||||
""" | """ | ||||
cb = partial(get_response_cb, datadir=datadir) | cb = partial(get_response_cb, datadir=datadir) | ||||
requests_mock_datadir.put(re.compile('https://'), body=cb) | requests_mock_datadir.put(re.compile('https://'), body=cb) | ||||
return requests_mock_datadir | return requests_mock_datadir |