fix tests

Author: dobrac

packages/js-sdk/tests/connectionConfig.test.ts

@@ -7,6 +7,7 @@ let originalEnv: { [key: string]: string | undefined }
 beforeEach(() => {
   originalEnv = {
     E2B_API_URL: process.env.E2B_API_URL,
+    E2B_DOMAIN: process.env.E2B_DOMAIN,
     E2B_DEBUG: process.env.E2B_DEBUG,
   }
 })
@@ -25,6 +26,7 @@ afterEach(() => {
 test('api_url defaults correctly', () => {
   // Ensure no env vars interfere
   delete process.env.E2B_API_URL
+  delete process.env.E2B_DOMAIN
   delete process.env.E2B_DEBUG
 
   const config = new ConnectionConfig()

packages/js-sdk/tests/setup.ts

@@ -101,7 +101,7 @@ export async function wait(ms: number) {
  * Supports msw path parameters like :templateID
  */
 export function apiUrl(path: string): string {
-  const domain = process.env.E2B_DOMAIN || 'e2b.dev'
+  const domain = process.env.E2B_DOMAIN || 'e2b.app'
   return `https://api.${domain}${path}`
 }
 

packages/python-sdk/e2b/template_async/main.py

@@ -58,11 +58,12 @@ async def _build(
 
         # Create template
         if on_build_logs:
+            tags_msg = f" with tags: {', '.join(tags)}" if tags else ""
             on_build_logs(
                 LogEntry(
                     timestamp=datetime.now(),
                     level="info",
-                    message=f"Requesting build for template: {name}{f' with tags: {", ".join(tags)}' if tags else ''}",
+                    message=f"Requesting build for template: {name}{tags_msg}",
                 )
             )
 

packages/python-sdk/e2b/template_sync/main.py

@@ -58,11 +58,12 @@ def _build(
 
         # Create template
         if on_build_logs:
+            tags_msg = f" with tags: {', '.join(tags)}" if tags else ""
             on_build_logs(
                 LogEntry(
                     timestamp=datetime.now(),
                     level="info",
-                    message=f"Requesting build for template: {name}{f' with tags: {", ".join(tags)}' if tags else ''}",
+                    message=f"Requesting build for template: {name}{tags_msg}",
                 )
             )
 

packages/python-sdk/tests/conftest.py

@@ -1,7 +1,7 @@
 import asyncio
 import os
 import uuid
-from typing import Callable, Optional
+from typing import Callable, Dict, Optional
 from uuid import uuid4
 
 import pytest
@@ -101,14 +101,38 @@ def _build(
         skip_cache: bool = False,
         on_build_logs: Optional[Callable[[LogEntry], None]] = None,
     ):
-        return Template.build(
-            template,
-            name or f"e2b-test-{uuid4()}",
-            cpu_count=1,
-            memory_mb=1024,
-            skip_cache=skip_cache,
-            on_build_logs=on_build_logs,
-        )
+        build_name = name or f"e2b-test-{uuid4()}"
+        build_info: Dict[str, Optional[str]] = {"template_id": None, "build_id": None}
+
+        def capture_logs(log: LogEntry):
+            import re
+
+            if "Template created with ID:" in log.message:
+                match = re.search(
+                    r"Template created with ID: ([^,]+), Build ID: (.+)", log.message
+                )
+                if match:
+                    build_info["template_id"] = match.group(1)
+                    build_info["build_id"] = match.group(2)
+            if on_build_logs:
+                on_build_logs(log)
+
+        try:
+            return Template.build(
+                template,
+                build_name,
+                cpu_count=1,
+                memory_mb=1024,
+                skip_cache=skip_cache,
+                on_build_logs=capture_logs,
+            )
+        except Exception as e:
+            print(
+                f"\n[BUILD FAILED] name={build_name}, "
+                f"template_id={build_info['template_id']}, "
+                f"build_id={build_info['build_id']}, error={e}"
+            )
+            raise
 
     return _build
 
@@ -121,14 +145,38 @@ async def _async_build(
         skip_cache: bool = False,
         on_build_logs: Optional[Callable[[LogEntry], None]] = None,
     ):
-        return await AsyncTemplate.build(
-            template,
-            name or f"e2b-test-{uuid4()}",
-            cpu_count=1,
-            memory_mb=1024,
-            skip_cache=skip_cache,
-            on_build_logs=on_build_logs,
-        )
+        build_name = name or f"e2b-test-{uuid4()}"
+        build_info: Dict[str, Optional[str]] = {"template_id": None, "build_id": None}
+
+        def capture_logs(log: LogEntry):
+            import re
+
+            if "Template created with ID:" in log.message:
+                match = re.search(
+                    r"Template created with ID: ([^,]+), Build ID: (.+)", log.message
+                )
+                if match:
+                    build_info["template_id"] = match.group(1)
+                    build_info["build_id"] = match.group(2)
+            if on_build_logs:
+                on_build_logs(log)
+
+        try:
+            return await AsyncTemplate.build(
+                template,
+                build_name,
+                cpu_count=1,
+                memory_mb=1024,
+                skip_cache=skip_cache,
+                on_build_logs=capture_logs,
+            )
+        except Exception as e:
+            print(
+                f"\n[BUILD FAILED] name={build_name}, "
+                f"template_id={build_info['template_id']}, "
+                f"build_id={build_info['build_id']}, error={e}"
+            )
+            raise
 
     return _async_build