@trigger.dev/core 0.0.0-chat-prerelease-20260417152143 → 0.0.0-chat-prerelease-20260418174118

package/dist/esm/v3/test/test-input-stream-manager.js

@@ -0,0 +1,188 @@
+import { InputStreamOncePromise, InputStreamTimeoutError } from "../inputStreams/types.js";
+/**
+ * In-memory implementation of `InputStreamManager` for unit tests.
+ *
+ * Tests push data via the driver's `.send(streamId, data)` method. Any
+ * pending `.once()` waiters resolve immediately, and all `.on()` handlers
+ * fire synchronously (awaited if they return a promise).
+ *
+ * Use this alongside {@link runInMockTaskContext} — not directly.
+ */
+export class TestInputStreamManager {
+    handlers = new Map();
+    onceWaiters = new Map();
+    latest = new Map();
+    lastSeqNums = new Map();
+    // Buffered sends that arrived before a `.once()` waiter was registered.
+    // `.once()` semantically means "wait for NEXT value" but tests often
+    // send data before the task has had a chance to reach the wait point.
+    // Buffering closes that race so the waiter picks up the pending send.
+    pendingSends = new Map();
+    setRunId(_runId, _streamsVersion) {
+        // No-op — the test driver tracks nothing about runs
+    }
+    on(streamId, handler) {
+        if (!this.handlers.has(streamId)) {
+            this.handlers.set(streamId, new Set());
+        }
+        this.handlers.get(streamId).add(handler);
+        return {
+            off: () => {
+                this.handlers.get(streamId)?.delete(handler);
+            },
+        };
+    }
+    once(streamId, options) {
+        return new InputStreamOncePromise((resolve) => {
+            if (options?.signal?.aborted) {
+                resolve({
+                    ok: false,
+                    error: new InputStreamTimeoutError(streamId, options.timeoutMs ?? 0),
+                });
+                return;
+            }
+            // Pick up any buffered send that arrived before this waiter.
+            const buffered = this.pendingSends.get(streamId);
+            if (buffered && buffered.length > 0) {
+                const next = buffered.shift();
+                if (buffered.length === 0)
+                    this.pendingSends.delete(streamId);
+                resolve({ ok: true, output: next });
+                return;
+            }
+            const waiter = {
+                resolve,
+                signal: options?.signal,
+            };
+            if (options?.timeoutMs !== undefined) {
+                waiter.timer = setTimeout(() => {
+                    this.removeWaiter(streamId, waiter);
+                    resolve({
+                        ok: false,
+                        error: new InputStreamTimeoutError(streamId, options.timeoutMs),
+                    });
+                }, options.timeoutMs);
+            }
+            if (options?.signal) {
+                const abortHandler = () => {
+                    this.removeWaiter(streamId, waiter);
+                    if (waiter.timer)
+                        clearTimeout(waiter.timer);
+                    resolve({
+                        ok: false,
+                        error: new InputStreamTimeoutError(streamId, options.timeoutMs ?? 0),
+                    });
+                };
+                waiter.abortHandler = abortHandler;
+                options.signal.addEventListener("abort", abortHandler, { once: true });
+            }
+            if (!this.onceWaiters.has(streamId)) {
+                this.onceWaiters.set(streamId, []);
+            }
+            this.onceWaiters.get(streamId).push(waiter);
+        });
+    }
+    peek(streamId) {
+        return this.latest.get(streamId);
+    }
+    lastSeqNum(streamId) {
+        return this.lastSeqNums.get(streamId);
+    }
+    setLastSeqNum(streamId, seqNum) {
+        this.lastSeqNums.set(streamId, seqNum);
+    }
+    shiftBuffer(_streamId) {
+        return false;
+    }
+    disconnectStream(_streamId) { }
+    clearHandlers() {
+        this.handlers.clear();
+    }
+    reset() {
+        // Cancel any pending waiters to avoid dangling promises leaking between tests
+        for (const waiters of this.onceWaiters.values()) {
+            for (const w of waiters) {
+                if (w.timer)
+                    clearTimeout(w.timer);
+                if (w.signal && w.abortHandler) {
+                    w.signal.removeEventListener("abort", w.abortHandler);
+                }
+            }
+        }
+        this.onceWaiters.clear();
+        this.handlers.clear();
+        this.latest.clear();
+        this.lastSeqNums.clear();
+        this.pendingSends.clear();
+    }
+    disconnect() {
+        this.reset();
+    }
+    connectTail(_runId, _fromSeq) { }
+    // ── Test driver API (not part of InputStreamManager interface) ──────────
+    /**
+     * Push data onto an input stream. Resolves pending `once()` waiters
+     * and fires all `on()` handlers (awaiting async handlers).
+     */
+    async __sendFromTest(streamId, data) {
+        this.latest.set(streamId, data);
+        const waiters = this.onceWaiters.get(streamId);
+        const handlers = this.handlers.get(streamId);
+        const hasWaiters = waiters && waiters.length > 0;
+        const hasHandlers = handlers && handlers.size > 0;
+        // If nothing is listening yet, buffer so the next `.once()` call picks it up.
+        if (!hasWaiters && !hasHandlers) {
+            if (!this.pendingSends.has(streamId)) {
+                this.pendingSends.set(streamId, []);
+            }
+            this.pendingSends.get(streamId).push(data);
+            return;
+        }
+        if (hasWaiters) {
+            // Drain every pending once() waiter — this mirrors the real manager's
+            // behavior where the stream tail delivers the same record to all listeners.
+            const pending = waiters.splice(0);
+            for (const w of pending) {
+                if (w.timer)
+                    clearTimeout(w.timer);
+                if (w.signal && w.abortHandler) {
+                    w.signal.removeEventListener("abort", w.abortHandler);
+                }
+                w.resolve({ ok: true, output: data });
+            }
+        }
+        if (hasHandlers) {
+            await Promise.all(Array.from(handlers).map((h) => Promise.resolve().then(() => h(data))));
+        }
+    }
+    /**
+     * Immediately resolve every pending `once()` waiter for a stream with a
+     * timeout error. Used to simulate closed streams (e.g. `exitAfterPreloadIdle`).
+     */
+    __closeFromTest(streamId) {
+        const waiters = this.onceWaiters.get(streamId);
+        if (!waiters)
+            return;
+        const pending = waiters.splice(0);
+        for (const w of pending) {
+            if (w.timer)
+                clearTimeout(w.timer);
+            if (w.signal && w.abortHandler) {
+                w.signal.removeEventListener("abort", w.abortHandler);
+            }
+            w.resolve({
+                ok: false,
+                error: new InputStreamTimeoutError(streamId, 0),
+            });
+        }
+    }
+    removeWaiter(streamId, waiter) {
+        const waiters = this.onceWaiters.get(streamId);
+        if (!waiters)
+            return;
+        const idx = waiters.indexOf(waiter);
+        if (idx >= 0)
+            waiters.splice(idx, 1);
+    }
+}
+//# sourceMappingURL=test-input-stream-manager.js.map

