update test

Author: mishushakov

js/tests/supervisord.test.ts

@@ -57,9 +57,13 @@ sandboxTest('restart after jupyter kernel kill', async ({ sandbox }) => {
     // Expected — the kill may terminate the command handle
   }
 
-  // Code execution still works but the variable is undefined
+  // First call after kill returns context restarted error
   const code2 = await sandbox.runCode('x')
-  expect(code2.error!.value).toEqual("name 'x' is not defined")
+  expect(code2.error!.value).toEqual('Context was restarted')
+
+  // Subsequent call works normally
+  const code3 = await sandbox.runCode('y = 1; y')
+  expect(code3.text).toEqual('1')
 })
 
 sandboxTest('restart after code-interpreter kill', async ({ sandbox }) => {

python/tests/async/test_async_supervisord.py

@@ -52,10 +52,14 @@ async def test_restart_after_jupyter_kernel_kill(async_sandbox: AsyncSandbox):
     except Exception:
         pass
 
-    # Code execution still works but the variable is undefined
+    # First call after kill returns context restarted error
     result = await async_sandbox.run_code("x")
     assert result.error is not None
-    assert result.error.value == "name 'x' is not defined"
+    assert result.error.value == "Context was restarted"
+
+    # Subsequent call works normally
+    result2 = await async_sandbox.run_code("y = 1; y")
+    assert result2.text == "1"
 
 
 async def test_restart_after_code_interpreter_kill(async_sandbox: AsyncSandbox):

python/tests/sync/test_supervisord.py

@@ -50,10 +50,14 @@ def test_restart_after_jupyter_kernel_kill(sandbox: Sandbox):
     except Exception:
         pass
 
-    # Code execution still works but the variable is undefined
+    # First call after kill returns context restarted error
     result = sandbox.run_code("x")
     assert result.error is not None
-    assert result.error.value == "name 'x' is not defined"
+    assert result.error.value == "Context was restarted"
+
+    # Subsequent call works normally
+    result2 = sandbox.run_code("y = 1; y")
+    assert result2.text == "1"
 
 
 def test_restart_after_code_interpreter_kill(sandbox: Sandbox):