diff --git a/worker-{{cookiecutter.slug}}/tests/conftest.py b/worker-{{cookiecutter.slug}}/tests/conftest.py index 8972c1b81b279bab3ccf73a218ff97fae080faa1..bca0563ef4bc7f367068e754ae7c3134715eb636 100644 --- a/worker-{{cookiecutter.slug}}/tests/conftest.py +++ b/worker-{{cookiecutter.slug}}/tests/conftest.py @@ -2,10 +2,13 @@ import os import pytest +from arkindex_worker.worker.base import BaseWorker + +from arkindex.mock import MockApiClient @pytest.fixture(autouse=True) -def setup_environment(responses): +def setup_environment(responses, monkeypatch): """Setup needed environment variables""" # Allow accessing remote API schemas @@ -20,3 +23,6 @@ def setup_environment(responses): os.environ["ARKINDEX_API_SCHEMA_URL"] = schema_url # Setup a fake worker run ID os.environ["ARKINDEX_WORKER_RUN_ID"] = "1234-{{ cookiecutter.slug }}" + + # Setup a mock api client instead of using a real one + monkeypatch.setattr(BaseWorker, "setup_api_client", lambda _: MockApiClient())