package/dist/esm/v3/test/test-input-stream-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-input-stream-manager.js","sourceRoot":"","sources":["../../../../src/v3/test/test-input-stream-manager.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,0BAA0B,CAAC;AAY3F;;;;;;;;GAQG;AACH,MAAM,OAAO,sBAAsB;IACzB,QAAQ,GAAG,IAAI,GAAG,EAAwB,CAAC;IAC3C,WAAW,GAAG,IAAI,GAAG,EAAwB,CAAC;IAC9C,MAAM,GAAG,IAAI,GAAG,EAAmB,CAAC;IACpC,WAAW,GAAG,IAAI,GAAG,EAAkB,CAAC;IAChD,wEAAwE;IACxE,qEAAqE;IACrE,sEAAsE;IACtE,sEAAsE;IAC9D,YAAY,GAAG,IAAI,GAAG,EAAqB,CAAC;IAEpD,QAAQ,CAAC,MAAc,EAAE,eAAwB;QAC/C,oDAAoD;IACtD,CAAC;IAED,EAAE,CAAC,QAAgB,EAAE,OAAgB;QACnC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;QACzC,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAE1C,OAAO;YACL,GAAG,EAAE,GAAG,EAAE;gBACR,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;YAC/C,CAAC;SACF,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,QAAgB,EAAE,OAAgC;QACrD,OAAO,IAAI,sBAAsB,CAAU,CAAC,OAAO,EAAE,EAAE;YACrD,IAAI,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;gBAC7B,OAAO,CAAC;oBACN,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE,IAAI,uBAAuB,CAAC,QAAQ,EAAE,OAAO,CAAC,SAAS,IAAI,CAAC,CAAC;iBACrE,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,6DAA6D;YAC7D,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACjD,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpC,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;gBAC9B,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;oBAAE,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBAC9D,OAAO,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBACpC,OAAO;YACT,CAAC;YAED,MAAM,MAAM,GAAe;gBACzB,OAAO;gBACP,MAAM,EAAE,OAAO,EAAE,MAAM;aACxB,CAAC;YAEF,IAAI,OAAO,EAAE,SAAS,KAAK,SAAS,EAAE,CAAC;gBACrC,MAAM,CAAC,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;oBAC7B,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;oBACpC,OAAO,CAAC;wBACN,EAAE,EAAE,KAAK;wBACT,KAAK,EAAE,IAAI,uBAAuB,CAAC,QAAQ,EAAE,OAAO,CAAC,SAAU,CAAC;qBACjE,CAAC,CAAC;gBACL,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;YACxB,CAAC;YAED,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;gBACpB,MAAM,YAAY,GAAG,GAAG,EAAE;oBACxB,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;oBACpC,IAAI,MAAM,CAAC,KAAK;wBAAE,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBAC7C,OAAO,CAAC;wBACN,EAAE,EAAE,KAAK;wBACT,KAAK,EAAE,IAAI,uBAAuB,CAAC,QAAQ,EAAE,OAAO,CAAC,SAAS,IAAI,CAAC,CAAC;qBACrE,CAAC,CAAC;gBACL,CAAC,CAAC;gBACF,MAAM,CAAC,YAAY,GAAG,YAAY,CAAC;gBACnC,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YACzE,CAAC;YAED,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACpC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YACrC,CAAC;YACD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI,CAAC,QAAgB;QACnB,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IAED,UAAU,CAAC,QAAgB;QACzB,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAED,aAAa,CAAC,QAAgB,EAAE,MAAc;QAC5C,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IACzC,CAAC;IAED,WAAW,CAAC,SAAiB;QAC3B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,gBAAgB,CAAC,SAAiB,IAAS,CAAC;IAE5C,aAAa;QACX,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;IAED,KAAK;QACH,8EAA8E;QAC9E,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;YAChD,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;gBACxB,IAAI,CAAC,CAAC,KAAK;oBAAE,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBACnC,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,YAAY,EAAE,CAAC;oBAC/B,CAAC,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC;gBACxD,CAAC;YACH,CAAC;QACH,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACpB,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;IAC5B,CAAC;IAED,UAAU;QACR,IAAI,CAAC,KAAK,EAAE,CAAC;IACf,CAAC;IAED,WAAW,CAAC,MAAc,EAAE,QAAiB,IAAS,CAAC;IAEvD,2EAA2E;IAE3E;;;OAGG;IACH,KAAK,CAAC,cAAc,CAAC,QAAgB,EAAE,IAAa;QAClD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAEhC,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,UAAU,GAAG,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;QACjD,MAAM,WAAW,GAAG,QAAQ,IAAI,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC;QAElD,8EAA8E;QAC9E,IAAI,CAAC,UAAU,IAAI,CAAC,WAAW,EAAE,CAAC;YAChC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACrC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YACtC,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5C,OAAO;QACT,CAAC;QAED,IAAI,UAAU,EAAE,CAAC;YACf,sEAAsE;YACtE,4EAA4E;YAC5E,MAAM,OAAO,GAAG,OAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACnC,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;gBACxB,IAAI,CAAC,CAAC,KAAK;oBAAE,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBACnC,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,YAAY,EAAE,CAAC;oBAC/B,CAAC,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC;gBACxD,CAAC;gBACD,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QAED,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,IAAI,CAAC,QAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CACxE,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,eAAe,CAAC,QAAgB;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAClC,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACxB,IAAI,CAAC,CAAC,KAAK;gBAAE,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACnC,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,YAAY,EAAE,CAAC;gBAC/B,CAAC,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC;YACxD,CAAC;YACD,CAAC,CAAC,OAAO,CAAC;gBACR,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE,IAAI,uBAAuB,CAAC,QAAQ,EAAE,CAAC,CAAC;aAChD,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,QAAgB,EAAE,MAAkB;QACvD,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,MAAM,GAAG,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,GAAG,IAAI,CAAC;YAAE,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IACvC,CAAC;CACF"}

