@agtlantis/core 0.5.0 → 0.6.0

package/dist/execution/testing/host-configs.js

@@ -0,0 +1,154 @@
+/**
+ * Configuration objects for ExecutionHost contract tests.
+ * Abstracts the differences between SimpleHost and StreamingHost
+ * to enable shared contract testing.
+ */
+import { vi } from 'vitest';
+import { SimpleExecutionHost } from '../simple-host.js';
+import { StreamingExecutionHost } from '../streaming-host.js';
+import { SimpleSession } from '../../session/simple-session.js';
+import { StreamingSession } from '../../session/streaming-session.js';
+import { TEST_PROVIDER_TYPE, createMockModel, createMockFileManager, } from './fixtures.js';
+// ============================================================================
+// SimpleHost Configuration
+// ============================================================================
+export const simpleHostConfig = {
+    name: 'SimpleExecutionHost',
+    createHost(factory, workload, signal) {
+        return new SimpleExecutionHost(factory, workload, signal);
+    },
+    createSessionFactory(logger) {
+        return (signal) => new SimpleSession({
+            defaultLanguageModel: createMockModel(),
+            providerType: TEST_PROVIDER_TYPE,
+            fileManager: createMockFileManager(),
+            signal,
+            logger,
+        });
+    },
+    createSessionFactorySpy() {
+        let passedSignal;
+        const factory = vi.fn().mockImplementation((signal) => {
+            passedSignal = signal;
+            return new SimpleSession({
+                defaultLanguageModel: createMockModel(),
+                providerType: TEST_PROVIDER_TYPE,
+                fileManager: createMockFileManager(),
+                signal,
+            });
+        });
+        return {
+            factory,
+            getPassedSignal: () => passedSignal,
+        };
+    },
+    createSuccessWorkload(result) {
+        return vi.fn().mockResolvedValue(result);
+    },
+    createErrorWorkload(error) {
+        return vi.fn().mockRejectedValue(error);
+    },
+    createCancelableWorkload(abortScenario, onCancel) {
+        return async () => {
+            await new Promise((_, reject) => {
+                const signal = abortScenario.signal;
+                if (signal.aborted) {
+                    onCancel?.();
+                    reject(new DOMException('Aborted', 'AbortError'));
+                    return;
+                }
+                signal.addEventListener('abort', () => {
+                    onCancel?.();
+                    reject(new DOMException('Aborted', 'AbortError'));
+                });
+            });
+            return 'should-not-reach';
+        };
+    },
+    createHookWorkload(hook, options) {
+        return async (session) => {
+            session.onDone(hook);
+            if (options?.shouldFail) {
+                throw new Error('Intentional failure');
+            }
+            return 'result';
+        };
+    },
+};
+// ============================================================================
+// StreamingHost Configuration
+// ============================================================================
+export const streamingHostConfig = {
+    name: 'StreamingExecutionHost',
+    createHost(factory, workload, signal) {
+        return new StreamingExecutionHost(factory, workload, signal);
+    },
+    createSessionFactory(logger) {
+        return (signal) => new StreamingSession({
+            defaultLanguageModel: createMockModel(),
+            providerType: TEST_PROVIDER_TYPE,
+            fileManager: createMockFileManager(),
+            signal,
+            logger,
+        });
+    },
+    createSessionFactorySpy() {
+        let passedSignal;
+        const factory = vi.fn().mockImplementation((signal) => {
+            passedSignal = signal;
+            return new StreamingSession({
+                defaultLanguageModel: createMockModel(),
+                providerType: TEST_PROVIDER_TYPE,
+                fileManager: createMockFileManager(),
+                signal,
+            });
+        });
+        return {
+            factory,
+            getPassedSignal: () => passedSignal,
+        };
+    },
+    createSuccessWorkload(result) {
+        return async function* (session) {
+            return session.done(result);
+        };
+    },
+    createErrorWorkload(error) {
+        return async function* () {
+            throw error;
+        };
+    },
+    createCancelableWorkload(abortScenario, onCancel) {
+        return async function* (session) {
+            yield session.emit({ type: 'start' });
+            await new Promise((_, reject) => {
+                const signal = abortScenario.signal;
+                if (signal.aborted) {
+                    onCancel?.();
+                    reject(new DOMException('Aborted', 'AbortError'));
+                    return;
+                }
+                signal.addEventListener('abort', () => {
+                    onCancel?.();
+                    reject(new DOMException('Aborted', 'AbortError'));
+                });
+            });
+            return session.done('should-not-reach');
+        };
+    },
+    createHookWorkload(hook, options) {
+        return async function* (session) {
+            session.onDone(hook);
+            if (options?.shouldFail) {
+                throw new Error('Intentional failure');
+            }
+            return session.done('result');
+        };
+    },
+};
+// ============================================================================
+// Exports
+// ============================================================================
+/** All host configurations for parameterized testing */
+export const allHostConfigs = [simpleHostConfig, streamingHostConfig];
+//# sourceMappingURL=host-configs.js.map

