ci: add unified SDK tests workflow with staging

Author: dobrac

.github/workflows/cli_tests.yml

@@ -2,12 +2,14 @@ name: Test CLI
 
 on:
   workflow_call:
+    inputs:
+      E2B_DOMAIN:
+        required: false
+        type: string
+        default: ''
     secrets:
       E2B_API_KEY:
         required: true
-  pull_request:
-    branches:
-      - main
 
 permissions:
   contents: read
@@ -67,3 +69,4 @@ jobs:
         working-directory: ./packages/cli
         env:
           E2B_API_KEY: ${{ secrets.E2B_API_KEY }}
+          E2B_DOMAIN: ${{ inputs.E2B_DOMAIN }}

.github/workflows/js_sdk_tests.yml

@@ -2,12 +2,14 @@ name: Test JS SDK
 
 on:
   workflow_call:
+    inputs:
+      E2B_DOMAIN:
+        required: false
+        type: string
+        default: ''
     secrets:
       E2B_API_KEY:
         required: true
-  pull_request:
-    branches:
-      - main
 
 permissions:
   contents: read
@@ -83,6 +85,7 @@ jobs:
         run: pnpm test
         env:
           E2B_API_KEY: ${{ secrets.E2B_API_KEY }}
+          E2B_DOMAIN: ${{ inputs.E2B_DOMAIN }}
 
       - name: Install Bun
         uses: oven-sh/setup-bun@v2
@@ -91,6 +94,7 @@ jobs:
         run: pnpm test:bun
         env:
           E2B_API_KEY: ${{ secrets.E2B_API_KEY }}
+          E2B_DOMAIN: ${{ inputs.E2B_DOMAIN }}
 
       - name: Install Deno
         uses: denoland/setup-deno@v1
@@ -101,3 +105,4 @@ jobs:
         run: pnpm test:deno
         env:
           E2B_API_KEY: ${{ secrets.E2B_API_KEY }}
+          E2B_DOMAIN: ${{ inputs.E2B_DOMAIN }}

.github/workflows/python_sdk_tests.yml

@@ -2,12 +2,14 @@ name: Test Python SDK
 
 on:
   workflow_call:
+    inputs:
+      E2B_DOMAIN:
+        required: false
+        type: string
+        default: ''
     secrets:
       E2B_API_KEY:
         required: true
-  pull_request:
-    branches:
-      - main
 
 permissions:
   contents: read
@@ -58,3 +60,4 @@ jobs:
         run: poetry run pytest --verbose --numprocesses=4
         env:
           E2B_API_KEY: ${{ secrets.E2B_API_KEY }}
+          E2B_DOMAIN: ${{ inputs.E2B_DOMAIN }}

.github/workflows/sdk_tests.yml

@@ -0,0 +1,72 @@
+name: SDK Tests
+
+on:
+  pull_request:
+    branches:
+      - main
+  workflow_dispatch:
+
+permissions:
+  contents: read
+
+jobs:
+  environments:
+    name: Define environments
+    runs-on: ubuntu-latest
+    outputs:
+      matrix: ${{ steps.set.outputs.matrix }}
+    steps:
+      - id: set
+        run: |
+          matrix='[
+            {
+              "env_name": "Production",
+              "e2b_domain": "",
+              "api_key_secret": "E2B_API_KEY"
+            },
+            {
+              "env_name": "Staging",
+              "e2b_domain": "${{ vars.E2B_DOMAIN_STAGING }}",
+              "api_key_secret": "E2B_API_KEY_STAGING"
+            }
+          ]'
+          echo "matrix=$(echo $matrix | jq -c .)" >> "$GITHUB_OUTPUT"
+
+  js-tests:
+    needs: environments
+    name: JS SDK Tests (${{ matrix.env_name }})
+    strategy:
+      fail-fast: false
+      matrix:
+        include: ${{ fromJSON(needs.environments.outputs.matrix) }}
+    uses: ./.github/workflows/js_sdk_tests.yml
+    with:
+      E2B_DOMAIN: ${{ matrix.e2b_domain }}
+    secrets:
+      E2B_API_KEY: ${{ secrets[matrix.api_key_secret] }}
+
+  python-tests:
+    needs: environments
+    name: Python SDK Tests (${{ matrix.env_name }})
+    strategy:
+      fail-fast: false
+      matrix:
+        include: ${{ fromJSON(needs.environments.outputs.matrix) }}
+    uses: ./.github/workflows/python_sdk_tests.yml
+    with:
+      E2B_DOMAIN: ${{ matrix.e2b_domain }}
+    secrets:
+      E2B_API_KEY: ${{ secrets[matrix.api_key_secret] }}
+
+  cli-tests:
+    needs: environments
+    name: CLI Tests (${{ matrix.env_name }})
+    strategy:
+      fail-fast: false
+      matrix:
+        include: ${{ fromJSON(needs.environments.outputs.matrix) }}
+    uses: ./.github/workflows/cli_tests.yml
+    with:
+      E2B_DOMAIN: ${{ matrix.e2b_domain }}
+    secrets:
+      E2B_API_KEY: ${{ secrets[matrix.api_key_secret] }}