package/dist/esm/v3/test/test-realtime-streams-manager.d.ts

@@ -0,0 +1,40 @@
+import type { RealtimeStreamInstance, RealtimeStreamOperationOptions, RealtimeStreamsManager } from "../realtimeStreams/types.js";
+/**
+ * In-memory implementation of `RealtimeStreamsManager` for unit tests.
+ * Collects every chunk that tasks write via `pipe()` or `append()` into
+ * per-stream buffers that tests can inspect.
+ *
+ * Use this alongside {@link runInMockTaskContext} — not directly.
+ */
+type WriteListener = (key: string, chunk: unknown) => void;
+export declare class TestRealtimeStreamsManager implements RealtimeStreamsManager {
+    private buffers;
+    private pipeWaits;
+    private writeListeners;
+    pipe<T>(key: string, source: AsyncIterable<T> | ReadableStream<T>, _options?: RealtimeStreamOperationOptions): RealtimeStreamInstance<T>;
+    append<TPart extends BodyInit>(key: string, part: TPart, _options?: RealtimeStreamOperationOptions): Promise<void>;
+    /**
+     * Register a listener fired for every chunk written to any stream.
+     * Returns an unsubscribe function.
+     *
+     * Intended for test harnesses that need to react to writes synchronously
+     * (e.g. resolving a "turn complete" latch).
+     */
+    onWrite(listener: WriteListener): () => void;
+    private notify;
+    /**
+     * Return all chunks written to the given stream key in order of write.
+     */
+    __chunksFromTest<T = unknown>(key: string): T[];
+    /**
+     * Return all chunks across every stream, keyed by stream id.
+     */
+    __allChunksFromTest(): Record<string, unknown[]>;
+    /**
+     * Clear the buffer for a specific stream or all streams.
+     */
+    __clearFromTest(key?: string): void;
+    reset(): void;
+    private getBuffer;
+}
+export {};

