From b043782248dc7f9dc94379451f359fd586f04aa1 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Fri, 7 Oct 2022 09:08:36 +0000
Subject: [PATCH] Speedup test by not loading a real client during tests
---
worker-{{cookiecutter.slug}}/tests/conftest.py | 8 +++++++-
1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/worker-{{cookiecutter.slug}}/tests/conftest.py b/worker-{{cookiecutter.slug}}/tests/conftest.py
index 8972c1b8..bca0563e 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())
--
GitLab