package/dist/execution/testing/host-configs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"host-configs.js","sourceRoot":"","sources":["../../../src/execution/testing/host-configs.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,EAAE,gBAAgB,EAAE,MAAM,oCAAoC,CAAC;AACtE,OAAO,EACL,kBAAkB,EAElB,eAAe,EACf,qBAAqB,GACtB,MAAM,eAAe,CAAC;AAkEvB,+EAA+E;AAC/E,2BAA2B;AAC3B,+EAA+E;AAE/E,MAAM,CAAC,MAAM,gBAAgB,GAAoC;IAC/D,IAAI,EAAE,qBAAqB;IAE3B,UAAU,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAO;QACnC,OAAO,IAAI,mBAAmB,CAC5B,OAA+B,EAC/B,QAA0B,EAC1B,MAAM,CACP,CAAC;IACJ,CAAC;IAED,oBAAoB,CAAC,MAAe;QAClC,OAAO,CAAC,MAAoB,EAAE,EAAE,CAC9B,IAAI,aAAa,CAAC;YAChB,oBAAoB,EAAE,eAAe,EAAE;YACvC,YAAY,EAAE,kBAAkB;YAChC,WAAW,EAAE,qBAAqB,EAAE;YACpC,MAAM;YACN,MAAM;SACP,CAAC,CAAC;IACP,CAAC;IAED,uBAAuB;QACrB,IAAI,YAAqC,CAAC;QAC1C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,MAAoB,EAAE,EAAE;YAClE,YAAY,GAAG,MAAM,CAAC;YACtB,OAAO,IAAI,aAAa,CAAC;gBACvB,oBAAoB,EAAE,eAAe,EAAE;gBACvC,YAAY,EAAE,kBAAkB;gBAChC,WAAW,EAAE,qBAAqB,EAAE;gBACpC,MAAM;aACP,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO;YACL,OAAO;YACP,eAAe,EAAE,GAAG,EAAE,CAAC,YAAY;SACpC,CAAC;IACJ,CAAC;IAED,qBAAqB,CAAC,MAAc;QAClC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAC3C,CAAC;IAED,mBAAmB,CAAC,KAAY;QAC9B,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAC1C,CAAC;IAED,wBAAwB,CAAC,aAA4B,EAAE,QAAqB;QAC1E,OAAO,KAAK,IAAI,EAAE;YAChB,MAAM,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gBACpC,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC;gBACpC,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnB,QAAQ,EAAE,EAAE,CAAC;oBACb,MAAM,CAAC,IAAI,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAC;oBAClD,OAAO;gBACT,CAAC;gBACD,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;oBACpC,QAAQ,EAAE,EAAE,CAAC;oBACb,MAAM,CAAC,IAAI,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAC;gBACpD,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YACH,OAAO,kBAAkB,CAAC;QAC5B,CAAC,CAAC;IACJ,CAAC;IAED,kBAAkB,CAAC,IAAgB,EAAE,OAAkC;QACrE,OAAO,KAAK,EAAE,OAAsB,EAAE,EAAE;YACtC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACrB,IAAI,OAAO,EAAE,UAAU,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzC,CAAC;YACD,OAAO,QAAQ,CAAC;QAClB,CAAC,CAAC;IACJ,CAAC;CACF,CAAC;AAEF,+EAA+E;AAC/E,8BAA8B;AAC9B,+EAA+E;AAE/E,MAAM,CAAC,MAAM,mBAAmB,GAAoC;IAClE,IAAI,EAAE,wBAAwB;IAE9B,UAAU,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAO;QACnC,OAAO,IAAI,sBAAsB,CAC/B,OAAkC,EAClC,QAA6B,EAC7B,MAAM,CACP,CAAC;IACJ,CAAC;IAED,oBAAoB,CAAC,MAAe;QAClC,OAAO,CAAC,MAAoB,EAAE,EAAE,CAC9B,IAAI,gBAAgB,CAAY;YAC9B,oBAAoB,EAAE,eAAe,EAAE;YACvC,YAAY,EAAE,kBAAkB;YAChC,WAAW,EAAE,qBAAqB,EAAE;YACpC,MAAM;YACN,MAAM;SACP,CAAC,CAAC;IACP,CAAC;IAED,uBAAuB;QACrB,IAAI,YAAqC,CAAC;QAC1C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,MAAoB,EAAE,EAAE;YAClE,YAAY,GAAG,MAAM,CAAC;YACtB,OAAO,IAAI,gBAAgB,CAAY;gBACrC,oBAAoB,EAAE,eAAe,EAAE;gBACvC,YAAY,EAAE,kBAAkB;gBAChC,WAAW,EAAE,qBAAqB,EAAE;gBACpC,MAAM;aACP,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO;YACL,OAAO;YACP,eAAe,EAAE,GAAG,EAAE,CAAC,YAAY;SACpC,CAAC;IACJ,CAAC;IAED,qBAAqB,CAAC,MAAc;QAClC,OAAO,KAAK,SAAS,CAAC,EAAE,OAAoC;YAC1D,OAAO,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC9B,CAAC,CAAC;IACJ,CAAC;IAED,mBAAmB,CAAC,KAAY;QAC9B,OAAO,KAAK,SAAS,CAAC;YACpB,MAAM,KAAK,CAAC;QACd,CAAC,CAAC;IACJ,CAAC;IAED,wBAAwB,CAAC,aAA4B,EAAE,QAAqB;QAC1E,OAAO,KAAK,SAAS,CAAC,EAAE,OAAoC;YAC1D,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;YAEtC,MAAM,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gBACpC,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC;gBACpC,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnB,QAAQ,EAAE,EAAE,CAAC;oBACb,MAAM,CAAC,IAAI,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAC;oBAClD,OAAO;gBACT,CAAC;gBACD,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;oBACpC,QAAQ,EAAE,EAAE,CAAC;oBACb,MAAM,CAAC,IAAI,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAC;gBACpD,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,OAAO,OAAO,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAC1C,CAAC,CAAC;IACJ,CAAC;IAED,kBAAkB,CAAC,IAAgB,EAAE,OAAkC;QACrE,OAAO,KAAK,SAAS,CAAC,EAAE,OAAoC;YAC1D,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACrB,IAAI,OAAO,EAAE,UAAU,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzC,CAAC;YACD,OAAO,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC,CAAC;IACJ,CAAC;CACF,CAAC;AAEF,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E,wDAAwD;AACxD,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,gBAAgB,EAAE,mBAAmB,CAAU,CAAC"}

