Test new error handling indicating when the sandbox is not found

packages/js-sdk/tests/sandbox/commands/sendStdin.test.ts

@@ -1,5 +1,5 @@
 import { assert } from 'vitest'
-import { sandboxTest } from '../../setup.js'
+import { sandboxTest, wait } from '../../setup.js'
 
 sandboxTest('send stdin to process', async ({ sandbox }) => {
   const text = 'Hello, World!'
@@ -37,6 +37,8 @@ sandboxTest('send special characters to stdin', async ({ sandbox }) => {
 
   await sandbox.commands.sendStdin(cmd.pid, text)
 
+  await wait(5_000)
+
   for (let i = 0; i < 5; i++) {
     if (cmd.stdout === text) {
       break

packages/js-sdk/tests/sandbox/connect.test.ts

@@ -1,4 +1,4 @@
-import { test, assert } from 'vitest'
+import { assert, test } from 'vitest'
 
 import { Sandbox } from '../../src'
 import { isDebug, template } from '../setup.js'
@@ -19,3 +19,25 @@ test('connect', async () => {
     }
   }
 })
+
+test('connect to non-running sandbox', async () => {
+  const sbx = await Sandbox.create(template, { timeoutMs: 10_000 })
+  let isKilled = false
+
+  try {
+    const isRunning = await sbx.isRunning()
+    assert.isTrue(isRunning)
+    await sbx.kill()
+    isKilled = true
+
+    const sbxConnection = await Sandbox.connect(sbx.sandboxId)
+    const isRunning2 = await sbxConnection.isRunning()
+    assert.isFalse(isRunning2)
+
+    await sbxConnection.commands.run('echo "hello"')
+  } finally {
+    if (!isKilled) {
+      await sbx.kill()
+    }
+  }
+})

packages/js-sdk/tests/sandbox/host.test.ts

@@ -2,30 +2,60 @@ import { assert } from 'vitest'
 
 import { isDebug, sandboxTest, wait } from '../setup.js'
 
-sandboxTest('ping server in sandbox', async ({ sandbox }) => {
-  const cmd = await sandbox.commands.run('python -m http.server 8000', { background: true })
+sandboxTest.skipIf(isDebug)(
+  'ping server in running sandbox',
+  async ({ sandbox }) => {
+    const cmd = await sandbox.commands.run('python -m http.server 8000', {
+      background: true,
+    })
 
-  try {
-    await wait(1000)
+    try {
+      await wait(10_000)
 
-    const host = sandbox.getHost(8000)
+      const host = sandbox.getHost(8000)
 
-    let res = await fetch(`${isDebug ? 'http' : 'https'}://${host}`)
+      const res = await fetch(`${isDebug ? 'http' : 'https'}://${host}`)
 
-    for (let i = 0; i < 10; i++) {
-      if (res.status === 200) {
-        break
+      assert.equal(res.status, 200)
+    } finally {
+      try {
+        await cmd.kill()
+      } catch (e) {
+        console.error(e)
       }
-
-      res = await fetch(`${isDebug ? 'http' : 'https'}://${host}`)
-      await wait(500)
     }
-    assert.equal(res.status, 200)
-  } finally {
+  }
+)
+
+sandboxTest.skipIf(isDebug)(
+  'ping server in non-running sandbox',
+  async ({ sandbox }) => {
+    const cmd = await sandbox.commands.run('python -m http.server 8000', {
+      background: true,
+    })
+
     try {
-      await cmd.kill()
-    } catch (e) {
-      console.error(e)
+      await wait(10_000)
+
+      const host = sandbox.getHost(8000)
+
+      const res = await fetch(`${isDebug ? 'http' : 'https'}://${host}`)
+
+      assert.equal(res.status, 200)
+
+      await sandbox.kill()
+
+      const res2 = await fetch(`${isDebug ? 'http' : 'https'}://${host}`)
+      assert.equal(res2.status, 502)
+
+      const text = await res2.text()
+      assert.equal(text, 'Sandbox does not exist.')
+    } finally {
+      try {
+        await cmd.kill()
+      } catch (e) {
+        console.error(e)
+      }
     }
   }
-}, { timeout: 60000 })
+)