Refactor sandbox fixtures for better async handling

Co-authored-by: jakub.dobry <jakub.dobry@e2b.dev>

Author: cursoragent

python/tests/conftest.py

@@ -11,26 +11,17 @@
 timeout = 60
 
 
-@pytest_asyncio.fixture(scope="function")
+# Override the event loop so it never closes during test execution
+# This helps with pytest-xdist and prevents "Event loop is closed" errors
+@pytest.fixture(scope="session")
 def event_loop():
-    """Create an instance of the default event loop for each test case."""
-    policy = asyncio.get_event_loop_policy()
-    loop = policy.new_event_loop()
-    yield loop
-    # Clean up any remaining tasks
-    try:
-        pending = asyncio.all_tasks(loop)
-        for task in pending:
-            task.cancel()
-        if pending:
-            loop.run_until_complete(asyncio.gather(*pending, return_exceptions=True))
-    except Exception:
-        pass
-    # Close the loop gracefully
+    """Create a session-scoped event loop for all async tests."""
     try:
-        loop.close()
-    except Exception:
-        pass
+        loop = asyncio.get_running_loop()
+    except RuntimeError:
+        loop = asyncio.new_event_loop()
+    yield loop
+    loop.close()
 
 
 @pytest.fixture()
@@ -54,24 +45,38 @@ def sandbox(template, debug):
                 )
 
 
-@pytest_asyncio.fixture(loop_scope="function")
-async def async_sandbox(template, debug):
-    async_sandbox = await AsyncSandbox.create(template, timeout=timeout, debug=debug)
+@pytest.fixture
+def async_sandbox_factory(request, template, debug, event_loop):
+    """Factory for creating async sandboxes with proper cleanup."""
+    async def factory(template_override=None, **kwargs):
+        template_name = template_override or template
+        kwargs.setdefault("timeout", timeout)
+        kwargs.setdefault("debug", debug)
+        
+        sandbox = await AsyncSandbox.create(template_name, **kwargs)
 
-    try:
-        yield async_sandbox
-    finally:
-        try:
-            await async_sandbox.kill()
-        except RuntimeError as e:
-            # Ignore "Event loop is closed" errors during cleanup in pytest-xdist
-            if "Event loop is closed" not in str(e):
-                raise
-        except:  # noqa: E722
-            if not debug:
-                warning(
-                    "Failed to kill sandbox — this is expected if the test runs with local envd."
-                )
+        def kill():
+            async def _kill():
+                try:
+                    await sandbox.kill()
+                except:  # noqa: E722
+                    if not debug:
+                        warning(
+                            "Failed to kill sandbox — this is expected if the test runs with local envd."
+                        )
+
+            event_loop.run_until_complete(_kill())
+
+        request.addfinalizer(kill)
+        return sandbox
+
+    return factory
+
+
+@pytest.fixture
+async def async_sandbox(async_sandbox_factory):
+    """Default async sandbox fixture."""
+    return await async_sandbox_factory()
 
 
 @pytest.fixture