package/dist/execution/testing/vitest-assertions.d.ts

@@ -0,0 +1,102 @@
+/**
+ * Vitest-specific assertion helpers for execution module tests.
+ * These helpers depend on vitest and should only be used in internal tests.
+ */
+import { vi } from 'vitest';
+import type { ExecutionResult, StreamingResult } from '../types.js';
+import type { Logger } from '../../observability/logger.js';
+export type MockLogger = Logger & {
+    onLLMCallStart: ReturnType<typeof vi.fn>;
+    onLLMCallEnd: ReturnType<typeof vi.fn>;
+    onExecutionStart: ReturnType<typeof vi.fn>;
+    onExecutionEmit: ReturnType<typeof vi.fn>;
+    onExecutionDone: ReturnType<typeof vi.fn>;
+    onExecutionError: ReturnType<typeof vi.fn>;
+};
+/**
+ * Assert that an execution result is successful and optionally check the value.
+ *
+ * @example
+ * ```typescript
+ * const result = await execution.result();
+ * expectSuccessResult(result, 'expected-value');
+ * // result is now narrowed to succeeded type
+ * console.log(result.value);
+ * ```
+ */
+export declare function expectSuccessResult<T>(result: ExecutionResult<T>, expectedValue?: T): asserts result is Extract<ExecutionResult<T>, {
+    status: 'succeeded';
+}>;
+/**
+ * Assert that an execution result is failed and optionally check the error.
+ *
+ * @example
+ * ```typescript
+ * const result = await execution.result();
+ * expectFailedResult(result, 'Expected error message');
+ * // or with regex
+ * expectFailedResult(result, /timeout/i);
+ * ```
+ */
+export declare function expectFailedResult<T>(result: ExecutionResult<T>, errorMatcher?: string | RegExp | Error): asserts result is Extract<ExecutionResult<T>, {
+    status: 'failed';
+}>;
+/**
+ * Assert that an execution result is canceled.
+ *
+ * @example
+ * ```typescript
+ * const result = await execution.result();
+ * expectCanceledResult(result);
+ * ```
+ */
+export declare function expectCanceledResult<T>(result: ExecutionResult<T>): asserts result is Extract<ExecutionResult<T>, {
+    status: 'canceled';
+}>;
+/**
+ * Assert streaming result status and optionally check events count.
+ *
+ * @example
+ * ```typescript
+ * const result = await execution.result();
+ * expectStreamingSuccessResult(result, 'value', 3);
+ * ```
+ */
+export declare function expectStreamingSuccessResult<TEvent, T>(result: StreamingResult<TEvent, T>, expectedValue?: T, expectedEventCount?: number): asserts result is Extract<StreamingResult<TEvent, T>, {
+    status: 'succeeded';
+}>;
+type LoggerEventType = 'start' | 'emit' | 'done' | 'error';
+/**
+ * Verify that logger was called in the expected sequence.
+ *
+ * @example
+ * ```typescript
+ * const logger = createMockLogger();
+ * // ... run execution ...
+ *
+ * verifyLoggerSequence(logger, ['start', 'emit', 'emit', 'done']);
+ * ```
+ */
+export declare function verifyLoggerSequence(logger: MockLogger, expectedSequence: LoggerEventType[]): void;
+/**
+ * Create a vitest mock logger.
+ */
+export declare function createVitestMockLogger(): MockLogger;
+/**
+ * @deprecated Use createVitestMockLogger instead
+ */
+export declare const createMockLogger: typeof createVitestMockLogger;
+/**
+ * Get vitest's vi.fn for use with framework-agnostic fixtures.
+ *
+ * @example
+ * ```typescript
+ * import { vitestMockFn } from './vitest-assertions';
+ * import { createStreamingSessionFactory } from './fixtures';
+ *
+ * const factory = createStreamingSessionFactory({ mockFn: vitestMockFn });
+ * ```
+ */
+export declare const vitestMockFn: () => (...args: unknown[]) => unknown;
+export {};
+//# sourceMappingURL=vitest-assertions.d.ts.map

