changed tests to use test template in CI

Author: mishushakov

.github/workflows/build_template.yml

@@ -6,7 +6,9 @@ on:
       E2B_API_KEY:
         required: true
       E2B_DOMAIN:
-        required: true
+        required: false
+      E2B_TEMPLATE_CONFIG:
+        required: false
 
 permissions:
   contents: read
@@ -54,8 +56,9 @@ jobs:
         run: npm install -g @e2b/cli
 
       - name: Build E2B template
-        run: e2b template build --config e2b.test.toml
+        run: e2b template build --config ${{ secrets.E2B_TEMPLATE_CONFIG }}
         working-directory: ./template
         env:
           E2B_ACCESS_TOKEN: ${{ secrets.E2B_ACCESS_TOKEN }}
-          E2B_DOMAIN: ${{ secrets.E2B_DOMAIN }}
+          E2B_DOMAIN: ${{ secrets.E2B_DOMAIN }}
+          E2B_TEMPLATE_CONFIG: ${{ secrets.E2B_TEMPLATE_CONFIG }}

.github/workflows/js_tests.yml

@@ -6,7 +6,9 @@ on:
       E2B_API_KEY:
         required: true
       E2B_DOMAIN:
-        required: true
+        required: false
+      E2B_TEMPLATE_CONFIG:
+        required: false
 
 permissions:
   contents: read

.github/workflows/pull_request.yml

@@ -19,6 +19,8 @@ jobs:
     secrets:
       E2B_API_KEY: ${{ secrets.E2B_API_KEY }}
       E2B_DOMAIN: ${{ secrets.E2B_DOMAIN }}
+      E2B_TEMPLATE: ${{ secrets.E2B_TEMPLATE }}
+      E2B_TEMPLATE_CONFIG: ${{ secrets.E2B_TEMPLATE_CONFIG }}
   js-sdk:
     uses: ./.github/workflows/js_tests.yml
     secrets:
@@ -29,8 +31,10 @@ jobs:
     secrets:
       E2B_API_KEY: ${{ secrets.E2B_API_KEY }}
       E2B_DOMAIN: ${{ secrets.E2B_DOMAIN }}
+      E2B_TEMPLATE: ${{ secrets.E2B_TEMPLATE }}
   charts-tests:
     uses: ./.github/workflows/charts_tests.yml
     secrets:
       E2B_API_KEY: ${{ secrets.E2B_API_KEY }}
       E2B_DOMAIN: ${{ secrets.E2B_DOMAIN }}
+      E2B_TEMPLATE: ${{ secrets.E2B_TEMPLATE }}

.github/workflows/python_tests.yml

@@ -5,6 +5,10 @@ on:
     secrets:
       E2B_API_KEY:
         required: true
+      E2B_DOMAIN:
+        required: false
+      E2B_TEMPLATE_CONFIG:
+        required: false
 
 permissions:
   contents: read

js/tests/setup.ts

@@ -3,14 +3,18 @@ import { test as base } from 'vitest'
 
 const timeoutMs = 60_000
 
+const template = process.env.E2B_TEMPLATE || 'code-interpreter-v1'
+
 interface SandboxFixture {
   sandbox: Sandbox
 }
 
 export const sandboxTest = base.extend<SandboxFixture>({
   sandbox: [
     async ({}, use) => {
-      const sandbox = await Sandbox.create({ timeoutMs })
+      const sandbox = await Sandbox.create(template, {
+        timeoutMs,
+      })
       try {
         await use(sandbox)
       } finally {

python/tests/conftest.py

@@ -9,10 +9,14 @@
 
 timeout = 60
 
+@pytest.fixture()
+def template():
+    return os.getenv("E2B_TEMPLATE", "code-interpreter-v1")
+
 
 @pytest.fixture()
-def sandbox(debug):
-    sandbox = Sandbox(timeout=timeout)
+def sandbox(template, debug):
+    sandbox = Sandbox(template, timeout=timeout)
 
     try:
         yield sandbox
@@ -27,8 +31,8 @@ def sandbox(debug):
 
 
 @pytest_asyncio.fixture
-async def async_sandbox(debug):
-    sandbox = await AsyncSandbox.create(timeout=timeout)
+async def async_sandbox(template, debug):
+    sandbox = await AsyncSandbox.create(template, timeout=timeout)
 
     try:
         yield sandbox