@trigger.dev/sdk 4.4.6 → 4.5.0-rc.1

package/dist/commonjs/v3/test/mock-chat-agent.js

@@ -0,0 +1,468 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.mockChatAgent = mockChatAgent;
+const v3_1 = require("@trigger.dev/core/v3");
+const test_1 = require("@trigger.dev/core/v3/test");
+const sessions_js_1 = require("../sessions.js");
+const ai_js_1 = require("../ai.js");
+const test_session_handle_js_1 = require("./test-session-handle.js");
+const CONTROL_CHUNK_TYPES = new Set([
+    "trigger:turn-complete",
+    "trigger:upgrade-required",
+]);
+function isControlChunk(chunk) {
+    if (typeof chunk !== "object" || chunk === null)
+        return false;
+    const type = chunk.type;
+    return typeof type === "string" && CONTROL_CHUNK_TYPES.has(type);
+}
+/**
+ * Create an offline test harness for a `chat.agent` task.
+ *
+ * The harness starts the agent's `run()` function in a mocked task context,
+ * waits in preload for the first message, then exposes driver methods for
+ * sending messages / actions / stop signals and awaiting turn completion.
+ *
+ * Users are responsible for mocking the language model themselves — use
+ * `MockLanguageModelV3` and `simulateReadableStream` from `ai/test` inside
+ * their agent's `run()` function (typically via DI through `clientData`).
+ *
+ * @example
+ * ```ts
+ * import { mockChatAgent } from "@trigger.dev/sdk/ai/test";
+ * import { MockLanguageModelV3, simulateReadableStream } from "ai/test";
+ * import { myAgent } from "./my-agent";
+ *
+ * test("says hello", async () => {
+ *   const harness = mockChatAgent(myAgent, { chatId: "test-1" });
+ *   try {
+ *     const turn = await harness.sendMessage({
+ *       id: "m1",
+ *       role: "user",
+ *       parts: [{ type: "text", text: "hi" }],
+ *     });
+ *     expect(turn.chunks).toContainEqual(
+ *       expect.objectContaining({ type: "text-delta", delta: "hello" })
+ *     );
+ *   } finally {
+ *     await harness.close();
+ *   }
+ * });
+ * ```
+ */
+function mockChatAgent(agent, options = {}) {
+    const chatId = options.chatId ?? "test-chat";
+    // The agent opens the session with `payload.sessionId ?? payload.chatId`.
+    // We pass no sessionId, so it falls back to chatId.
+    const sessionId = chatId;
+    // `continuation: true` without an explicit mode auto-selects "continuation"
+    // — the canonical shape for a continuation-run boot.
+    const mode = options.mode ??
+        (options.continuation === true
+            ? "continuation"
+            : options.preload === false
+                ? "submit-message"
+                : "preload");
+    const clientData = options.clientData;
+    const taskEntry = v3_1.resourceCatalog.getTask(agent.id);
+    if (!taskEntry) {
+        throw new Error(`mockChatAgent: no task registered with id "${agent.id}". ` +
+            `Import "@trigger.dev/sdk/ai/test" before your agent module so tasks register correctly.`);
+    }
+    const runFn = taskEntry.fns.run;
+    // Session .out state: chunks + listener registry. Shared between the
+    // harness and the TestSessionOutputChannel installed via the open-override.
+    const sessionOutState = {
+        chunks: [],
+        listeners: new Set(),
+    };
+    // Buffers that survive across harness method calls
+    const allRawChunks = [];
+    const allChunks = [];
+    // Promise that resolves when the background task run() function returns.
+    let taskFinished;
+    let sendSessionInput;
+    let closeSessionInput;
+    let runSignal;
+    // A latch that resolves every time `trigger:turn-complete` appears on the chat stream.
+    // We use a shared pending promise and replace it after each completion.
+    let turnCompleteResolvers = [];
+    const waitForTurnComplete = () => new Promise((resolve) => {
+        turnCompleteResolvers.push(resolve);
+    });
+    // Signal that the caller is ready to observe output
+    let harnessReadyResolve;
+    const harnessReady = new Promise((resolve) => {
+        harnessReadyResolve = resolve;
+    });
+    // ── Snapshot read/write override state ───────────────────────────────
+    // The runtime's snapshot read returns whatever `seededSnapshot` is at
+    // boot time. The runtime's snapshot write captures into
+    // `lastWrittenSnapshot` for harness consumers to assert via
+    // `getSnapshot()`. Installed below alongside the session overrides;
+    // cleared on close in the same finally block.
+    let seededSnapshot = options.snapshot;
+    let lastWrittenSnapshot;
+    let seededReplayChunks = [];
+    let seededReplayPartial;
+    let seededSessionInMessages = [];
+    (0, ai_js_1.__setReadChatSnapshotImplForTests)((_id) => {
+        return seededSnapshot;
+    });
+    (0, ai_js_1.__setWriteChatSnapshotImplForTests)((_id, snapshot) => {
+        lastWrittenSnapshot = snapshot;
+    });
+    // Replay override: install a default that returns whatever
+    // `seededReplayChunks` reduces to. `mockChatAgent` doesn't model the
+    // settled-vs-partial split — seeded chunks always reduce to the
+    // `settled` array with `partial: undefined`. Recovery-specific
+    // tests can install their own override to seed a partial.
+    // Cleared in the same `finally` block as the other test overrides.
+    (0, ai_js_1.__setReplaySessionOutTailImplForTests)(async () => {
+        const settled = seededReplayChunks.length === 0
+            ? []
+            : (await reduceChunksToMessages(seededReplayChunks));
+        // For the mock harness, `partialRaw` is the same as `partial` — we
+        // don't model cleanupAbortedParts separately. Recovery tests that
+        // need a partialRaw distinct from partial install their own stub.
+        return {
+            settled,
+            partial: seededReplayPartial,
+            partialRaw: seededReplayPartial,
+        };
+    });
+    // session.in tail override: each seeded UIMessage becomes a
+    // { message, metadata: undefined, seqNum: i+1 } entry. Mirrors the
+    // seq-num pattern from the out-tail stub so cursor-advance logic is
+    // exercised correctly. `metadata` is `undefined` for seeded users —
+    // the boot path falls back to `payload.metadata` for those.
+    (0, ai_js_1.__setReplaySessionInTailImplForTests)(async () => {
+        return seededSessionInMessages.map((message, i) => ({
+            message,
+            metadata: undefined,
+            seqNum: i + 1,
+        }));
+    });
+    // Install the session open override so `sessions.open(id)` returns a
+    // SessionHandle with an in-memory `.out` that captures writes. The
+    // `.in` channel routes record subscriptions (`on`/`once`/`peek`)
+    // through the `sessionStreams` global — the mock task context
+    // installs a `TestSessionStreamManager` there — and stubs `wait()`
+    // so the suspend path resolves cleanly on `runSignal.abort()` without
+    // touching the api client.
+    (0, sessions_js_1.__setSessionOpenImplForTests)((id) => (0, test_session_handle_js_1.createTestSessionHandle)(id, sessionOutState, () => runSignal?.signal));
+    // Install the session start override so any test path that invokes
+    // `sessions.start()` (typically through a server action shim like
+    // `chat.createStartSessionAction`) becomes a no-op fixture instead of
+    // hitting a real API. Most chat.agent tests trigger the run directly
+    // via `sendPayloadAndWait` and never go through this path, but the
+    // stub keeps the API safe to call from inside tested code.
+    (0, sessions_js_1.__setSessionStartImplForTests)((body) => {
+        if (process.env.TRIGGER_CHAT_TEST_DEBUG === "1") {
+            console.log("[mockChatAgent] sessions.start override:", body);
+        }
+        const fakeRunId = `run_test_${body.externalId ?? "anon"}`;
+        return {
+            id: `session_test_${body.externalId ?? "anon"}`,
+            externalId: body.externalId ?? null,
+            type: body.type,
+            taskIdentifier: body.taskIdentifier,
+            triggerConfig: body.triggerConfig,
+            currentRunId: fakeRunId,
+            runId: fakeRunId,
+            publicAccessToken: "tr_test_session_pat",
+            tags: body.tags ?? [],
+            metadata: (body.metadata ?? null),
+            closedAt: null,
+            closedReason: null,
+            expiresAt: null,
+            createdAt: new Date(0),
+            updatedAt: new Date(0),
+            isCached: false,
+        };
+    });
+    taskFinished = (0, test_1.runInMockTaskContext)(async (drivers) => {
+        runSignal = new AbortController();
+        // For `mode: "continuation"`, omit `trigger` from the wire payload —
+        // mirrors what the server's `ensureRunForSession` / `swapSessionRun`
+        // produces (the continuation overrides clear `trigger` so the SDK
+        // boot path falls into the continuation-wait branch instead of
+        // re-firing the basePayload's stale first-run trigger). `continuation:
+        // true` is set unconditionally for this mode so the boot path's
+        // continuation-wait condition matches.
+        const isContinuationMode = mode === "continuation";
+        const initialPayload = {
+            chatId,
+            ...(isContinuationMode
+                ? { trigger: undefined, continuation: true }
+                : { trigger: mode }),
+            metadata: clientData,
+            ...(!isContinuationMode && options.continuation ? { continuation: true } : {}),
+            ...(options.previousRunId ? { previousRunId: options.previousRunId } : {}),
+        };
+        sendSessionInput = drivers.sessions.in.send;
+        closeSessionInput = drivers.sessions.in.close;
+        // Record every chunk written to session.out, detect turn-complete.
+        const listener = (chunk) => {
+            allRawChunks.push(chunk);
+            if (!isControlChunk(chunk)) {
+                allChunks.push(chunk);
+            }
+            if (typeof chunk === "object" &&
+                chunk !== null &&
+                chunk.type === "trigger:turn-complete") {
+                const resolvers = turnCompleteResolvers;
+                turnCompleteResolvers = [];
+                for (const resolve of resolvers)
+                    resolve();
+            }
+        };
+        sessionOutState.listeners.add(listener);
+        const unsubscribe = () => sessionOutState.listeners.delete(listener);
+        if (options.setupLocals) {
+            await options.setupLocals({ set: drivers.locals.set });
+        }
+        harnessReadyResolve();
+        try {
+            if (process.env.TRIGGER_CHAT_TEST_DEBUG === "1") {
+                console.log("[mockChatAgent] Starting runFn with payload:", initialPayload);
+            }
+            await runFn(initialPayload, {
+                ctx: drivers.ctx,
+                signal: runSignal.signal,
+            });
+            if (process.env.TRIGGER_CHAT_TEST_DEBUG === "1") {
+                console.log("[mockChatAgent] runFn returned");
+            }
+        }
+        catch (err) {
+            if (process.env.TRIGGER_CHAT_TEST_DEBUG === "1") {
+                console.log("[mockChatAgent] runFn threw:", err);
+            }
+            throw err;
+        }
+        finally {
+            unsubscribe();
+            // Resolve any outstanding turn-complete waiters so callers don't hang
+            const resolvers = turnCompleteResolvers;
+            turnCompleteResolvers = [];
+            for (const resolve of resolvers)
+                resolve();
+        }
+    }, options.taskContext)
+        .catch((err) => {
+        // Propagate errors to pending turn waiters instead of dropping them
+        const resolvers = turnCompleteResolvers;
+        turnCompleteResolvers = [];
+        for (const resolve of resolvers)
+            resolve();
+        throw err;
+    })
+        .finally(() => {
+        // Always clear the test overrides, even if the task threw.
+        (0, sessions_js_1.__setSessionOpenImplForTests)(undefined);
+        (0, sessions_js_1.__setSessionStartImplForTests)(undefined);
+        (0, ai_js_1.__setReadChatSnapshotImplForTests)(undefined);
+        (0, ai_js_1.__setWriteChatSnapshotImplForTests)(undefined);
+        (0, ai_js_1.__setReplaySessionOutTailImplForTests)(undefined);
+        (0, ai_js_1.__setReplaySessionInTailImplForTests)(undefined);
+    });
+    const sendPayloadAndWait = async (payload) => {
+        await harnessReady;
+        const before = allRawChunks.length;
+        const turnComplete = waitForTurnComplete();
+        await sendSessionInput(sessionId, { kind: "message", payload });
+        await turnComplete;
+        const rawChunks = allRawChunks.slice(before);
+        const chunks = rawChunks.filter((c) => !isControlChunk(c));
+        return { chunks, rawChunks };
+    };
+    const harness = {
+        chatId,
+        async sendMessage(message) {
+            return sendPayloadAndWait({
+                message,
+                chatId,
+                trigger: "submit-message",
+                metadata: clientData,
+            });
+        },
+        async sendRegenerate() {
+            return sendPayloadAndWait({
+                chatId,
+                trigger: "regenerate-message",
+                metadata: clientData,
+            });
+        },
+        async sendHeadStart({ messages }) {
+            return sendPayloadAndWait({
+                headStartMessages: messages,
+                chatId,
+                trigger: "handover-prepare",
+                metadata: clientData,
+            });
+        },
+        async sendAction(action) {
+            return sendPayloadAndWait({
+                chatId,
+                trigger: "action",
+                action,
+                metadata: clientData,
+            });
+        },
+        async sendStop(message) {
+            await harnessReady;
+            await sendSessionInput(sessionId, { kind: "stop", message });
+        },
+        async sendHandover(args) {
+            await harnessReady;
+            const before = allRawChunks.length;
+            const turnComplete = waitForTurnComplete();
+            await sendSessionInput(sessionId, {
+                kind: "handover",
+                partialAssistantMessage: args.partialAssistantMessage,
+                messageId: args.messageId,
+                isFinal: args.isFinal ?? false,
+            });
+            await turnComplete;
+            const rawChunks = allRawChunks.slice(before);
+            const chunks = rawChunks.filter((c) => !isControlChunk(c));
+            return { chunks, rawChunks };
+        },
+        async sendHandoverSkip() {
+            await harnessReady;
+            // No turn-complete on skip — the agent exits without firing hooks.
+            // Send the chunk and wait for the run to finish.
+            await sendSessionInput(sessionId, { kind: "handover-skip" });
+            await Promise.race([
+                taskFinished.catch(() => { }),
+                new Promise((resolve) => setTimeout(resolve, 1000)),
+            ]);
+        },
+        seedSnapshot(snapshot) {
+            seededSnapshot = snapshot;
+        },
+        seedSessionOutTail(chunks) {
+            seededReplayChunks = chunks ?? [];
+        },
+        seedSessionOutPartial(partial) {
+            seededReplayPartial = partial;
+        },
+        seedSessionInTail(messages) {
+            seededSessionInMessages = messages;
+        },
+        getSnapshot() {
+            return lastWrittenSnapshot;
+        },
+        async close() {
+            await harnessReady;
+            // Send a close trigger wrapped as a `kind: "message"` ChatInputChunk.
+            // The turn loop checks for this after a successful turn and exits
+            // cleanly. On error-recovery paths the loop just loops back with
+            // the close payload, so we also close the session input below to
+            // unblock any pending once() waiters.
+            try {
+                await sendSessionInput(sessionId, {
+                    kind: "message",
+                    payload: {
+                        chatId,
+                        trigger: "close",
+                    },
+                });
+            }
+            catch {
+                // best-effort
+            }
+            // Resolve any pending once() waiters on the session input with a
+            // timeout error — that makes waitWithIdleTimeout return
+            // `{ ok: false }` and the turn loop exits cleanly.
+            closeSessionInput?.(sessionId);
+            // Also abort the run signal so anything downstream (streamText,
+            // deferred work) unwinds promptly.
+            runSignal?.abort("close");
+            // Wait for run() to return. The loop's error recovery path will
+            // see !next.ok and exit. Use a bounded wait so tests never hang.
+            await Promise.race([
+                taskFinished.catch(() => { }),
+                new Promise((resolve) => setTimeout(resolve, 1000)),
+            ]);
+        },
+        get allChunks() {
+            return allChunks.slice();
+        },
+        get allRawChunks() {
+            return allRawChunks.slice();
+        },
+    };
+    return harness;
+}
+/**
+ * Reduce a synthetic UIMessageChunk[] sequence into the UIMessage[] that
+ * the runtime's `replaySessionOutTail` would produce. Splits chunks at
+ * `start` boundaries and feeds each segment through AI SDK's
+ * `readUIMessageStream`. The trailing un-finished segment goes through
+ * `cleanupAbortedParts`. Mirrors the production reducer used in
+ * `ai.ts:replaySessionOutTail`.
+ */
+async function reduceChunksToMessages(chunks) {
+    if (chunks.length === 0)
+        return [];
+    const aiModule = (await import("ai"));
+    const readUIMessageStream = aiModule.readUIMessageStream;
+    const cleanupAbortedParts = aiModule.cleanupAbortedParts;
+    if (!readUIMessageStream)
+        return [];
+    const segments = [];
+    let current;
+    for (const chunk of chunks) {
+        if (chunk.type === "start") {
+            current = { chunks: [chunk], closed: false };
+            segments.push(current);
+            continue;
+        }
+        if (!current) {
+            current = { chunks: [], closed: false };
+            segments.push(current);
+        }
+        current.chunks.push(chunk);
+        if (chunk.type === "finish") {
+            current.closed = true;
+            current = undefined;
+        }
+    }
+    const out = [];
+    for (let i = 0; i < segments.length; i++) {
+        const seg = segments[i];
+        const isTrailing = i === segments.length - 1 && !seg.closed;
+        const segmentStream = new ReadableStream({
+            start(controller) {
+                for (const c of seg.chunks)
+                    controller.enqueue(c);
+                controller.close();
+            },
+        });
+        let last;
+        try {
+            for await (const snapshot of readUIMessageStream({ stream: segmentStream })) {
+                last = snapshot;
+            }
+        }
+        catch {
+            // Skip malformed segment — tests can assert by inspecting what makes it through.
+            continue;
+        }
+        if (!last)
+            continue;
+        if (isTrailing && cleanupAbortedParts) {
+            const cleaned = cleanupAbortedParts(last);
+            if (!cleaned.parts || cleaned.parts.length === 0)
+                continue;
+            out.push(cleaned);
+        }
+        else {
+            out.push(last);
+        }
+    }
+    return out;
+}
+//# sourceMappingURL=mock-chat-agent.js.map

