testdriverai 7.1.4 → 7.2.2

package/lib/vitest/hooks.mjs

@@ -19,6 +19,7 @@ import fs from 'fs';
 import os from 'os';
 import path from 'path';
 import { vi } from 'vitest';
+import chalk from 'chalk';
 import TestDriverSDK from '../../sdk.js';
 
 /**
@@ -103,7 +104,6 @@ function setupConsoleSpy(client, taskId) {
   // Store spies on client for cleanup
   client._consoleSpies = { logSpy, errorSpy, warnSpy, infoSpy };
 
-  console.log(`[testdriver] Console spy set up for task: ${taskId}`);
 }
 
 /**
@@ -162,6 +162,13 @@ export function TestDriver(context, options = {}) {
   // Merge options: plugin global options < test-specific options
   const mergedOptions = { ...pluginOptions, ...options };
   
+  // Support TD_OS environment variable for specifying target OS (linux, mac, windows)
+  // Priority: test options > plugin options > environment variable > default (linux)
+  if (!mergedOptions.os && process.env.TD_OS) {
+    mergedOptions.os = process.env.TD_OS;
+    console.log(`[testdriver] Set mergedOptions.os = ${mergedOptions.os} from TD_OS environment variable`);
+  }
+  
   // Extract TestDriver-specific options
   const apiKey = mergedOptions.apiKey || process.env.TD_API_KEY;
   
@@ -191,9 +198,7 @@ export function TestDriver(context, options = {}) {
         
         await testdriver.auth();
         await testdriver.connect();
-        
-        console.log('[testdriver] ✅ Connected to sandbox');
-        
+                
         if (debugConsoleSpy) {
           console.log('[DEBUG] After connect - sandbox.instanceSocketConnected:', testdriver.sandbox?.instanceSocketConnected);
           console.log('[DEBUG] After connect - sandbox.send:', typeof testdriver.sandbox?.send);
@@ -213,7 +218,6 @@ export function TestDriver(context, options = {}) {
           : `touch ${logPath}`;
         
         await testdriver.exec(shell, createLogCmd, 10000, true);
-        console.log('[testdriver] ✅ Created log file:', logPath);
         
         // Add automatic log tracking when dashcam starts
         // Store original start method
@@ -226,56 +230,45 @@ export function TestDriver(context, options = {}) {
   // Register cleanup handler with dashcam.stop()
   if (!lifecycleHandlers.has(context.task)) {
     const cleanup = async () => {
-      console.log('[testdriver] Cleaning up TestDriver client...');
       try {
         // Stop dashcam if it was started - with timeout to prevent hanging
         if (testdriver._dashcam && testdriver._dashcam.recording) {
           try {
-            // Add a timeout wrapper to prevent dashcam.stop from hanging indefinitely
-            const stopWithTimeout = Promise.race([
-              testdriver.dashcam.stop(),
-              new Promise((_, reject) => 
-                setTimeout(() => reject(new Error('Dashcam stop timed out after 30s')), 30000)
-              )
-            ]);
+            const dashcamUrl = await testdriver.dashcam.stop();
+            console.log('');
+            console.log('🎥' + chalk.yellow(` Dashcam URL`) + `: ${dashcamUrl}`);
+            console.log('');
+            // Write test result to file for the reporter (cross-process communication)
+            // This should happen regardless of whether dashcam succeeded, to ensure platform info is available
+            const testId = context.task.id;
+            const platform = testdriver.os || 'linux';
+            const absolutePath = context.task.file?.filepath || context.task.file?.name || 'unknown';
+            const projectRoot = process.cwd();
+            const testFile = absolutePath !== 'unknown'
+              ? path.relative(projectRoot, absolutePath)
+              : absolutePath;
+            
+            // Create results directory if it doesn't exist
+            const resultsDir = path.join(os.tmpdir(), 'testdriver-results');
+            if (!fs.existsSync(resultsDir)) {
+              fs.mkdirSync(resultsDir, { recursive: true });
+            }
+            
+            // Write test result file
+            const testResultFile = path.join(resultsDir, `${testId}.json`);
+            const testResult = {
+              dashcamUrl: dashcamUrl || null,
+              platform,
+              testFile,
+              testOrder: 0,
+              sessionId: testdriver.getSessionId(),
+            };
             
-            const dashcamUrl = await stopWithTimeout;
-            console.log('🎥 Dashcam URL:', dashcamUrl);
+            fs.writeFileSync(testResultFile, JSON.stringify(testResult, null, 2));
             
-            // Write dashcam URL to file for the reporter (cross-process communication)
-            if (dashcamUrl) {
-              const testId = context.task.id;
-              const platform = testdriver.os || 'linux';
-              const absolutePath = context.task.file?.filepath || context.task.file?.name || 'unknown';
-              const projectRoot = process.cwd();
-              const testFile = absolutePath !== 'unknown'
-                ? path.relative(projectRoot, absolutePath)
-                : absolutePath;
-              
-              // Create results directory if it doesn't exist
-              const resultsDir = path.join(os.tmpdir(), 'testdriver-results');
-              if (!fs.existsSync(resultsDir)) {
-                fs.mkdirSync(resultsDir, { recursive: true });
-              }
-              
-              // Write test result file
-              const testResultFile = path.join(resultsDir, `${testId}.json`);
-              const testResult = {
-                dashcamUrl,
-                platform,
-                testFile,
-                testOrder: 0,
-                sessionId: testdriver.getSessionId(),
-              };
-              
-              fs.writeFileSync(testResultFile, JSON.stringify(testResult, null, 2));
-              console.log(`[testdriver] ✅ Wrote dashcam URL to ${testResultFile}`);
-              
-              // Also register in memory if plugin is available
-              if (globalThis.__testdriverPlugin?.registerDashcamUrl) {
-                globalThis.__testdriverPlugin.registerDashcamUrl(testId, dashcamUrl, platform);
-                console.log(`[testdriver] ✅ Registered dashcam URL in memory for test ${testId}`);
-              }
+            // Also register in memory if plugin is available
+            if (globalThis.__testdriverPlugin?.registerDashcamUrl) {
+              globalThis.__testdriverPlugin.registerDashcamUrl(testId, dashcamUrl, platform);
             }
           } catch (error) {
             // Log more detailed error information for debugging
@@ -305,7 +298,6 @@ export function TestDriver(context, options = {}) {
           testdriver.disconnect(),
           new Promise((resolve) => setTimeout(resolve, 5000)) // 5s timeout for disconnect
         ]);
-        console.log('✅ Client disconnected');
       } catch (error) {
         console.error('Error disconnecting client:', error);
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "testdriverai",
-  "version": "7.1.4",
+  "version": "7.2.2",
   "description": "Next generation autonomous AI agent for end-to-end testing of web & desktop",
   "main": "sdk.js",
   "exports": {