@agentforge/testing 0.16.20 → 0.16.22

package/dist/index.cjs

@@ -12808,14 +12808,14 @@ function withTimeout(fn2, timeout, isHook = false) {
   if (timeout <= 0 || timeout === Number.POSITIVE_INFINITY) {
     return fn2;
   }
-  const { setTimeout: setTimeout2, clearTimeout } = getSafeTimers();
+  const { setTimeout: setTimeout2, clearTimeout: clearTimeout2 } = getSafeTimers();
   return function runWithTimeout(...args) {
     return Promise.race([
       fn2(...args),
       new Promise((resolve4, reject) => {
         var _a;
         const timer = setTimeout2(() => {
-          clearTimeout(timer);
+          clearTimeout2(timer);
           reject(new Error(makeTimeoutMsg(isHook, timeout)));
         }, timeout);
         (_a = timer.unref) == null ? void 0 : _a.call(timer);
@@ -15673,9 +15673,9 @@ function createExpectPoll(expect2) {
           const promise = () => new Promise((resolve4, reject) => {
             let intervalId;
             let lastError;
-            const { setTimeout: setTimeout2, clearTimeout } = getSafeTimers();
+            const { setTimeout: setTimeout2, clearTimeout: clearTimeout2 } = getSafeTimers();
             const timeoutId = setTimeout2(() => {
-              clearTimeout(intervalId);
+              clearTimeout2(intervalId);
               reject(
                 copyStackTrace$1(
                   new Error(`Matcher did not succeed in ${timeout}ms`, {
@@ -15691,8 +15691,8 @@ function createExpectPoll(expect2) {
                 const obj = await fn2();
                 utils_exports.flag(assertion, "object", obj);
                 resolve4(await assertionFunction.call(assertion, ...args));
-                clearTimeout(intervalId);
-                clearTimeout(timeoutId);
+                clearTimeout2(intervalId);
+                clearTimeout2(timeoutId);
               } catch (err) {
                 lastError = err;
                 intervalId = setTimeout2(check, interval);
@@ -17283,7 +17283,7 @@ To automatically clean-up native timers, use \`shouldClearNativeTimers\`.`
           });
         };
       }
-      clock.clearTimeout = function clearTimeout(timerId) {
+      clock.clearTimeout = function clearTimeout2(timerId) {
         return clearTimer(clock, timerId, "Timeout");
       };
       clock.nextTick = function nextTick(func) {
@@ -17926,7 +17926,7 @@ function copyStackTrace(target, source) {
   return target;
 }
 function waitFor(callback, options = {}) {
-  const { setTimeout: setTimeout2, setInterval, clearTimeout, clearInterval } = getSafeTimers();
+  const { setTimeout: setTimeout2, setInterval, clearTimeout: clearTimeout2, clearInterval } = getSafeTimers();
   const { interval = 50, timeout = 1e3 } = typeof options === "number" ? { timeout: options } : options;
   const STACK_TRACE_ERROR = new Error("STACK_TRACE_ERROR");
   return new Promise((resolve4, reject) => {
@@ -17936,7 +17936,7 @@ function waitFor(callback, options = {}) {
     let intervalId;
     const onResolve = (result) => {
       if (timeoutId) {
-        clearTimeout(timeoutId);
+        clearTimeout2(timeoutId);
       }
       if (intervalId) {
         clearInterval(intervalId);
@@ -17994,7 +17994,7 @@ function waitFor(callback, options = {}) {
   });
 }
 function waitUntil(callback, options = {}) {
-  const { setTimeout: setTimeout2, setInterval, clearTimeout, clearInterval } = getSafeTimers();
+  const { setTimeout: setTimeout2, setInterval, clearTimeout: clearTimeout2, clearInterval } = getSafeTimers();
   const { interval = 50, timeout = 1e3 } = typeof options === "number" ? { timeout: options } : options;
   const STACK_TRACE_ERROR = new Error("STACK_TRACE_ERROR");
   return new Promise((resolve4, reject) => {
@@ -18018,7 +18018,7 @@ function waitUntil(callback, options = {}) {
         return;
       }
       if (timeoutId) {
-        clearTimeout(timeoutId);
+        clearTimeout2(timeoutId);
       }
       if (intervalId) {
         clearInterval(intervalId);
@@ -18604,20 +18604,21 @@ var AgentTestRunner = class {
     let messages = [];
     let passed = true;
     let error;
+    let timeoutId;
     try {
-      const timeout = this.config.timeout || 3e4;
+      const timeout = this.config.timeout ?? 3e4;
       const timeoutPromise = new Promise((_, reject) => {
-        setTimeout(() => reject(new Error("Agent test timeout")), timeout);
+        timeoutId = setTimeout(() => reject(new Error("Agent test timeout")), timeout);
       });
       const runPromise = (async () => {
         if (this.config.captureSteps) {
           const result = await this.agent.invoke(input);
           finalState = result;
-          messages = result.messages || [];
+          messages = extractMessages(result);
         } else {
           const result = await this.agent.invoke(input);
           finalState = result;
-          messages = result.messages || [];
+          messages = extractMessages(result);
         }
         if (this.config.validateState && this.config.stateValidator) {
           const isValid = await this.config.stateValidator(finalState);
@@ -18626,7 +18627,13 @@ var AgentTestRunner = class {
           }
         }
       })();
-      await Promise.race([runPromise, timeoutPromise]);
+      try {
+        await Promise.race([runPromise, timeoutPromise]);
+      } finally {
+        if (timeoutId !== void 0) {
+          clearTimeout(timeoutId);
+        }
+      }
     } catch (err) {
       passed = false;
       error = err;
@@ -18651,6 +18658,13 @@ var AgentTestRunner = class {
 function createAgentTestRunner(agent, config2) {
   return new AgentTestRunner(agent, config2);
 }
+function extractMessages(state) {
+  if (typeof state !== "object" || state === null) {
+    return [];
+  }
+  const { messages } = state;
+  return Array.isArray(messages) ? messages : [];
+}
 var ConversationSimulator = class {
   constructor(agent, config2 = {}) {
     this.agent = agent;