package/dist/esm/v3/test/test-realtime-streams-manager.js

@@ -0,0 +1,135 @@
+import { createAsyncIterableStreamFromAsyncIterable, } from "../streams/asyncIterableStream.js";
+export class TestRealtimeStreamsManager {
+    buffers = new Map();
+    pipeWaits = new Map();
+    writeListeners = new Set();
+    pipe(key, source, _options) {
+        const buffer = this.getBuffer(key);
+        const self = this;
+        // Eagerly drain the source in the background so chunks land in the
+        // buffer + notify listeners even when the caller never consumes the
+        // returned stream. This mirrors the real SDK behavior: `streams.writer`
+        // awaits `instance.wait()`, it doesn't read the returned stream.
+        //
+        // The source is read ONCE (into a chunks array) and replayed into a
+        // ReadableStream so the caller can still consume it if they want.
+        const readChunks = [];
+        let resolveDone;
+        const done = new Promise((resolve) => {
+            resolveDone = resolve;
+        });
+        (async () => {
+            try {
+                const iter = source instanceof ReadableStream
+                    ? (async function* () {
+                        const reader = source.getReader();
+                        try {
+                            while (true) {
+                                const { done: d, value } = await reader.read();
+                                if (d)
+                                    return;
+                                yield value;
+                            }
+                        }
+                        finally {
+                            reader.releaseLock();
+                        }
+                    })()
+                    : source;
+                for await (const chunk of iter) {
+                    readChunks.push(chunk);
+                    buffer.push(chunk);
+                    self.notify(key, chunk);
+                }
+            }
+            catch {
+                // Swallow — tests can inspect what made it into the buffer
+            }
+            finally {
+                resolveDone();
+            }
+        })();
+        const replayStream = (async function* () {
+            // Wait for all chunks to be drained, then replay from our snapshot
+            await done;
+            for (const chunk of readChunks)
+                yield chunk;
+        })();
+        const wrappedStream = createAsyncIterableStreamFromAsyncIterable(replayStream);
+        if (!this.pipeWaits.has(key))
+            this.pipeWaits.set(key, []);
+        this.pipeWaits.get(key).push(done);
+        return {
+            wait: () => done.then(() => ({})),
+            get stream() {
+                return wrappedStream;
+            },
+        };
+    }
+    async append(key, part, _options) {
+        this.getBuffer(key).push(part);
+        this.notify(key, part);
+    }
+    /**
+     * Register a listener fired for every chunk written to any stream.
+     * Returns an unsubscribe function.
+     *
+     * Intended for test harnesses that need to react to writes synchronously
+     * (e.g. resolving a "turn complete" latch).
+     */
+    onWrite(listener) {
+        this.writeListeners.add(listener);
+        return () => {
+            this.writeListeners.delete(listener);
+        };
+    }
+    notify(key, chunk) {
+        for (const listener of this.writeListeners) {
+            try {
+                listener(key, chunk);
+            }
+            catch {
+                // Never let a listener error break stream writes
+            }
+        }
+    }
+    // ── Test driver API (not part of RealtimeStreamsManager interface) ──────
+    /**
+     * Return all chunks written to the given stream key in order of write.
+     */
+    __chunksFromTest(key) {
+        return (this.buffers.get(key) ?? []).slice();
+    }
+    /**
+     * Return all chunks across every stream, keyed by stream id.
+     */
+    __allChunksFromTest() {
+        const result = {};
+        for (const [key, chunks] of this.buffers.entries()) {
+            result[key] = chunks.slice();
+        }
+        return result;
+    }
+    /**
+     * Clear the buffer for a specific stream or all streams.
+     */
+    __clearFromTest(key) {
+        if (key === undefined) {
+            this.buffers.clear();
+        }
+        else {
+            this.buffers.delete(key);
+        }
+    }
+    reset() {
+        this.buffers.clear();
+        this.pipeWaits.clear();
+    }
+    getBuffer(key) {
+        if (!this.buffers.has(key)) {
+            this.buffers.set(key, []);
+        }
+        return this.buffers.get(key);
+    }
+}
+//# sourceMappingURL=test-realtime-streams-manager.js.map

