updated tests

Author: mishushakov

js/tests/contexts.test.ts

@@ -5,32 +5,43 @@ import { sandboxTest } from './setup'
 sandboxTest('create context with no options', async ({ sandbox }) => {
   const context = await sandbox.createCodeContext()
 
-  expect(context.id).toBeDefined()
-  expect(context.language).toBe('python')
-  expect(context.cwd).toBe('/home/user')
+  const contexts = await sandbox.listCodeContexts()
+  const lastContext = contexts[contexts.length - 1]
+
+  expect(lastContext.id).toBeDefined()
+  expect(lastContext.language).toBe(context.language)
+  expect(lastContext.cwd).toBe(context.cwd)
 })
 
 sandboxTest('create context with options', async ({ sandbox }) => {
   const context = await sandbox.createCodeContext({
     language: 'python',
-    cwd: '/home/user/test',
+    cwd: '/root',
   })
 
-  expect(context.id).toBeDefined()
-  expect(context.language).toBe('python')
-  expect(context.cwd).toBe('/home/user/test')
+  const contexts = await sandbox.listCodeContexts()
+  const lastContext = contexts[contexts.length - 1]
+
+  expect(lastContext.id).toBeDefined()
+  expect(lastContext.language).toBe(context.language)
+  expect(lastContext.cwd).toBe(context.cwd)
 })
 
 sandboxTest('remove context', async ({ sandbox }) => {
   const context = await sandbox.createCodeContext()
 
   await sandbox.removeCodeContext(context.id)
+  const contexts = await sandbox.listCodeContexts()
+
+  expect(contexts.map((context) => context.id)).not.toContain(context.id)
 })
 
 sandboxTest('list contexts', async ({ sandbox }) => {
   const contexts = await sandbox.listCodeContexts()
 
-  expect(contexts.length).toBeGreaterThan(0)
+  // default contexts should include python and javascript
+  expect(contexts.map((context) => context.language)).toContain('python')
+  expect(contexts.map((context) => context.language)).toContain('javascript')
 })
 
 sandboxTest('restart context', async ({ sandbox }) => {

python/tests/async/test_async_contexts.py

@@ -4,9 +4,12 @@
 async def test_create_context_with_no_options(async_sandbox: AsyncSandbox):
     context = await async_sandbox.create_code_context()
 
-    assert context.id is not None
-    assert context.language == "python"
-    assert context.cwd == "/home/user"
+    contexts = await async_sandbox.list_code_contexts()
+    last_context = contexts[-1]
+
+    assert last_context.id is not None
+    assert last_context.language == context.language
+    assert last_context.cwd == context.cwd
 
 
 async def test_create_context_with_options(async_sandbox: AsyncSandbox):
@@ -15,9 +18,12 @@ async def test_create_context_with_options(async_sandbox: AsyncSandbox):
         cwd="/home/user/test",
     )
 
-    assert context.id is not None
-    assert context.language == "python"
-    assert context.cwd == "/home/user/test"
+    contexts = await async_sandbox.list_code_contexts()
+    last_context = contexts[-1]
+
+    assert last_context.id is not None
+    assert last_context.language == context.language
+    assert last_context.cwd == context.cwd
 
 
 async def test_remove_context(async_sandbox: AsyncSandbox):
@@ -29,7 +35,10 @@ async def test_remove_context(async_sandbox: AsyncSandbox):
 async def test_list_contexts(async_sandbox: AsyncSandbox):
     contexts = await async_sandbox.list_code_contexts()
 
-    assert len(contexts) > 0
+    # default contexts should include python and javascript
+    languages = [context.language for context in contexts]
+    assert "python" in languages
+    assert "javascript" in languages
 
 
 async def test_restart_context(async_sandbox: AsyncSandbox):

python/tests/sync/test_contexts.py

@@ -4,9 +4,12 @@
 def test_create_context_with_no_options(sandbox: Sandbox):
     context = sandbox.create_code_context()
 
-    assert context.id is not None
-    assert context.language == "python"
-    assert context.cwd == "/home/user"
+    contexts = sandbox.list_code_contexts()
+    last_context = contexts[-1]
+
+    assert last_context.id is not None
+    assert last_context.language == context.language
+    assert last_context.cwd == context.cwd
 
 
 def test_create_context_with_options(sandbox: Sandbox):
@@ -15,9 +18,12 @@ def test_create_context_with_options(sandbox: Sandbox):
         cwd="/home/user/test",
     )
 
-    assert context.id is not None
-    assert context.language == "python"
-    assert context.cwd == "/home/user/test"
+    contexts = sandbox.list_code_contexts()
+    last_context = contexts[-1]
+
+    assert last_context.id is not None
+    assert last_context.language == context.language
+    assert last_context.cwd == context.cwd
 
 
 def test_remove_context(sandbox: Sandbox):
@@ -29,7 +35,10 @@ def test_remove_context(sandbox: Sandbox):
 def test_list_contexts(sandbox: Sandbox):
     contexts = sandbox.list_code_contexts()
 
-    assert len(contexts) > 0
+    # default contexts should include python and javascript
+    languages = [context.language for context in contexts]
+    assert "python" in languages
+    assert "javascript" in languages
 
 
 def test_restart_context(sandbox: Sandbox):