Set token id + secret in test fixtures w subprocesses

modal-labs · Oct 18, 2024 · 8458296 · 8458296
1 parent 25ce086
commit 8458296
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/test/conftest.py b/test/conftest.py
@@ -1768,6 +1768,8 @@ async def container_client(servicer):
 @pytest_asyncio.fixture(scope="function")
 async def server_url_env(servicer, monkeypatch):
     monkeypatch.setenv("MODAL_SERVER_URL", servicer.client_addr)
+    monkeypatch.setenv("MODAL_TOKEN_ID", "ak-123")
+    monkeypatch.setenv("MODAL_TOKEN_SECRET", "as-123")
     yield
 
 

diff --git a/test/helpers.py b/test/helpers.py
@@ -26,7 +26,14 @@ def deploy_app_externally(
             **{"PYTHONUTF8": "1"},
         }  # windows apparently needs a bunch of env vars to start python...
 
-    env = {**windows_support, "MODAL_SERVER_URL": servicer.client_addr, "MODAL_ENVIRONMENT": "main", **env}
+    env = {
+        **windows_support,
+        "MODAL_SERVER_URL": servicer.client_addr,
+        "MODAL_ENVIRONMENT": "main",
+        "MODAL_TOKEN_ID": "ak-123",
+        "MODAL_TOKEN_SECRET": "as-123",
+        **env,
+    }
     if cwd is None:
         cwd = pathlib.Path(__file__).parent.parent