package/dist/esm/v3/test/test-realtime-streams-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-realtime-streams-manager.js","sourceRoot":"","sources":["../../../../src/v3/test/test-realtime-streams-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,0CAA0C,GAC3C,MAAM,mCAAmC,CAAC;AAgB3C,MAAM,OAAO,0BAA0B;IAC7B,OAAO,GAAG,IAAI,GAAG,EAAqB,CAAC;IACvC,SAAS,GAAG,IAAI,GAAG,EAA2B,CAAC;IAC/C,cAAc,GAAG,IAAI,GAAG,EAAiB,CAAC;IAElD,IAAI,CACF,GAAW,EACX,MAA4C,EAC5C,QAAyC;QAEzC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,IAAI,GAAG,IAAI,CAAC;QAElB,mEAAmE;QACnE,oEAAoE;QACpE,wEAAwE;QACxE,iEAAiE;QACjE,EAAE;QACF,oEAAoE;QACpE,kEAAkE;QAClE,MAAM,UAAU,GAAQ,EAAE,CAAC;QAC3B,IAAI,WAAwB,CAAC;QAC7B,MAAM,IAAI,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YACzC,WAAW,GAAG,OAAO,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,CAAC,KAAK,IAAI,EAAE;YACV,IAAI,CAAC;gBACH,MAAM,IAAI,GACR,MAAM,YAAY,cAAc;oBAC9B,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC;wBACd,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;wBAClC,IAAI,CAAC;4BACH,OAAO,IAAI,EAAE,CAAC;gCACZ,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gCAC/C,IAAI,CAAC;oCAAE,OAAO;gCACd,MAAM,KAAU,CAAC;4BACnB,CAAC;wBACH,CAAC;gCAAS,CAAC;4BACT,MAAM,CAAC,WAAW,EAAE,CAAC;wBACvB,CAAC;oBACH,CAAC,CAAC,EAAE;oBACN,CAAC,CAAC,MAAM,CAAC;gBAEb,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,EAAE,CAAC;oBAC/B,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACvB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACnB,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAC1B,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,2DAA2D;YAC7D,CAAC;oBAAS,CAAC;gBACT,WAAW,EAAE,CAAC;YAChB,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;QAEL,MAAM,YAAY,GAAG,CAAC,KAAK,SAAS,CAAC;YACnC,mEAAmE;YACnE,MAAM,IAAI,CAAC;YACX,KAAK,MAAM,KAAK,IAAI,UAAU;gBAAE,MAAM,KAAK,CAAC;QAC9C,CAAC,CAAC,EAAE,CAAC;QACL,MAAM,aAAa,GAAG,0CAA0C,CAAC,YAAY,CAAC,CAAC;QAE/E,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC1D,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEpC,OAAO;YACL,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YACjC,IAAI,MAAM;gBACR,OAAO,aAAa,CAAC;YACvB,CAAC;SACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CACV,GAAW,EACX,IAAW,EACX,QAAyC;QAEzC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IACzB,CAAC;IAED;;;;;;OAMG;IACH,OAAO,CAAC,QAAuB;QAC7B,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClC,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,GAAW,EAAE,KAAc;QACxC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YAC3C,IAAI,CAAC;gBACH,QAAQ,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YACvB,CAAC;YAAC,MAAM,CAAC;gBACP,iDAAiD;YACnD,CAAC;QACH,CAAC;IACH,CAAC;IAED,2EAA2E;IAE3E;;OAEG;IACH,gBAAgB,CAAc,GAAW;QACvC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,EAAS,CAAC;IACtD,CAAC;IAED;;OAEG;IACH,mBAAmB;QACjB,MAAM,MAAM,GAA8B,EAAE,CAAC;QAC7C,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACnD,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC;QAC/B,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,GAAY;QAC1B,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACtB,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACvB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAEO,SAAS,CAAC,GAAW;QAC3B,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC;IAChC,CAAC;CACF"}