package/dist/commonjs/v3/test/mock-chat-agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-chat-agent.js","sourceRoot":"","sources":["../../../../src/v3/test/mock-chat-agent.ts"],"names":[],"mappings":";;AAwUA,sCA+YC;AAttBD,6CAAuD;AAEvD,oDAGmC;AACnC,gDAGwB;AACxB,oCAMkB;AAClB,qEAGkC;AAsQlC,MAAM,mBAAmB,GAAG,IAAI,GAAG,CAAC;IAClC,uBAAuB;IACvB,0BAA0B;CAC3B,CAAC,CAAC;AAEH,SAAS,cAAc,CAAC,KAAc;IACpC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC;IAC9D,MAAM,IAAI,GAAI,KAA2B,CAAC,IAAI,CAAC;IAC/C,OAAO,OAAO,IAAI,KAAK,QAAQ,IAAI,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;AACnE,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAiCG;AACH,SAAgB,aAAa,CAC3B,KAAsB,EACtB,UAAgC,EAAE;IAElC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,WAAW,CAAC;IAC7C,0EAA0E;IAC1E,oDAAoD;IACpD,MAAM,SAAS,GAAG,MAAM,CAAC;IACzB,4EAA4E;IAC5E,qDAAqD;IACrD,MAAM,IAAI,GACR,OAAO,CAAC,IAAI;QACZ,CAAC,OAAO,CAAC,YAAY,KAAK,IAAI;YAC5B,CAAC,CAAC,cAAc;YAChB,CAAC,CAAC,OAAO,CAAC,OAAO,KAAK,KAAK;gBACzB,CAAC,CAAC,gBAAgB;gBAClB,CAAC,CAAC,SAAS,CAAC,CAAC;IACnB,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;IAEtC,MAAM,SAAS,GAAG,oBAAe,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACpD,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACb,8CAA8C,KAAK,CAAC,EAAE,KAAK;YACzD,yFAAyF,CAC5F,CAAC;IACJ,CAAC;IAED,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC;IAEhC,qEAAqE;IACrE,4EAA4E;IAC5E,MAAM,eAAe,GAAwB;QAC3C,MAAM,EAAE,EAAE;QACV,SAAS,EAAE,IAAI,GAAG,EAAE;KACrB,CAAC;IAEF,mDAAmD;IACnD,MAAM,YAAY,GAAc,EAAE,CAAC;IACnC,MAAM,SAAS,GAAqB,EAAE,CAAC;IAEvC,yEAAyE;IACzE,IAAI,YAA4B,CAAC;IACjC,IAAI,gBAAsE,CAAC;IAC3E,IAAI,iBAA4D,CAAC;IACjE,IAAI,SAA2B,CAAC;IAEhC,uFAAuF;IACvF,wEAAwE;IACxE,IAAI,qBAAqB,GAAsB,EAAE,CAAC;IAClD,MAAM,mBAAmB,GAAG,GAAG,EAAE,CAC/B,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;QAC5B,qBAAqB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEL,oDAAoD;IACpD,IAAI,mBAAgC,CAAC;IACrC,MAAM,YAAY,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;QACjD,mBAAmB,GAAG,OAAO,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,wEAAwE;IACxE,sEAAsE;IACtE,wDAAwD;IACxD,4DAA4D;IAC5D,oEAAoE;IACpE,8CAA8C;IAC9C,IAAI,cAAc,GAA+B,OAAO,CAAC,QAAQ,CAAC;IAClE,IAAI,mBAA+C,CAAC;IACpD,IAAI,kBAAkB,GAAqB,EAAE,CAAC;IAC9C,IAAI,mBAA0C,CAAC;IAC/C,IAAI,uBAAuB,GAAgB,EAAE,CAAC;IAE9C,IAAA,yCAAiC,EAAC,CAAsB,GAAW,EAAE,EAAE;QACrE,OAAO,cAA+C,CAAC;IACzD,CAAC,CAAC,CAAC;IACH,IAAA,0CAAkC,EAAC,CAAsB,GAAW,EAAE,QAA2B,EAAE,EAAE;QACnG,mBAAmB,GAAG,QAA0B,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,2DAA2D;IAC3D,qEAAqE;IACrE,gEAAgE;IAChE,+DAA+D;IAC/D,0DAA0D;IAC1D,mEAAmE;IACnE,IAAA,6CAAqC,EAAC,KAAK,IAAI,EAAE;QAC/C,MAAM,OAAO,GACX,kBAAkB,CAAC,MAAM,KAAK,CAAC;YAC7B,CAAC,CAAC,EAAE;YACJ,CAAC,CAAE,CAAC,MAAM,sBAAsB,CAAC,kBAAkB,CAAC,CAAe,CAAC;QACxE,mEAAmE;QACnE,kEAAkE;QAClE,kEAAkE;QAClE,OAAO;YACL,OAAO;YACP,OAAO,EAAE,mBAAmB;YAC5B,UAAU,EAAE,mBAAmB;SACvB,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,4DAA4D;IAC5D,mEAAmE;IACnE,oEAAoE;IACpE,oEAAoE;IACpE,4DAA4D;IAC5D,IAAA,4CAAoC,EAAC,KAAK,IAAI,EAAE;QAC9C,OAAO,uBAAuB,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAClD,OAAO;YACP,QAAQ,EAAE,SAAS;YACnB,MAAM,EAAE,CAAC,GAAG,CAAC;SACd,CAAC,CAAU,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,qEAAqE;IACrE,mEAAmE;IACnE,iEAAiE;IACjE,8DAA8D;IAC9D,mEAAmE;IACnE,sEAAsE;IACtE,2BAA2B;IAC3B,IAAA,0CAA4B,EAAC,CAAC,EAAE,EAAE,EAAE,CAClC,IAAA,gDAAuB,EAAC,EAAE,EAAE,eAAe,EAAE,GAAG,EAAE,CAAC,SAAS,EAAE,MAAM,CAAC,CACtE,CAAC;IAEF,mEAAmE;IACnE,kEAAkE;IAClE,sEAAsE;IACtE,qEAAqE;IACrE,mEAAmE;IACnE,2DAA2D;IAC3D,IAAA,2CAA6B,EAAC,CAAC,IAAI,EAAE,EAAE;QACrC,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAuB,KAAK,GAAG,EAAE,CAAC;YAChD,OAAO,CAAC,GAAG,CAAC,0CAA0C,EAAE,IAAI,CAAC,CAAC;QAChE,CAAC;QACD,MAAM,SAAS,GAAG,YAAY,IAAI,CAAC,UAAU,IAAI,MAAM,EAAE,CAAC;QAC1D,OAAO;YACL,EAAE,EAAE,gBAAgB,IAAI,CAAC,UAAU,IAAI,MAAM,EAAE;YAC/C,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI;YACnC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,YAAY,EAAE,SAAS;YACvB,KAAK,EAAE,SAAS;YAChB,iBAAiB,EAAE,qBAAqB;YACxC,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE;YACrB,QAAQ,EAAE,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAmC;YACnE,QAAQ,EAAE,IAAI;YACd,YAAY,EAAE,IAAI;YAClB,SAAS,EAAE,IAAI;YACf,SAAS,EAAE,IAAI,IAAI,CAAC,CAAC,CAAC;YACtB,SAAS,EAAE,IAAI,IAAI,CAAC,CAAC,CAAC;YACtB,QAAQ,EAAE,KAAK;SAChB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,YAAY,GAAG,IAAA,2BAAoB,EACjC,KAAK,EAAE,OAAO,EAAE,EAAE;QAChB,SAAS,GAAG,IAAI,eAAe,EAAE,CAAC;QAElC,qEAAqE;QACrE,qEAAqE;QACrE,kEAAkE;QAClE,+DAA+D;QAC/D,uEAAuE;QACvE,gEAAgE;QAChE,uCAAuC;QACvC,MAAM,kBAAkB,GAAG,IAAI,KAAK,cAAc,CAAC;QACnD,MAAM,cAAc,GAAoB;YACtC,MAAM;YACN,GAAG,CAAC,kBAAkB;gBACpB,CAAC,CAAC,EAAE,OAAO,EAAE,SAAkB,EAAE,YAAY,EAAE,IAAI,EAAE;gBACrD,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACtB,QAAQ,EAAE,UAAU;YACpB,GAAG,CAAC,CAAC,kBAAkB,IAAI,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9E,GAAG,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC3E,CAAC;QAEF,gBAAgB,GAAG,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAC;QAC5C,iBAAiB,GAAG,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,KAAK,CAAC;QAE9C,mEAAmE;QACnE,MAAM,QAAQ,GAAG,CAAC,KAAc,EAAE,EAAE;YAClC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC3B,SAAS,CAAC,IAAI,CAAC,KAAuB,CAAC,CAAC;YAC1C,CAAC;YACD,IACE,OAAO,KAAK,KAAK,QAAQ;gBACzB,KAAK,KAAK,IAAI;gBACb,KAA2B,CAAC,IAAI,KAAK,uBAAuB,EAC7D,CAAC;gBACD,MAAM,SAAS,GAAG,qBAAqB,CAAC;gBACxC,qBAAqB,GAAG,EAAE,CAAC;gBAC3B,KAAK,MAAM,OAAO,IAAI,SAAS;oBAAE,OAAO,EAAE,CAAC;YAC7C,CAAC;QACH,CAAC,CAAC;QACF,eAAe,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACxC,MAAM,WAAW,GAAG,GAAG,EAAE,CAAC,eAAe,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAErE,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YACxB,MAAM,OAAO,CAAC,WAAW,CAAC,EAAE,GAAG,EAAE,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC;QACzD,CAAC;QAED,mBAAmB,EAAE,CAAC;QAEtB,IAAI,CAAC;YACH,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAuB,KAAK,GAAG,EAAE,CAAC;gBAChD,OAAO,CAAC,GAAG,CAAC,8CAA8C,EAAE,cAAc,CAAC,CAAC;YAC9E,CAAC;YACD,MAAM,KAAK,CAAC,cAAc,EAAE;gBAC1B,GAAG,EAAE,OAAO,CAAC,GAAG;gBAChB,MAAM,EAAE,SAAS,CAAC,MAAM;aACzB,CAAC,CAAC;YACH,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAuB,KAAK,GAAG,EAAE,CAAC;gBAChD,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAuB,KAAK,GAAG,EAAE,CAAC;gBAChD,OAAO,CAAC,GAAG,CAAC,8BAA8B,EAAE,GAAG,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;gBAAS,CAAC;YACT,WAAW,EAAE,CAAC;YACd,sEAAsE;YACtE,MAAM,SAAS,GAAG,qBAAqB,CAAC;YACxC,qBAAqB,GAAG,EAAE,CAAC;YAC3B,KAAK,MAAM,OAAO,IAAI,SAAS;gBAAE,OAAO,EAAE,CAAC;QAC7C,CAAC;IACH,CAAC,EACD,OAAO,CAAC,WAAW,CACpB;SACE,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;QACb,oEAAoE;QACpE,MAAM,SAAS,GAAG,qBAAqB,CAAC;QACxC,qBAAqB,GAAG,EAAE,CAAC;QAC3B,KAAK,MAAM,OAAO,IAAI,SAAS;YAAE,OAAO,EAAE,CAAC;QAC3C,MAAM,GAAG,CAAC;IACZ,CAAC,CAAC;SACD,OAAO,CAAC,GAAG,EAAE;QACZ,2DAA2D;QAC3D,IAAA,0CAA4B,EAAC,SAAS,CAAC,CAAC;QACxC,IAAA,2CAA6B,EAAC,SAAS,CAAC,CAAC;QACzC,IAAA,yCAAiC,EAAC,SAAS,CAAC,CAAC;QAC7C,IAAA,0CAAkC,EAAC,SAAS,CAAC,CAAC;QAC9C,IAAA,6CAAqC,EAAC,SAAS,CAAC,CAAC;QACjD,IAAA,4CAAoC,EAAC,SAAS,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEL,MAAM,kBAAkB,GAAG,KAAK,EAC9B,OAAwB,EACI,EAAE;QAC9B,MAAM,YAAY,CAAC;QACnB,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;QACnC,MAAM,YAAY,GAAG,mBAAmB,EAAE,CAAC;QAC3C,MAAM,gBAAgB,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;QAChE,MAAM,YAAY,CAAC;QACnB,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAC7B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CACN,CAAC;QACtB,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IAC/B,CAAC,CAAC;IAEF,MAAM,OAAO,GAAyB;QACpC,MAAM;QAEN,KAAK,CAAC,WAAW,CAAC,OAAO;YACvB,OAAO,kBAAkB,CAAC;gBACxB,OAAO;gBACP,MAAM;gBACN,OAAO,EAAE,gBAAgB;gBACzB,QAAQ,EAAE,UAAU;aACrB,CAAC,CAAC;QACL,CAAC;QAED,KAAK,CAAC,cAAc;YAClB,OAAO,kBAAkB,CAAC;gBACxB,MAAM;gBACN,OAAO,EAAE,oBAAoB;gBAC7B,QAAQ,EAAE,UAAU;aACrB,CAAC,CAAC;QACL,CAAC;QAED,KAAK,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE;YAC9B,OAAO,kBAAkB,CAAC;gBACxB,iBAAiB,EAAE,QAAQ;gBAC3B,MAAM;gBACN,OAAO,EAAE,kBAAkB;gBAC3B,QAAQ,EAAE,UAAU;aACrB,CAAC,CAAC;QACL,CAAC;QAED,KAAK,CAAC,UAAU,CAAC,MAAM;YACrB,OAAO,kBAAkB,CAAC;gBACxB,MAAM;gBACN,OAAO,EAAE,QAAQ;gBACjB,MAAM;gBACN,QAAQ,EAAE,UAAU;aACrB,CAAC,CAAC;QACL,CAAC;QAED,KAAK,CAAC,QAAQ,CAAC,OAAO;YACpB,MAAM,YAAY,CAAC;YACnB,MAAM,gBAAgB,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;QAC/D,CAAC;QAED,KAAK,CAAC,YAAY,CAAC,IAAI;YACrB,MAAM,YAAY,CAAC;YACnB,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;YACnC,MAAM,YAAY,GAAG,mBAAmB,EAAE,CAAC;YAC3C,MAAM,gBAAgB,CAAC,SAAS,EAAE;gBAChC,IAAI,EAAE,UAAU;gBAChB,uBAAuB,EAAE,IAAI,CAAC,uBAAuB;gBACrD,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,KAAK;aAC/B,CAAC,CAAC;YACH,MAAM,YAAY,CAAC;YACnB,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAqB,CAAC;YAC/E,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QAC/B,CAAC;QAED,KAAK,CAAC,gBAAgB;YACpB,MAAM,YAAY,CAAC;YACnB,mEAAmE;YACnE,iDAAiD;YACjD,MAAM,gBAAgB,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC;YAC7D,MAAM,OAAO,CAAC,IAAI,CAAC;gBACjB,YAAY,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;gBAC5B,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;aAC1D,CAAC,CAAC;QACL,CAAC;QAED,YAAY,CAAC,QAAQ;YACnB,cAAc,GAAG,QAAQ,CAAC;QAC5B,CAAC;QAED,kBAAkB,CAAC,MAAM;YACvB,kBAAkB,GAAG,MAAM,IAAI,EAAE,CAAC;QACpC,CAAC;QAED,qBAAqB,CAAC,OAAO;YAC3B,mBAAmB,GAAG,OAAO,CAAC;QAChC,CAAC;QAED,iBAAiB,CAAC,QAAQ;YACxB,uBAAuB,GAAG,QAAQ,CAAC;QACrC,CAAC;QAED,WAAW;YACT,OAAO,mBAAmB,CAAC;QAC7B,CAAC;QAED,KAAK,CAAC,KAAK;YACT,MAAM,YAAY,CAAC;YAEnB,sEAAsE;YACtE,kEAAkE;YAClE,iEAAiE;YACjE,iEAAiE;YACjE,sCAAsC;YACtC,IAAI,CAAC;gBACH,MAAM,gBAAgB,CAAC,SAAS,EAAE;oBAChC,IAAI,EAAE,SAAS;oBACf,OAAO,EAAE;wBACP,MAAM;wBACN,OAAO,EAAE,OAAO;qBACjB;iBACF,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;gBACP,cAAc;YAChB,CAAC;YACD,iEAAiE;YACjE,wDAAwD;YACxD,mDAAmD;YACnD,iBAAiB,EAAE,CAAC,SAAS,CAAC,CAAC;YAE/B,gEAAgE;YAChE,mCAAmC;YACnC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YAE1B,gEAAgE;YAChE,iEAAiE;YACjE,MAAM,OAAO,CAAC,IAAI,CAAC;gBACjB,YAAY,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;gBAC5B,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;aAC1D,CAAC,CAAC;QACL,CAAC;QAED,IAAI,SAAS;YACX,OAAO,SAAS,CAAC,KAAK,EAAE,CAAC;QAC3B,CAAC;QAED,IAAI,YAAY;YACd,OAAO,YAAY,CAAC,KAAK,EAAE,CAAC;QAC9B,CAAC;KACF,CAAC;IAEF,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;;;;GAOG;AACH,KAAK,UAAU,sBAAsB,CAAC,MAAwB;IAC5D,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IACnC,MAAM,QAAQ,GAAG,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,CAGnC,CAAC;IACF,MAAM,mBAAmB,GAAG,QAAQ,CAAC,mBAAmB,CAAC;IACzD,MAAM,mBAAmB,GAAG,QAAQ,CAAC,mBAAmB,CAAC;IACzD,IAAI,CAAC,mBAAmB;QAAE,OAAO,EAAE,CAAC;IAGpC,MAAM,QAAQ,GAAc,EAAE,CAAC;IAC/B,IAAI,OAA4B,CAAC;IACjC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC3B,OAAO,GAAG,EAAE,MAAM,EAAE,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;YAC7C,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACvB,SAAS;QACX,CAAC;QACD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;YACxC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACzB,CAAC;QACD,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC5B,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;YACtB,OAAO,GAAG,SAAS,CAAC;QACtB,CAAC;IACH,CAAC;IAED,MAAM,GAAG,GAAgB,EAAE,CAAC;IAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAE,CAAC;QACzB,MAAM,UAAU,GAAG,CAAC,KAAK,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC;QAC5D,MAAM,aAAa,GAAG,IAAI,cAAc,CAAiB;YACvD,KAAK,CAAC,UAAU;gBACd,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,MAAM;oBAAE,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBAClD,UAAU,CAAC,KAAK,EAAE,CAAC;YACrB,CAAC;SACF,CAAC,CAAC;QACH,IAAI,IAA2B,CAAC;QAChC,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,mBAAmB,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC;gBAC5E,IAAI,GAAG,QAAQ,CAAC;YAClB,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,iFAAiF;YACjF,SAAS;QACX,CAAC;QACD,IAAI,CAAC,IAAI;YAAE,SAAS;QACpB,IAAI,UAAU,IAAI,mBAAmB,EAAE,CAAC;YACtC,MAAM,OAAO,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAC1C,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YAC3D,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpB,CAAC;aAAM,CAAC;YACN,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC"}

