test: mock all template build tests

packages/js-sdk/tests/setup.ts

@@ -56,6 +56,9 @@ async function buildTemplate(
       memoryMB: 1024,
       skipCache: options?.skipCache,
       onBuildLogs: captureLogs,
+      // The placeholder key keeps the mocked template tests independent of
+      // E2B_API_KEY being set in the environment.
+      apiKey: process.env.E2B_API_KEY ?? TEST_API_KEY,
     })
   } catch (e) {
     console.error(

packages/js-sdk/tests/template/backgroundBuild.test.ts

@@ -1,6 +1,18 @@
 import { randomUUID } from 'node:crypto'
-import { expect, test } from 'vitest'
+import { afterAll, beforeAll, expect, test } from 'vitest'
+import { setupServer } from 'msw/node'
 import { Template, waitForTimeout } from '../../src'
+import { TEST_API_KEY } from '../setup'
+import { createMockBuildApi } from './mockBuildApi'
+
+const server = setupServer(...createMockBuildApi().handlers)
+
+beforeAll(() => server.listen({ onUnhandledRequest: 'error' }))
+afterAll(() => server.close())
+
+// The placeholder key keeps the mocked template tests independent of
+// E2B_API_KEY being set in the environment.
+const apiKey = process.env.E2B_API_KEY ?? TEST_API_KEY
 
 test('build template in background', async () => {
   const template = Template()
@@ -14,12 +26,13 @@ test('build template in background', async () => {
   const buildInfo = await Template.buildInBackground(template, name, {
     cpuCount: 1,
     memoryMB: 1024,
+    apiKey,
   })
 
   // Should return quickly (within a few seconds), not wait for the full build
   expect(buildInfo).toBeDefined()
 
   // Verify the build is actually running
-  const status = await Template.getBuildStatus(buildInfo)
+  const status = await Template.getBuildStatus(buildInfo, { apiKey })
   expect(status.status).toEqual('building')
 }, 10_000)

packages/js-sdk/tests/template/build.test.ts

@@ -2,8 +2,15 @@ import fs from 'node:fs'
 import os from 'node:os'
 import path from 'node:path'
 import { afterAll, beforeAll } from 'vitest'
+import { setupServer } from 'msw/node'
 import { defaultBuildLogger, Template, waitForTimeout } from '../../src'
 import { buildTemplateTest } from '../setup'
+import { createMockBuildApi } from './mockBuildApi'
+
+const server = setupServer(...createMockBuildApi().handlers)
+
+beforeAll(() => server.listen({ onUnhandledRequest: 'error' }))
+afterAll(() => server.close())
 
 // The file context lives in a temp directory so a test run never writes into
 // the repository tree. It is created in beforeAll rather than at module load so

packages/js-sdk/tests/template/exists.test.ts

@@ -1,14 +1,26 @@
 import { randomUUID } from 'node:crypto'
-import { expect, test } from 'vitest'
+import { afterAll, beforeAll, expect, test } from 'vitest'
+import { setupServer } from 'msw/node'
 import { Template } from '../../src'
+import { TEST_API_KEY } from '../setup'
+import { createMockBuildApi } from './mockBuildApi'
+
+const server = setupServer(...createMockBuildApi().handlers)
+
+beforeAll(() => server.listen({ onUnhandledRequest: 'error' }))
+afterAll(() => server.close())
+
+// The placeholder key keeps the mocked template tests independent of
+// E2B_API_KEY being set in the environment.
+const apiKey = process.env.E2B_API_KEY ?? TEST_API_KEY
 
 test('check if base template name exists', async () => {
-  const exists = await Template.exists('base')
+  const exists = await Template.exists('base', { apiKey })
   expect(exists).toBe(true)
 })
 
 test('check non existing name', async () => {
   const nonExistingName = `nonexistent-${randomUUID()}`
-  const exists = await Template.exists(nonExistingName)
+  const exists = await Template.exists(nonExistingName, { apiKey })
   expect(exists).toBe(false)
 })

packages/js-sdk/tests/template/methods/makeSymlink.test.ts

@@ -1,5 +1,13 @@
+import { afterAll, beforeAll } from 'vitest'
+import { setupServer } from 'msw/node'
 import { Template } from '../../../src'
 import { buildTemplateTest } from '../../setup'
+import { createMockBuildApi } from '../mockBuildApi'
+
+const server = setupServer(...createMockBuildApi().handlers)
+
+beforeAll(() => server.listen({ onUnhandledRequest: 'error' }))
+afterAll(() => server.close())
 
 buildTemplateTest('make symlink', async ({ buildTemplate }) => {
   const template = Template()

packages/js-sdk/tests/template/methods/runCmd.test.ts

@@ -1,6 +1,13 @@
-import { expect } from 'vitest'
+import { afterAll, beforeAll, expect } from 'vitest'
+import { setupServer } from 'msw/node'
 import { Template } from '../../../src'
 import { buildTemplateTest } from '../../setup'
+import { createMockBuildApi } from '../mockBuildApi'
+
+const server = setupServer(...createMockBuildApi().handlers)
+
+beforeAll(() => server.listen({ onUnhandledRequest: 'error' }))
+afterAll(() => server.close())
 
 buildTemplateTest('run command', async ({ buildTemplate }) => {
   const template = Template()