Fix integration tests

openzim · Dec 6, 2024 · 34b596f · 34b596f
1 parent bc18cd0
commit 34b596f
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 3 deletions.
diff --git a/scraper/tests-integration/conftest.py b/scraper/tests-integration/conftest.py
@@ -1,9 +1,29 @@
 import tempfile
+import threading
 from collections.abc import Generator
 from pathlib import Path
 from typing import Any
 
 import pytest
+from zimscraperlib.download import get_session
+
+from mindtouch2zim.context import Context
+
+CONTEXT_DEFAULTS = {
+    "web_session": get_session(),
+    "tmp_folder": None,
+    "cache_folder": None,
+    "_current_thread_workitem": threading.local(),
+    "library_url": None,
+    "creator": None,
+    "name": None,
+    "title": None,
+    "description": None,
+}
+
+
+# initialize a context since it is a requirement for most modules to load
+Context.setup(**CONTEXT_DEFAULTS)
 
 
 @pytest.fixture(scope="module")

diff --git a/scraper/tests-integration/test_client.py b/scraper/tests-integration/test_client.py
@@ -11,14 +11,16 @@
     MindtouchClient,
     MindtouchHome,
 )
-from mindtouch2zim.context import CONTEXT
+from mindtouch2zim.context import Context
 from mindtouch2zim.download import stream_file
 
+context = Context.get()
+
 
 @pytest.fixture(scope="module")
 def client(libretexts_url: str, cache_folder: Path) -> MindtouchClient:
-    CONTEXT.library_url = libretexts_url
-    CONTEXT.cache_folder = cache_folder
+    context.library_url = libretexts_url
+    context.cache_folder = cache_folder
     return MindtouchClient()