fixes local Docker testing setup

js/tests/reconnect.test.ts

@@ -1,9 +1,9 @@
 import { expect } from 'vitest'
 
 import { Sandbox } from '../src'
-import { sandboxTest } from './setup'
+import { isDebug, sandboxTest } from './setup'
 
-sandboxTest('reconnect', async ({ sandbox }) => {
+sandboxTest.skipIf(isDebug)('reconnect', async ({ sandbox }) => {
   sandbox = await Sandbox.connect(sandbox.sandboxId)
 
   const result = await sandbox.runCode('x =1; x')

js/tests/systemd.test.ts

@@ -1,5 +1,5 @@
 import { expect } from 'vitest'
-import { sandboxTest, wait } from './setup'
+import { isDebug, sandboxTest, wait } from './setup'
 
 async function waitForHealth(sandbox: any, maxRetries = 10, intervalMs = 100) {
   for (let i = 0; i < maxRetries; i++) {
@@ -18,7 +18,7 @@ async function waitForHealth(sandbox: any, maxRetries = 10, intervalMs = 100) {
   return false
 }
 
-sandboxTest('restart after jupyter kill', async ({ sandbox }) => {
+sandboxTest.skipIf(isDebug)('restart after jupyter kill', async ({ sandbox }) => {
   // Verify health is up initially
   const initialHealth = await waitForHealth(sandbox)
   expect(initialHealth).toBe(true)
@@ -43,7 +43,7 @@ sandboxTest('restart after jupyter kill', async ({ sandbox }) => {
   expect(result.text).toEqual('1')
 })
 
-sandboxTest('restart after code-interpreter kill', async ({ sandbox }) => {
+sandboxTest.skipIf(isDebug)('restart after code-interpreter kill', async ({ sandbox }) => {
   // Verify health is up initially
   const initialHealth = await waitForHealth(sandbox)
   expect(initialHealth).toBe(true)

python/tests/async/test_async_reconnect.py

@@ -1,6 +1,9 @@
+import pytest
+
 from e2b_code_interpreter.code_interpreter_async import AsyncSandbox
 
 
+@pytest.mark.skip_debug
 async def test_reconnect(async_sandbox: AsyncSandbox):
     sandbox_id = async_sandbox.sandbox_id
 

python/tests/async/test_async_systemd.py

@@ -1,5 +1,7 @@
 import asyncio
 
+import pytest
+
 from e2b_code_interpreter.code_interpreter_async import AsyncSandbox
 
 
@@ -17,6 +19,7 @@ async def wait_for_health(sandbox: AsyncSandbox, max_retries=10, interval_ms=100
     return False
 
 
+@pytest.mark.skip_debug
 async def test_restart_after_jupyter_kill(async_sandbox: AsyncSandbox):
     # Verify health is up initially
     assert await wait_for_health(async_sandbox)
@@ -39,6 +42,7 @@ async def test_restart_after_jupyter_kill(async_sandbox: AsyncSandbox):
     assert result.text == "1"
 
 
+@pytest.mark.skip_debug
 async def test_restart_after_code_interpreter_kill(async_sandbox: AsyncSandbox):
     # Verify health is up initially
     assert await wait_for_health(async_sandbox)

python/tests/sync/test_reconnect.py

@@ -1,6 +1,9 @@
+import pytest
+
 from e2b_code_interpreter.code_interpreter_sync import Sandbox
 
 
+@pytest.mark.skip_debug
 def test_reconnect(sandbox: Sandbox):
     sandbox_id = sandbox.sandbox_id
 

python/tests/sync/test_systemd.py

@@ -1,5 +1,7 @@
 import time
 
+import pytest
+
 from e2b_code_interpreter.code_interpreter_sync import Sandbox
 
 
@@ -17,6 +19,7 @@ def wait_for_health(sandbox: Sandbox, max_retries=10, interval_ms=100):
     return False
 
 
+@pytest.mark.skip_debug
 def test_restart_after_jupyter_kill(sandbox: Sandbox):
     # Verify health is up initially
     assert wait_for_health(sandbox)
@@ -37,6 +40,7 @@ def test_restart_after_jupyter_kill(sandbox: Sandbox):
     assert result.text == "1"
 
 
+@pytest.mark.skip_debug
 def test_restart_after_code_interpreter_kill(sandbox: Sandbox):
     # Verify health is up initially
     assert wait_for_health(sandbox)

template/template.py

@@ -144,7 +144,7 @@ def make_template(
     template = template.set_user("user").set_workdir("/home/user")
 
     if is_docker:
-        start_cmd = "sudo /root/.jupyter/start-up.sh"
+        start_cmd = "sudo --preserve-env=E2B_LOCAL /root/.jupyter/start-up.sh"
     else:
         start_cmd = "sudo systemctl start jupyter"