package/dist/esm/v3/test/test-run-metadata-manager.d.ts

@@ -0,0 +1,31 @@
+import type { DeserializedJson } from "../../schemas/json.js";
+import type { AsyncIterableStream } from "../streams/asyncIterableStream.js";
+import type { RunMetadataManager, RunMetadataUpdater } from "../runMetadata/types.js";
+/**
+ * In-memory implementation of `RunMetadataManager` for unit tests.
+ *
+ * Just stores metadata in a Map — no API calls, no queue. Good enough
+ * for tests that read/write metadata via `runMetadata.getKey()` /
+ * `runMetadata.set()`, including the IDLE_TIMEOUT and TURN_TIMEOUT
+ * checks inside `chat.agent()`.
+ */
+export declare class TestRunMetadataManager implements RunMetadataManager {
+    private store;
+    enterWithMetadata(metadata: Record<string, DeserializedJson>): void;
+    current(): Record<string, DeserializedJson> | undefined;
+    getKey(key: string): DeserializedJson | undefined;
+    set(key: string, value: DeserializedJson): this;
+    del(key: string): this;
+    append(key: string, value: DeserializedJson): this;
+    remove(key: string, value: DeserializedJson): this;
+    increment(key: string, value: number): this;
+    decrement(key: string, value: number): this;
+    update(metadata: Record<string, DeserializedJson>): this;
+    flush(): Promise<void>;
+    refresh(): Promise<void>;
+    stream<T>(_key: string, value: AsyncIterable<T> | ReadableStream<T>): Promise<AsyncIterable<T>>;
+    fetchStream<T>(_key: string): Promise<AsyncIterableStream<T>>;
+    get parent(): RunMetadataUpdater;
+    get root(): RunMetadataUpdater;
+    reset(): void;
+}

