added tests for kernel restart

Author: mishushakov

js/tests/supervisord.test.ts

@@ -43,6 +43,25 @@ sandboxTest('restart after jupyter kill', async ({ sandbox }) => {
   expect(result.text).toEqual('1')
 })
 
+sandboxTest('restart after jupyter kernel kill', async ({ sandbox }) => {
+  // Verify code execution works initially
+  const code = await sandbox.runCode('x = 42; x')
+  expect(code.text).toEqual('42')
+
+  // Kill all jupyter kernel processes as root
+  try {
+    await sandbox.commands.run("kill -9 $(pgrep -f 'ipykernel_launcher')", {
+      user: 'root',
+    })
+  } catch {
+    // Expected — the kill may terminate the command handle
+  }
+
+  // Code execution still works but the variable is undefined
+  const code2 = await sandbox.runCode('x')
+  expect(code2.error!.value).toEqual("name 'x' is not defined")
+})
+
 sandboxTest('restart after code-interpreter kill', async ({ sandbox }) => {
   // Verify health is up initially
   const initialHealth = await waitForHealth(sandbox)

python/tests/async/test_async_supervisord.py

@@ -39,6 +39,25 @@ async def test_restart_after_jupyter_kill(async_sandbox: AsyncSandbox):
     assert result.text == "1"
 
 
+async def test_restart_after_jupyter_kernel_kill(async_sandbox: AsyncSandbox):
+    # Verify code execution works initially
+    initial = await async_sandbox.run_code("x = 42; x")
+    assert initial.text == "42"
+
+    # Kill all jupyter kernel processes as root
+    try:
+        await async_sandbox.commands.run(
+            "kill -9 $(pgrep -f 'ipykernel_launcher')", user="root"
+        )
+    except Exception:
+        pass
+
+    # Code execution still works but the variable is undefined
+    result = await async_sandbox.run_code("x")
+    assert result.error is not None
+    assert result.error.value == "name 'x' is not defined"
+
+
 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_supervisord.py

@@ -37,6 +37,25 @@ def test_restart_after_jupyter_kill(sandbox: Sandbox):
     assert result.text == "1"
 
 
+def test_restart_after_jupyter_kernel_kill(sandbox: Sandbox):
+    # Verify code execution works initially
+    initial = sandbox.run_code("x = 42; x")
+    assert initial.text == "42"
+
+    # Kill all jupyter kernel processes as root
+    try:
+        sandbox.commands.run(
+            "kill -9 $(pgrep -f 'ipykernel_launcher')", user="root"
+        )
+    except Exception:
+        pass
+
+    # Code execution still works but the variable is undefined
+    result = sandbox.run_code("x")
+    assert result.error is not None
+    assert result.error.value == "name 'x' is not defined"
+
+
 def test_restart_after_code_interpreter_kill(sandbox: Sandbox):
     # Verify health is up initially
     assert wait_for_health(sandbox)