wip5

Sema4AI · Dec 12, 2024 · a034385 · a034385
1 parent 8b220df
commit a034385
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/.github/workflows/tests-sema4ai-sdk-data-server-integration.yml b/.github/workflows/tests-sema4ai-sdk-data-server-integration.yml
@@ -68,6 +68,7 @@ jobs:
         # Big timeout to create environment in windows.
         RUN_TESTS_TIMEOUT: 3000
         ACTION_SERVER_TEST_ACCESS_CREDENTIALS: ${{ secrets.ACTION_SERVER_TEST_ACCESS_CREDENTIALS }}
+        PYTEST_CAN_RUN_DATA_SERVER: 1
 
       run: poetry run python -u ../../sema4ai-python-ls-core/tests/run_tests.py -rfE -n 1 -otests_output_data_server -vv -m data_server .
 

diff --git a/sema4ai/tests/sema4ai_code_tests/data_server_fixtures.py b/sema4ai/tests/sema4ai_code_tests/data_server_fixtures.py
@@ -169,6 +169,15 @@ def _is_current_db_data_valid(
 
 @pytest.fixture(scope="session")
 def data_server_cli(request, tmpdir_factory) -> Iterator["DataServerCliWrapper"]:
+    import os
+
+    if os.getenv("GITHUB_ACTIONS"):
+        # Detect if running in github actions
+        if not os.getenv("PYTEST_CAN_RUN_DATA_SERVER"):
+            raise RuntimeError(
+                "Test must be marked with @pytest.mark.data_server to run in github actions to use `data_server_cli` fixture"
+            )
+
     from sema4ai_code_tests.data_server_cli_wrapper import DataServerCliWrapper
     from sema4ai_ls_core.system_mutex import timed_acquire_mutex