package/dist/execution/testing/vitest-assertions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vitest-assertions.d.ts","sourceRoot":"","sources":["../../../src/execution/testing/vitest-assertions.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,EAAE,EAAU,MAAM,QAAQ,CAAC;AACpC,OAAO,KAAK,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AACpE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,+BAA+B,CAAC;AAM5D,MAAM,MAAM,UAAU,GAAG,MAAM,GAAG;IAChC,cAAc,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IACzC,YAAY,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IACvC,gBAAgB,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAC3C,eAAe,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAC1C,eAAe,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAC1C,gBAAgB,EAAE,UAAU,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;CAC5C,CAAC;AAMF;;;;;;;;;;GAUG;AACH,wBAAgB,mBAAmB,CAAC,CAAC,EACnC,MAAM,EAAE,eAAe,CAAC,CAAC,CAAC,EAC1B,aAAa,CAAC,EAAE,CAAC,GAChB,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE;IAAE,MAAM,EAAE,WAAW,CAAA;CAAE,CAAC,CAQxE;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,kBAAkB,CAAC,CAAC,EAClC,MAAM,EAAE,eAAe,CAAC,CAAC,CAAC,EAC1B,YAAY,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,KAAK,GACrC,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE;IAAE,MAAM,EAAE,QAAQ,CAAA;CAAE,CAAC,CAcrE;AAED;;;;;;;;GAQG;AACH,wBAAgB,oBAAoB,CAAC,CAAC,EACpC,MAAM,EAAE,eAAe,CAAC,CAAC,CAAC,GACzB,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE;IAAE,MAAM,EAAE,UAAU,CAAA;CAAE,CAAC,CAEvE;AAED;;;;;;;;GAQG;AACH,wBAAgB,4BAA4B,CAAC,MAAM,EAAE,CAAC,EACpD,MAAM,EAAE,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,EAClC,aAAa,CAAC,EAAE,CAAC,EACjB,kBAAkB,CAAC,EAAE,MAAM,GAC1B,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE;IAAE,MAAM,EAAE,WAAW,CAAA;CAAE,CAAC,CAKhF;AAMD,KAAK,eAAe,GAAG,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;AAE3D;;;;;;;;;;GAUG;AACH,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,UAAU,EAClB,gBAAgB,EAAE,eAAe,EAAE,GAClC,IAAI,CAsBN;AAED;;GAEG;AACH,wBAAgB,sBAAsB,IAAI,UAAU,CASnD;AAED;;GAEG;AACH,eAAO,MAAM,gBAAgB,+BAAyB,CAAC;AAEvD;;;;;;;;;;GAUG;AACH,eAAO,MAAM,YAAY,EAAE,MAAM,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAuB,CAAC"}