package/dist/commonjs/v3/test/setup-catalog.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/commonjs/v3/test/setup-catalog.js

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const v3_1 = require("@trigger.dev/core/v3");
+const workers_1 = require("@trigger.dev/core/v3/workers");
+/**
+ * Installs an in-memory `StandardResourceCatalog` and seeds a fake file
+ * context so task definitions (`task()`, `chat.agent()`, etc.) register
+ * their run functions where the test harness can look them up.
+ *
+ * This is invoked as a side-effect of importing `@trigger.dev/sdk/ai/test`.
+ *
+ * Without this, `registerTaskMetadata` short-circuits on a missing
+ * `_currentFileContext` and tasks silently fail to register.
+ */
+const catalog = new workers_1.StandardResourceCatalog();
+v3_1.resourceCatalog.setGlobalResourceCatalog(catalog);
+v3_1.resourceCatalog.setCurrentFileContext("__test__.ts", "__test__");
+//# sourceMappingURL=setup-catalog.js.map

package/dist/commonjs/v3/test/setup-catalog.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup-catalog.js","sourceRoot":"","sources":["../../../../src/v3/test/setup-catalog.ts"],"names":[],"mappings":";;AAAA,6CAAuD;AACvD,0DAAuE;AAEvE;;;;;;;;;GASG;AACH,MAAM,OAAO,GAAG,IAAI,iCAAuB,EAAE,CAAC;AAC9C,oBAAe,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC;AAClD,oBAAe,CAAC,qBAAqB,CAAC,aAAa,EAAE,UAAU,CAAC,CAAC"}

