Changeset View
Changeset View
Standalone View
Standalone View
swh/deposit/tests/conftest.py
| Show First 20 Lines • Show All 72 Lines • ▼ Show 20 Lines | |||||
| def deposit_config(): | def deposit_config(): | ||||
| return TEST_CONFIG | return TEST_CONFIG | ||||
| @pytest.fixture(autouse=True) | @pytest.fixture(autouse=True) | ||||
| def deposit_autoconfig(monkeypatch, deposit_config, swh_scheduler_config): | def deposit_autoconfig(monkeypatch, deposit_config, swh_scheduler_config): | ||||
| """Enforce config for deposit classes inherited from APIConfig.""" | """Enforce config for deposit classes inherited from APIConfig.""" | ||||
| def mock_parse_config(*args, **kw): | # add the scheduler to configuration | ||||
| config = deposit_config.copy() | config = deposit_config.copy() | ||||
| config["scheduler"] = { | config["scheduler"] = { | ||||
| "cls": "local", | "cls": "local", | ||||
| "args": swh_scheduler_config, | "args": swh_scheduler_config, | ||||
| } | } | ||||
| return config | |||||
| monkeypatch.setattr(APIConfig, "parse_config_file", mock_parse_config) | |||||
| scheduler = get_scheduler("local", swh_scheduler_config) | scheduler = get_scheduler("local", swh_scheduler_config) | ||||
| task_type = { | task_type = { | ||||
| "type": "load-deposit", | "type": "load-deposit", | ||||
| "backend_name": "swh.loader.packages.deposit.tasks.LoadDeposit", | "backend_name": "swh.loader.packages.deposit.tasks.LoadDeposit", | ||||
| "description": "why does this have not-null constraint?", | "description": "why does this have not-null constraint?", | ||||
| } | } | ||||
| scheduler.create_task_type(task_type) | scheduler.create_task_type(task_type) | ||||
| def init_api_config(self): | |||||
| self.config = config | |||||
| self.scheduler = scheduler | |||||
| monkeypatch.setattr(APIConfig, "__init__", init_api_config) | |||||
| @pytest.fixture(scope="session") | @pytest.fixture(scope="session") | ||||
| def django_db_setup(request, django_db_blocker, postgresql_proc): | def django_db_setup(request, django_db_blocker, postgresql_proc): | ||||
| from django.conf import settings | from django.conf import settings | ||||
| settings.DATABASES["default"].update( | settings.DATABASES["default"].update( | ||||
| { | { | ||||
| ("ENGINE", "django.db.backends.postgresql"), | ("ENGINE", "django.db.backends.postgresql"), | ||||
| ▲ Show 20 Lines • Show All 321 Lines • Show Last 20 Lines | |||||