package/dist/execution/testing/vitest-assertions.js

@@ -0,0 +1,142 @@
+/**
+ * Vitest-specific assertion helpers for execution module tests.
+ * These helpers depend on vitest and should only be used in internal tests.
+ */
+import { vi, expect } from 'vitest';
+// ============================================================================
+// Result Assertion Helpers
+// ============================================================================
+/**
+ * Assert that an execution result is successful and optionally check the value.
+ *
+ * @example
+ * ```typescript
+ * const result = await execution.result();
+ * expectSuccessResult(result, 'expected-value');
+ * // result is now narrowed to succeeded type
+ * console.log(result.value);
+ * ```
+ */
+export function expectSuccessResult(result, expectedValue) {
+    expect(result.status).toBe('succeeded');
+    if (result.status !== 'succeeded') {
+        throw new Error('Result is not succeeded');
+    }
+    if (expectedValue !== undefined) {
+        expect(result.value).toBe(expectedValue);
+    }
+}
+/**
+ * Assert that an execution result is failed and optionally check the error.
+ *
+ * @example
+ * ```typescript
+ * const result = await execution.result();
+ * expectFailedResult(result, 'Expected error message');
+ * // or with regex
+ * expectFailedResult(result, /timeout/i);
+ * ```
+ */
+export function expectFailedResult(result, errorMatcher) {
+    expect(result.status).toBe('failed');
+    if (result.status !== 'failed') {
+        throw new Error('Result is not failed');
+    }
+    if (errorMatcher !== undefined) {
+        if (typeof errorMatcher === 'string') {
+            expect(result.error.message).toBe(errorMatcher);
+        }
+        else if (errorMatcher instanceof RegExp) {
+            expect(result.error.message).toMatch(errorMatcher);
+        }
+        else {
+            expect(result.error).toBe(errorMatcher);
+        }
+    }
+}
+/**
+ * Assert that an execution result is canceled.
+ *
+ * @example
+ * ```typescript
+ * const result = await execution.result();
+ * expectCanceledResult(result);
+ * ```
+ */
+export function expectCanceledResult(result) {
+    expect(result.status).toBe('canceled');
+}
+/**
+ * Assert streaming result status and optionally check events count.
+ *
+ * @example
+ * ```typescript
+ * const result = await execution.result();
+ * expectStreamingSuccessResult(result, 'value', 3);
+ * ```
+ */
+export function expectStreamingSuccessResult(result, expectedValue, expectedEventCount) {
+    expectSuccessResult(result, expectedValue);
+    if (expectedEventCount !== undefined) {
+        expect(result.events).toHaveLength(expectedEventCount);
+    }
+}
+/**
+ * Verify that logger was called in the expected sequence.
+ *
+ * @example
+ * ```typescript
+ * const logger = createMockLogger();
+ * // ... run execution ...
+ *
+ * verifyLoggerSequence(logger, ['start', 'emit', 'emit', 'done']);
+ * ```
+ */
+export function verifyLoggerSequence(logger, expectedSequence) {
+    const eventMap = {
+        start: 'onExecutionStart',
+        emit: 'onExecutionEmit',
+        done: 'onExecutionDone',
+        error: 'onExecutionError',
+    };
+    // Count expected occurrences
+    const expectedCounts = expectedSequence.reduce((acc, event) => {
+        acc[event] = (acc[event] || 0) + 1;
+        return acc;
+    }, {});
+    // Verify each event type was called the expected number of times
+    for (const [event, count] of Object.entries(expectedCounts)) {
+        const methodName = eventMap[event];
+        expect(logger[methodName]).toHaveBeenCalledTimes(count);
+    }
+}
+/**
+ * Create a vitest mock logger.
+ */
+export function createVitestMockLogger() {
+    return {
+        onLLMCallStart: vi.fn(),
+        onLLMCallEnd: vi.fn(),
+        onExecutionStart: vi.fn(),
+        onExecutionEmit: vi.fn(),
+        onExecutionDone: vi.fn(),
+        onExecutionError: vi.fn(),
+    };
+}
+/**
+ * @deprecated Use createVitestMockLogger instead
+ */
+export const createMockLogger = createVitestMockLogger;
+/**
+ * Get vitest's vi.fn for use with framework-agnostic fixtures.
+ *
+ * @example
+ * ```typescript
+ * import { vitestMockFn } from './vitest-assertions';
+ * import { createStreamingSessionFactory } from './fixtures';
+ *
+ * const factory = createStreamingSessionFactory({ mockFn: vitestMockFn });
+ * ```
+ */
+export const vitestMockFn = () => vi.fn();
+//# sourceMappingURL=vitest-assertions.js.map