package/dist/commonjs/v3/test/test-session-handle.d.ts

@@ -0,0 +1,53 @@
+import type { AsyncIterableStream, PipeStreamResult, StreamWriteResult, WriterStreamOptions } from "@trigger.dev/core/v3";
+import { SessionHandle, SessionOutputChannel, SessionPipeStreamOptions, SessionSubscribeOptions } from "../sessions.js";
+/**
+ * Per-session in-memory state collected from `.out` writes during a test.
+ * Owned by the mock-chat-agent harness; updated by {@link TestSessionOutputChannel}.
+ */
+export type TestSessionOutState = {
+    /** Every chunk written to `.out`, in order of write. */
+    chunks: unknown[];
+    /** Registered write listeners (fired for each chunk). */
+    listeners: Set<(chunk: unknown) => void>;
+};
+/**
+ * `.out` channel that captures writes in memory instead of piping to S2.
+ * Mirrors {@link SessionOutputChannel}'s public shape — `pipe` / `writer`
+ * / `append` / `read` — so the agent's existing code paths work unchanged.
+ */
+export declare class TestSessionOutputChannel extends SessionOutputChannel {
+    private readonly state;
+    constructor(sessionId: string, state: TestSessionOutState);
+    append<T>(value: T, _options?: SessionPipeStreamOptions): Promise<void>;
+    pipe<T>(value: AsyncIterable<T> | ReadableStream<T>, _options?: SessionPipeStreamOptions): PipeStreamResult<T>;
+    writer<T>(options: WriterStreamOptions<T>): PipeStreamResult<T>;
+    read<T>(_options?: SessionSubscribeOptions<T>): Promise<AsyncIterableStream<T>>;
+    /**
+     * Override the one-shot control-record path. In production this goes
+     * direct to S2 with header-form records; in tests we project it back
+     * into the chunk-shape the harness already understands (the listener
+     * watches for `{type: "trigger:turn-complete"}` to drive turn-complete
+     * latches). Returns an empty `StreamWriteResult` — tests don't observe
+     * the seq_num, and trim seeding only matters in production.
+     */
+    writeControl(subtype: string, extraHeaders?: ReadonlyArray<readonly [string, string]>): Promise<StreamWriteResult>;
+    /**
+     * No-op in the mock harness. Production trims keep `session.out` bounded;
+     * the in-memory `state.chunks` array doesn't need trimming and tests
+     * that care about trim behaviour exercise it via the real S2 code path.
+     */
+    trimTo(_earliestSeqNum: number): Promise<void>;
+}
+/**
+ * Construct a {@link SessionHandle} whose `.out` channel captures writes in
+ * memory and whose `.in` channel routes through the `sessionStreams`
+ * global for record subscriptions (`on` / `once` / `peek`) but stubs
+ * `wait()` to skip the apiClient round-trip — see
+ * {@link TestSessionInputChannel}.
+ *
+ * `getAbortSignal` lets the channel observe the harness's run signal so
+ * `wait()` resolves cleanly on close. Pass a getter (not the signal
+ * directly) so the channel reads it lazily — the harness creates its
+ * `AbortController` after the override is installed.
+ */
+export declare function createTestSessionHandle(sessionId: string, state: TestSessionOutState, getAbortSignal?: () => AbortSignal | undefined): SessionHandle;