updated tests

Author: mishushakov

js/src/sandbox.ts

@@ -398,6 +398,10 @@ export class Sandbox extends BaseSandbox {
           'Content-Type': 'application/json',
           ...this.connectionConfig.headers,
         },
+        keepalive: true,
+        signal: this.connectionConfig.getSignal(
+          this.connectionConfig.requestTimeoutMs
+        ),
       })
 
       const error = await extractError(res)

js/tests/contexts.test.ts

@@ -47,5 +47,17 @@ sandboxTest('list contexts', async ({ sandbox }) => {
 sandboxTest('restart context', async ({ sandbox }) => {
   const context = await sandbox.createCodeContext()
 
+  // set a variable in the context
+  await sandbox.runCode('x = 1', { context: context })
+
+  // restart the context
   await sandbox.restartCodeContext(context.id)
+
+  // check that the variable no longer exists
+  const execution = await sandbox.runCode('x', { context: context })
+
+  // check for an NameError with message "name 'x' is not defined"
+  expect(execution.error).toBeDefined()
+  expect(execution.error?.name).toBe('NameError')
+  expect(execution.error?.value).toBe("name 'x' is not defined")
 })

python/tests/async/test_async_contexts.py

@@ -31,6 +31,9 @@ async def test_remove_context(async_sandbox: AsyncSandbox):
 
     await async_sandbox.remove_code_context(context.id)
 
+    contexts = await async_sandbox.list_code_contexts()
+    assert context.id not in [ctx.id for ctx in contexts]
+
 
 async def test_list_contexts(async_sandbox: AsyncSandbox):
     contexts = await async_sandbox.list_code_contexts()
@@ -44,4 +47,16 @@ async def test_list_contexts(async_sandbox: AsyncSandbox):
 async def test_restart_context(async_sandbox: AsyncSandbox):
     context = await async_sandbox.create_code_context()
 
+    # set a variable in the context
+    await async_sandbox.run_code("x = 1", context=context)
+
+    # restart the context
     await async_sandbox.restart_code_context(context.id)
+
+    # check that the variable no longer exists
+    execution = await async_sandbox.run_code("x", context=context)
+
+    # check for a NameError with message "name 'x' is not defined"
+    assert execution.error is not None
+    assert execution.error.name == "NameError"
+    assert execution.error.value == "name 'x' is not defined"

python/tests/sync/test_contexts.py

@@ -31,6 +31,9 @@ def test_remove_context(sandbox: Sandbox):
 
     sandbox.remove_code_context(context.id)
 
+    contexts = sandbox.list_code_contexts()
+    assert context.id not in [ctx.id for ctx in contexts]
+
 
 def test_list_contexts(sandbox: Sandbox):
     contexts = sandbox.list_code_contexts()
@@ -44,4 +47,16 @@ def test_list_contexts(sandbox: Sandbox):
 def test_restart_context(sandbox: Sandbox):
     context = sandbox.create_code_context()
 
+    # set a variable in the context
+    sandbox.run_code("x = 1", context=context)
+
+    # restart the context
     sandbox.restart_code_context(context.id)
+
+    # check that the variable no longer exists
+    execution = sandbox.run_code("x", context=context)
+
+    # check for a NameError with message "name 'x' is not defined"
+    assert execution.error is not None
+    assert execution.error.name == "NameError"
+    assert execution.error.value == "name 'x' is not defined"