Changeset View
Changeset View
Standalone View
Standalone View
swh/loader/pytest_plugin.py
- This file was copied from conftest.py.
# Copyright (C) 2019-2020 The Software Heritage developers | # Copyright (C) 2019-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 os | import os | ||||
import pytest | import pytest | ||||
import yaml | import yaml | ||||
from typing import Any, Dict | from typing import Any, Dict | ||||
pytest_plugins = ["swh.scheduler.pytest_plugin", "swh.storage.pytest_plugin"] | |||||
@pytest.fixture | @pytest.fixture | ||||
def swh_loader_config(swh_storage_postgresql) -> Dict[str, Any]: | def swh_loader_config(swh_storage_postgresql) -> Dict[str, Any]: | ||||
return { | return { | ||||
"storage": { | "storage": { | ||||
"cls": "pipeline", | "cls": "pipeline", | ||||
"steps": [ | "steps": [ | ||||
{"cls": "retry"}, | {"cls": "retry"}, | ||||
Show All 25 Lines | |||||
@pytest.fixture(autouse=True, scope="session") | @pytest.fixture(autouse=True, scope="session") | ||||
def swh_proxy(): | def swh_proxy(): | ||||
"""Automatically inject this fixture in all tests to ensure no outside | """Automatically inject this fixture in all tests to ensure no outside | ||||
connection takes place. | connection takes place. | ||||
""" | """ | ||||
os.environ["http_proxy"] = "http://localhost:999" | os.environ["http_proxy"] = "http://localhost:999" | ||||
os.environ["https_proxy"] = "http://localhost:999" | os.environ["https_proxy"] = "http://localhost:999" | ||||
@pytest.fixture(scope="session") | |||||
def swh_scheduler_celery_includes(swh_scheduler_celery_includes): | |||||
return swh_scheduler_celery_includes + [ | |||||
"swh.loader.package.archive.tasks", | |||||
"swh.loader.package.cran.tasks", | |||||
"swh.loader.package.debian.tasks", | |||||
"swh.loader.package.deposit.tasks", | |||||
"swh.loader.package.npm.tasks", | |||||
"swh.loader.package.pypi.tasks", | |||||
"swh.loader.package.nixguix.tasks", | |||||
] |