@testdriverai/agent 7.11.85-test → 7.11.87-canary

package/agent/index.js

@@ -1729,21 +1729,31 @@ ${regression}
 
     // Set sandbox ID for reconnection (only if not creating new and recent ID exists)
     if (this.ip) {
+      // Reuse a known sandboxId for this machine so a re-auth re-attaches to the
+      // same Ably channel instead of minting a new slot and orphaning the old
+      // one (self-hosted: one machine → one sandbox).
       let instance = await this.sandbox.send({
         type: "direct",
         resolution: this.config.TD_RESOLUTION,
         ci: this.config.CI,
         ip: this.ip,
+        sandboxId: this.sandboxId || undefined,
         instanceId: this.instanceId || undefined,
       });
 
+      // Remember the slot so a later re-auth reuses it instead of minting a new one.
+      this.sandboxId =
+        instance?.instance?.sandboxId ||
+        instance?.instance?.instanceId ||
+        this.sandboxId;
+
       // Store connection params for reconnection
       // For direct IP connections, store as a direct type so reconnection
       // sends a 'direct' message instead of 'connect' with an IP as sandboxId
       this.sandbox.setConnectionParams({
         type: 'direct',
         ip: this.ip,
-        sandboxId: instance?.instance?.instanceId || instance?.instance?.sandboxId || null,
+        sandboxId: this.sandboxId,
         persist: true,
         keepAlive: this.keepAlive,
       });

package/agent/lib/provision-commands.js

@@ -125,7 +125,7 @@ function windowsProvisionCommands({ channel, configJson, sandboxId, s3DownloadUr
       commands.push(
         "Write-Host 'Checking installed runner version...'",
         "$installedVersion = ''",
-        "try { $pkg = Get-Content 'node_modules/@testdriverai/runner/package.json' -Raw | ConvertFrom-Json; $installedVersion = $pkg.version } catch {}",
+        "try { $pkg = Get-Content 'node_modules/@testdriverai/runner/package.json' -Raw -ErrorAction Stop | ConvertFrom-Json; $installedVersion = $pkg.version } catch {}",
         "$installedMinor = if ($installedVersion) { ($installedVersion -split '[.-]')[0..1] -join '.' } else { '' }",
         "if (('" + expectedMinor + "' -ne '') -and ($installedMinor -eq '" + expectedMinor + "')) {",
         "  Write-Host \"Runner already at v$installedVersion (minor " + expectedMinor + "), skipping update\"",
@@ -135,7 +135,7 @@ function windowsProvisionCommands({ channel, configJson, sandboxId, s3DownloadUr
         "  Write-Host 'Installing @testdriverai/runner@" + runnerTag + "...'",
         '  npm install @testdriverai/runner@' + runnerTag + ' --omit=dev 2>&1 | Write-Host',
         "  $newVersion = ''",
-        "  try { $newPkg = Get-Content 'node_modules/@testdriverai/runner/package.json' -Raw | ConvertFrom-Json; $newVersion = $newPkg.version } catch {}",
+        "  try { $newPkg = Get-Content 'node_modules/@testdriverai/runner/package.json' -Raw -ErrorAction Stop | ConvertFrom-Json; $newVersion = $newPkg.version } catch {}",
         "  Write-Host (\"RUNNER_VERSION_CHECK:\" + (ConvertTo-Json -Compress @{ action='updated'; previousVersion=$installedVersion; expectedVersion='" + imageVersion + "'; expectedMinor='" + expectedMinor + "'; newVersion=$newVersion; channel='" + channel + "'; sandboxId='" + sandboxId + "' }))",
         "  Write-Host 'Runner install complete'",
         "}"

package/agent/lib/sandbox.js

@@ -303,7 +303,16 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
       };
       this._fileSubscription = await this._sessionChannel.subscribe("file", this._onFileMsg);
 
-      this.heartbeat = setInterval(function () { }, 5000);
+      // Idle keepalive: publish a control message every 30s so the API renews the
+      // Redis lease (LEASE_RENEWAL_WINDOW_MS = 2min). Without this, an idle session
+      // (no commands) lets the lease lapse and the sandbox is reaped mid-test on
+      // both Linux (E2B) and Windows (EC2).
+      this.heartbeat = setInterval(function () {
+        if (!self._sessionChannel || !self._ably || self._ably.connection.state !== "connected") return;
+        self._sessionChannel.publish("control", { type: "keepalive" }).catch(function (err) {
+          logger.debug("[realtime] keepalive publish failed (non-fatal): " + (err.message || err));
+        });
+      }, 30000);
       if (this.heartbeat.unref) this.heartbeat.unref();
 
       // ─── Periodic stats logging ────────────────────────────────────────
@@ -590,6 +599,8 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
         body.resolution = message.resolution;
         body.ci = message.ci;
         if (message.instanceId) body.instanceId = message.instanceId;
+        // Reuse the same slot on re-auth so the old Ably channel isn't orphaned.
+        if (message.sandboxId) body.sandboxId = message.sandboxId;
       }
 
       var reply = await this._httpPostWithConcurrencyRetry(

package/examples/idle-reconnect.test.mjs

@@ -0,0 +1,42 @@
+/**
+ * TestDriver SDK - Idle 10 Minute Test (Vitest)
+ *
+ * Spawns a VM, opens testdriver.ai, waits 10 minutes, then asserts that
+ * testdriver.ai is still visible on screen. Verifies the sandbox and session
+ * stay alive during a long idle period.
+ *
+ * Runs on both Linux and Windows acceptance suites (Windows via TD_OS=windows).
+ */
+
+import { describe, expect, it } from "vitest";
+import { TestDriver } from "../lib/vitest/hooks.mjs";
+import { getDefaults } from "./config.mjs";
+
+const DELAY = 10 * 60 * 1000;
+
+describe("Idle 10 Minute Test", () => {
+  it(
+    "should keep testdriver.ai visible after idling for 10 minutes",
+    { timeout: DELAY + 5 * 60 * 1000 },
+    async (context) => {
+      const testdriver = TestDriver(context, {
+        ...getDefaults(context),
+      });
+
+      // Spawn the VM and open testdriver.ai
+      await testdriver.provision.chrome({
+        url: "https://testdriver.ai",
+      });
+
+      // Idle for 10 minutes
+      await testdriver.wait(DELAY);
+
+      // Assert testdriver.ai is still visible on screen
+      const result = await testdriver.assert(
+        "the testdriver.ai website is visible on the screen",
+      );
+
+      expect(result).toBeTruthy();
+    },
+  );
+});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@testdriverai/agent",
-  "version": "7.11.85-test",
+  "version": "7.11.87-canary",
   "description": "Next generation autonomous AI agent for end-to-end testing of web & desktop",
   "main": "sdk.js",
   "types": "sdk.d.ts",