Some general pytest cleanup (#1021)

djeebus · web-flow · commit d2e22e3748de · 2025-11-13T17:58:30.000-08:00
- Individual tests must complete in less than 5 minutes
- Add a `make test` option that runs tests
- Upgrade poetry to 2.1.1 (the lock file was generated by this version,
so this just matches what we already expect)
diff --git a/.github/workflows/python_sdk_tests.yml b/.github/workflows/python_sdk_tests.yml
@@ -50,6 +50,6 @@ jobs:
         run: poetry build
 
       - name: Run tests
-        run: poetry run pytest --verbose -x
+        run: make test
         env:
           E2B_API_KEY: ${{ secrets.E2B_API_KEY }}
diff --git a/.tool-versions b/.tool-versions
@@ -2,4 +2,4 @@ deno 1.46.3
 nodejs 20.19.5
 pnpm 9.15.5
 python 3.9
-poetry 1.8.3
+poetry 2.1.1
diff --git a/packages/python-sdk/Makefile b/packages/python-sdk/Makefile
@@ -39,4 +39,12 @@ generate-mcp:
 			--class-name McpServer \
 			--use-field-description \
 			--disable-timestamp \
-			--extra-fields forbid
+			--extra-fields forbid
+
+.PHONY: setup
+setup:
+	poetry install
+
+.PHONY: test
+test: setup
+	poetry run pytest --verbose --numprocesses=4
diff --git a/packages/python-sdk/poetry.lock b/packages/python-sdk/poetry.lock
diff --git a/packages/python-sdk/pyproject.toml b/packages/python-sdk/pyproject.toml
@@ -31,6 +31,7 @@ pytest-asyncio = "^0.23.7"
 pydoc-markdown = "^4.8.2"
 datamodel-code-generator = "^0.34.0"
 ruff = "^0.11.12"
+pytest-timeout = "^2.4.0"
 
 [build-system]
 requires = ["poetry-core"]
diff --git a/packages/python-sdk/pytest.ini b/packages/python-sdk/pytest.ini
@@ -4,4 +4,5 @@ markers =
     skip_debug: skip test if E2B_DEBUG is set.
 
 asyncio_mode=auto
-addopts = "--import-mode=importlib" "--numprocesses=4"
+addopts = "--import-mode=importlib"
+timeout = 300
diff --git a/packages/python-sdk/tests/test_connection_config.py b/packages/python-sdk/tests/test_connection_config.py
@@ -1,7 +1,9 @@
 from e2b import ConnectionConfig
 
 
-def test_api_url_defaults_correctly():
+def test_api_url_defaults_correctly(monkeypatch):
+    monkeypatch.setenv("E2B_DOMAIN", "")
+
     config = ConnectionConfig()
     assert config.api_url == "https://api.e2b.app"
 