package/dist/execution/testing/vitest-assertions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vitest-assertions.js","sourceRoot":"","sources":["../../../src/execution/testing/vitest-assertions.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAiBpC,+EAA+E;AAC/E,2BAA2B;AAC3B,+EAA+E;AAE/E;;;;;;;;;;GAUG;AACH,MAAM,UAAU,mBAAmB,CACjC,MAA0B,EAC1B,aAAiB;IAEjB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACxC,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;IACD,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC3C,CAAC;AACH,CAAC;AAED;;;;;;;;;;GAUG;AACH,MAAM,UAAU,kBAAkB,CAChC,MAA0B,EAC1B,YAAsC;IAEtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACrC,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC1C,CAAC;IACD,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;QAC/B,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAClD,CAAC;aAAM,IAAI,YAAY,YAAY,MAAM,EAAE,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QACrD,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;AACH,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,oBAAoB,CAClC,MAA0B;IAE1B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;AACzC,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,4BAA4B,CAC1C,MAAkC,EAClC,aAAiB,EACjB,kBAA2B;IAE3B,mBAAmB,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;IAC3C,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC;IACzD,CAAC;AACH,CAAC;AAQD;;;;;;;;;;GAUG;AACH,MAAM,UAAU,oBAAoB,CAClC,MAAkB,EAClB,gBAAmC;IAEnC,MAAM,QAAQ,GAA8C;QAC1D,KAAK,EAAE,kBAAkB;QACzB,IAAI,EAAE,iBAAiB;QACvB,IAAI,EAAE,iBAAiB;QACvB,KAAK,EAAE,kBAAkB;KAC1B,CAAC;IAEF,6BAA6B;IAC7B,MAAM,cAAc,GAAG,gBAAgB,CAAC,MAAM,CAC5C,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;QACb,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACnC,OAAO,GAAG,CAAC;IACb,CAAC,EACD,EAAqC,CACtC,CAAC;IAEF,iEAAiE;IACjE,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QAC5D,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAwB,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,sBAAsB;IACpC,OAAO;QACL,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;QACvB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;QACrB,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;QACzB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;QACxB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;QACxB,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;KACZ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,sBAAsB,CAAC;AAEvD;;;;;;;;;;GAUG;AACH,MAAM,CAAC,MAAM,YAAY,GAA0C,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC"}