package/dist/esm/v3/test/test-run-metadata-manager.js

@@ -0,0 +1,82 @@
+/**
+ * In-memory implementation of `RunMetadataManager` for unit tests.
+ *
+ * Just stores metadata in a Map — no API calls, no queue. Good enough
+ * for tests that read/write metadata via `runMetadata.getKey()` /
+ * `runMetadata.set()`, including the IDLE_TIMEOUT and TURN_TIMEOUT
+ * checks inside `chat.agent()`.
+ */
+export class TestRunMetadataManager {
+    store = {};
+    enterWithMetadata(metadata) {
+        this.store = { ...metadata };
+    }
+    current() {
+        return { ...this.store };
+    }
+    getKey(key) {
+        return this.store[key];
+    }
+    set(key, value) {
+        this.store[key] = value;
+        return this;
+    }
+    del(key) {
+        delete this.store[key];
+        return this;
+    }
+    append(key, value) {
+        const existing = this.store[key];
+        if (Array.isArray(existing)) {
+            existing.push(value);
+        }
+        else {
+            this.store[key] = [value];
+        }
+        return this;
+    }
+    remove(key, value) {
+        const existing = this.store[key];
+        if (Array.isArray(existing)) {
+            this.store[key] = existing.filter((v) => v !== value);
+        }
+        return this;
+    }
+    increment(key, value) {
+        const existing = this.store[key];
+        const current = typeof existing === "number" ? existing : 0;
+        this.store[key] = current + value;
+        return this;
+    }
+    decrement(key, value) {
+        return this.increment(key, -value);
+    }
+    update(metadata) {
+        this.store = { ...metadata };
+        return this;
+    }
+    async flush() { }
+    async refresh() { }
+    async stream(_key, value) {
+        return value;
+    }
+    async fetchStream(_key) {
+        // Return an empty async iterable — tests can override if needed
+        const empty = {
+            [Symbol.asyncIterator]: () => ({
+                next: () => Promise.resolve({ done: true, value: undefined }),
+            }),
+        };
+        return empty;
+    }
+    get parent() {
+        return this;
+    }
+    get root() {
+        return this;
+    }
+    reset() {
+        this.store = {};
+    }
+}
+//# sourceMappingURL=test-run-metadata-manager.js.map

package/dist/esm/v3/test/test-run-metadata-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-run-metadata-manager.js","sourceRoot":"","sources":["../../../../src/v3/test/test-run-metadata-manager.ts"],"names":[],"mappings":"AAIA;;;;;;;GAOG;AACH,MAAM,OAAO,sBAAsB;IACzB,KAAK,GAAqC,EAAE,CAAC;IAErD,iBAAiB,CAAC,QAA0C;QAC1D,IAAI,CAAC,KAAK,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;IAC/B,CAAC;IAED,OAAO;QACL,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED,MAAM,CAAC,GAAW;QAChB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAED,GAAG,CAAC,GAAW,EAAE,KAAuB;QACtC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACxB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,GAAG,CAAC,GAAW;QACb,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,CAAC,GAAW,EAAE,KAAuB;QACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5B,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,CAAC,GAAW,EAAE,KAAuB;QACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,KAAK,CAAqB,CAAC;QAC5E,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,SAAS,CAAC,GAAW,EAAE,KAAa;QAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACjC,MAAM,OAAO,GAAG,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5D,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,GAAG,KAAK,CAAC;QAClC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,SAAS,CAAC,GAAW,EAAE,KAAa;QAClC,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC;IAED,MAAM,CAAC,QAA0C;QAC/C,IAAI,CAAC,KAAK,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,KAAK,KAAmB,CAAC;IAC/B,KAAK,CAAC,OAAO,KAAmB,CAAC;IAEjC,KAAK,CAAC,MAAM,CACV,IAAY,EACZ,KAA2C;QAE3C,OAAO,KAAyB,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,WAAW,CAAI,IAAY;QAC/B,gEAAgE;QAChE,MAAM,KAAK,GAAG;YACZ,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;gBAC7B,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,IAAa,EAAE,KAAK,EAAE,SAAc,EAAE,CAAC;aAC5E,CAAC;SACH,CAAC;QACF,OAAO,KAA0C,CAAC;IACpD,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK;QACH,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;IAClB,CAAC;CACF"}

package/dist/esm/version.js

@@ -1,2 +1,2 @@
-export const VERSION = "0.0.0-chat-prerelease-20260417152143";
+export const VERSION = "0.0.0-chat-prerelease-20260418174118";
 //# sourceMappingURL=version.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@trigger.dev/core",
-  "version": "0.0.0-chat-prerelease-20260417152143",
+  "version": "0.0.0-chat-prerelease-20260418174118",
   "description": "Core code used across the Trigger.dev SDK and platform",
   "license": "MIT",
   "publishConfig": {
@@ -42,6 +42,7 @@
       "./v3/utils/retries": "./src/v3/utils/retries.ts",
       "./v3/utils/structuredLogger": "./src/v3/utils/structuredLogger.ts",
       "./v3/chat-client": "./src/v3/chat-client.ts",
+      "./v3/test": "./src/v3/test/index.ts",
       "./v3/zodfetch": "./src/v3/zodfetch.ts",
       "./v3/zodMessageHandler": "./src/v3/zodMessageHandler.ts",
       "./v3/zodNamespace": "./src/v3/zodNamespace.ts",
@@ -153,6 +154,9 @@
       ],
       "v3/isomorphic": [
         "dist/commonjs/v3/isomorphic/index.d.ts"
+      ],
+      "v3/test": [
+        "dist/commonjs/v3/test/index.d.ts"
       ]
     }
   },
@@ -447,6 +451,17 @@
         "default": "./dist/commonjs/v3/chat-client.js"
       }
     },
+    "./v3/test": {
+      "import": {
+        "@triggerdotdev/source": "./src/v3/test/index.ts",
+        "types": "./dist/esm/v3/test/index.d.ts",
+        "default": "./dist/esm/v3/test/index.js"
+      },
+      "require": {
+        "types": "./dist/commonjs/v3/test/index.d.ts",
+        "default": "./dist/commonjs/v3/test/index.js"
+      }
+    },
     "./v3/zodfetch": {
       "import": {
         "@triggerdotdev/source": "./src/v3/zodfetch.ts",