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

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

@@ -0,0 +1,139 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TestRealtimeStreamsManager = void 0;
+const asyncIterableStream_js_1 = require("../streams/asyncIterableStream.js");
+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 = (0, asyncIterableStream_js_1.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);
+    }
+}
+exports.TestRealtimeStreamsManager = TestRealtimeStreamsManager;
+//# sourceMappingURL=test-realtime-streams-manager.js.map

package/dist/commonjs/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,8EAG2C;AAgB3C,MAAa,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,IAAA,mEAA0C,EAAC,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;AArJD,gEAqJC"}

package/dist/commonjs/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/commonjs/v3/test/test-run-metadata-manager.js

@@ -0,0 +1,86 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TestRunMetadataManager = void 0;
+/**
+ * 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()`.
+ */
+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 = {};
+    }
+}
+exports.TestRunMetadataManager = TestRunMetadataManager;
+//# sourceMappingURL=test-run-metadata-manager.js.map

package/dist/commonjs/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,MAAa,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;AA1FD,wDA0FC"}

package/dist/commonjs/version.js

@@ -1,5 +1,5 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.VERSION = void 0;
-exports.VERSION = "0.0.0-chat-prerelease-20260417152143";
+exports.VERSION = "0.0.0-chat-prerelease-20260418083610";
 //# sourceMappingURL=version.js.map

package/dist/esm/v3/test/index.d.ts

@@ -0,0 +1,4 @@
+export { runInMockTaskContext, type MockTaskContextDrivers, type MockTaskContextOptions, } from "./mock-task-context.js";
+export { TestInputStreamManager } from "./test-input-stream-manager.js";
+export { TestRealtimeStreamsManager } from "./test-realtime-streams-manager.js";
+export { TestRunMetadataManager } from "./test-run-metadata-manager.js";

package/dist/esm/v3/test/index.js

@@ -0,0 +1,5 @@
+export { runInMockTaskContext, } from "./mock-task-context.js";
+export { TestInputStreamManager } from "./test-input-stream-manager.js";
+export { TestRealtimeStreamsManager } from "./test-realtime-streams-manager.js";
+export { TestRunMetadataManager } from "./test-run-metadata-manager.js";
+//# sourceMappingURL=index.js.map

package/dist/esm/v3/test/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/v3/test/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,oBAAoB,GAGrB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,MAAM,gCAAgC,CAAC;AACxE,OAAO,EAAE,0BAA0B,EAAE,MAAM,oCAAoC,CAAC;AAChF,OAAO,EAAE,sBAAsB,EAAE,MAAM,gCAAgC,CAAC"}

package/dist/esm/v3/test/mock-task-context.d.ts

@@ -0,0 +1,105 @@
+import type { ServerBackgroundWorker, TaskRunContext } from "../schemas/index.js";
+import type { LocalsKey } from "../locals/types.js";
+/**
+ * Shallow-partial overrides applied on top of the default mock
+ * `TaskRunContext`. Each sub-object is a partial of its real shape —
+ * unset fields get sensible defaults.
+ */
+export type MockTaskRunContextOverrides = {
+    task?: Partial<TaskRunContext["task"]>;
+    attempt?: Partial<TaskRunContext["attempt"]>;
+    run?: Partial<TaskRunContext["run"]>;
+    machine?: Partial<TaskRunContext["machine"]>;
+    queue?: Partial<TaskRunContext["queue"]>;
+    environment?: Partial<TaskRunContext["environment"]>;
+    organization?: Partial<TaskRunContext["organization"]>;
+    project?: Partial<TaskRunContext["project"]>;
+    batch?: TaskRunContext["batch"];
+};
+/**
+ * Options for overriding parts of the mock task context.
+ */
+export type MockTaskContextOptions = {
+    /** Overrides applied on top of the default mock `TaskRunContext`. */
+    ctx?: MockTaskRunContextOverrides;
+    /** Overrides applied on top of the default `ServerBackgroundWorker`. */
+    worker?: Partial<ServerBackgroundWorker>;
+    /** Whether this is a warm start. */
+    isWarmStart?: boolean;
+};
+/**
+ * Drivers passed to the function running inside `runInMockTaskContext`.
+ */
+export type MockTaskContextDrivers = {
+    /** Push data into input streams — simulates realtime input from outside the task. */
+    inputs: {
+        /**
+         * Send `data` to the named input stream. Resolves when all `.on()`
+         * handlers have run.
+         */
+        send(streamId: string, data: unknown): Promise<void>;
+        /** Resolve any pending `.once()` waiters with a timeout error. */
+        close(streamId: string): void;
+    };
+    /** Inspect chunks written to output (realtime) streams. */
+    outputs: {
+        /** All chunks for a given stream, in the order they were written. */
+        chunks<T = unknown>(streamId: string): T[];
+        /** All chunks across every stream, keyed by stream id. */
+        all(): Record<string, unknown[]>;
+        /** Clear chunks for one stream, or all streams if no id is provided. */
+        clear(streamId?: string): void;
+        /**
+         * Register a listener fired for every chunk written to any stream.
+         * Returns an unsubscribe function.
+         */
+        onWrite(listener: (streamId: string, chunk: unknown) => void): () => void;
+    };
+    /** Read or seed locals for the run. */
+    locals: {
+        /** Read a local set by either the task or `set()` below. */
+        get<T>(key: LocalsKey<T>): T | undefined;
+        /**
+         * Pre-seed a local before the task runs. Use this for dependency
+         * injection — e.g. supply a test database client that the agent's
+         * hooks read via `locals.get()` instead of constructing the prod one.
+         */
+        set<T>(key: LocalsKey<T>, value: T): void;
+    };
+    /** The mock `TaskRunContext` assembled from defaults + user overrides. */
+    ctx: TaskRunContext;
+};
+/**
+ * Run a function inside a fully mocked task runtime context.
+ *
+ * Installs in-memory test managers for `locals`, `inputStreams`,
+ * `realtimeStreams`, `lifecycleHooks`, and `runtime`, sets a mock
+ * `TaskContext`, and tears everything down when the function returns.
+ *
+ * Inside the function, any code that reads from `locals`, `inputStreams`,
+ * `realtimeStreams`, or `taskContext.ctx` will see the mock context —
+ * so you can directly invoke the internal `run` function of any task
+ * (including `chat.agent`) without hitting the Trigger.dev runtime.
+ *
+ * @example
+ * ```ts
+ * import { runInMockTaskContext } from "@trigger.dev/core/v3/test";
+ *
+ * await runInMockTaskContext(
+ *   async ({ inputs, outputs, ctx }) => {
+ *     // Fire an input stream from the "outside"
+ *     setTimeout(() => {
+ *       inputs.send("chat-messages", { messages: [], chatId: "c1" });
+ *     }, 0);
+ *
+ *     // Run task code that reads from inputStreams.once(...)
+ *     await myTask.fns.run(payload, { ctx, signal: new AbortController().signal });
+ *
+ *     // Inspect chunks written to the output stream
+ *     expect(outputs.chunks("chat")).toContainEqual({ type: "text-delta", delta: "hi" });
+ *   },
+ *   { ctx: { run: { id: "run_abc" } } }
+ * );
+ * ```
+ */
+export declare function runInMockTaskContext<T>(fn: (drivers: MockTaskContextDrivers) => T | Promise<T>, options?: MockTaskContextOptions): Promise<T>;

package/dist/esm/v3/test/mock-task-context.js

@@ -0,0 +1,176 @@
+import { inputStreams } from "../input-streams-api.js";
+import { realtimeStreams } from "../realtime-streams-api.js";
+import { localsAPI } from "../locals-api.js";
+import { runMetadata } from "../run-metadata-api.js";
+import { taskContext } from "../task-context-api.js";
+import { lifecycleHooks } from "../lifecycle-hooks-api.js";
+import { runtime } from "../runtime-api.js";
+import { StandardLocalsManager } from "../locals/manager.js";
+import { StandardLifecycleHooksManager } from "../lifecycleHooks/manager.js";
+import { NoopRuntimeManager } from "../runtime/noopRuntimeManager.js";
+import { unregisterGlobal } from "../utils/globals.js";
+import { TestInputStreamManager } from "./test-input-stream-manager.js";
+import { TestRealtimeStreamsManager } from "./test-realtime-streams-manager.js";
+import { TestRunMetadataManager } from "./test-run-metadata-manager.js";
+function defaultTaskRunContext(overrides) {
+    return {
+        task: {
+            id: "test-task",
+            filePath: "test-task.ts",
+            ...overrides?.task,
+        },
+        attempt: {
+            number: 1,
+            startedAt: new Date(),
+            ...overrides?.attempt,
+        },
+        run: {
+            id: "run_test",
+            tags: [],
+            isTest: false,
+            createdAt: new Date(),
+            startedAt: new Date(),
+            ...overrides?.run,
+        },
+        machine: {
+            name: "micro",
+            cpu: 1,
+            memory: 0.5,
+            centsPerMs: 0,
+            ...overrides?.machine,
+        },
+        queue: {
+            name: "test-queue",
+            id: "test-queue-id",
+            ...overrides?.queue,
+        },
+        environment: {
+            id: "test-env-id",
+            slug: "test-env",
+            type: "DEVELOPMENT",
+            ...overrides?.environment,
+        },
+        organization: {
+            id: "test-org-id",
+            slug: "test-org",
+            name: "Test Org",
+            ...overrides?.organization,
+        },
+        project: {
+            id: "test-project-id",
+            ref: "test-project-ref",
+            slug: "test-project",
+            name: "Test Project",
+            ...overrides?.project,
+        },
+        batch: overrides?.batch,
+    };
+}
+function defaultWorker(overrides) {
+    return {
+        id: "test-worker-id",
+        version: "test-version",
+        contentHash: "test-content-hash",
+        engine: "V2",
+        ...overrides,
+    };
+}
+/**
+ * Run a function inside a fully mocked task runtime context.
+ *
+ * Installs in-memory test managers for `locals`, `inputStreams`,
+ * `realtimeStreams`, `lifecycleHooks`, and `runtime`, sets a mock
+ * `TaskContext`, and tears everything down when the function returns.
+ *
+ * Inside the function, any code that reads from `locals`, `inputStreams`,
+ * `realtimeStreams`, or `taskContext.ctx` will see the mock context —
+ * so you can directly invoke the internal `run` function of any task
+ * (including `chat.agent`) without hitting the Trigger.dev runtime.
+ *
+ * @example
+ * ```ts
+ * import { runInMockTaskContext } from "@trigger.dev/core/v3/test";
+ *
+ * await runInMockTaskContext(
+ *   async ({ inputs, outputs, ctx }) => {
+ *     // Fire an input stream from the "outside"
+ *     setTimeout(() => {
+ *       inputs.send("chat-messages", { messages: [], chatId: "c1" });
+ *     }, 0);
+ *
+ *     // Run task code that reads from inputStreams.once(...)
+ *     await myTask.fns.run(payload, { ctx, signal: new AbortController().signal });
+ *
+ *     // Inspect chunks written to the output stream
+ *     expect(outputs.chunks("chat")).toContainEqual({ type: "text-delta", delta: "hi" });
+ *   },
+ *   { ctx: { run: { id: "run_abc" } } }
+ * );
+ * ```
+ */
+export async function runInMockTaskContext(fn, options) {
+    const ctx = defaultTaskRunContext(options?.ctx);
+    const worker = defaultWorker(options?.worker);
+    const localsManager = new StandardLocalsManager();
+    const lifecycleManager = new StandardLifecycleHooksManager();
+    const runtimeManager = new NoopRuntimeManager();
+    const metadataManager = new TestRunMetadataManager();
+    const inputManager = new TestInputStreamManager();
+    const outputManager = new TestRealtimeStreamsManager();
+    // Unregister any previously-installed managers so `setGlobal*` wins —
+    // `registerGlobal` returns false silently if an entry already exists.
+    unregisterGlobal("locals");
+    unregisterGlobal("lifecycle-hooks");
+    unregisterGlobal("runtime");
+    unregisterGlobal("run-metadata");
+    unregisterGlobal("input-streams");
+    unregisterGlobal("realtime-streams");
+    unregisterGlobal("task-context");
+    localsAPI.setGlobalLocalsManager(localsManager);
+    lifecycleHooks.setGlobalLifecycleHooksManager(lifecycleManager);
+    runtime.setGlobalRuntimeManager(runtimeManager);
+    runMetadata.setGlobalManager(metadataManager);
+    inputStreams.setGlobalManager(inputManager);
+    realtimeStreams.setGlobalManager(outputManager);
+    taskContext.setGlobalTaskContext({
+        ctx,
+        worker,
+        isWarmStart: options?.isWarmStart ?? false,
+    });
+    const drivers = {
+        inputs: {
+            send: (streamId, data) => inputManager.__sendFromTest(streamId, data),
+            close: (streamId) => inputManager.__closeFromTest(streamId),
+        },
+        outputs: {
+            chunks: (streamId) => outputManager.__chunksFromTest(streamId),
+            all: () => outputManager.__allChunksFromTest(),
+            clear: (streamId) => outputManager.__clearFromTest(streamId),
+            onWrite: (listener) => outputManager.onWrite(listener),
+        },
+        locals: {
+            get: (key) => localsManager.getLocal(key),
+            set: (key, value) => localsManager.setLocal(key, value),
+        },
+        ctx,
+    };
+    try {
+        return await fn(drivers);
+    }
+    finally {
+        localsAPI.disable();
+        lifecycleHooks.disable();
+        runtime.disable();
+        // taskContext.disable() only sets a flag — unregister the global so
+        // `taskContext.ctx` returns undefined after the harness returns.
+        unregisterGlobal("task-context");
+        unregisterGlobal("input-streams");
+        unregisterGlobal("realtime-streams");
+        unregisterGlobal("run-metadata");
+        localsManager.reset();
+        inputManager.reset();
+        outputManager.reset();
+        metadataManager.reset();
+    }
+}
+//# sourceMappingURL=mock-task-context.js.map

package/dist/esm/v3/test/mock-task-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock-task-context.js","sourceRoot":"","sources":["../../../../src/v3/test/mock-task-context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAC7D,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAC3D,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAC5C,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,EAAE,6BAA6B,EAAE,MAAM,8BAA8B,CAAC;AAC7E,OAAO,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAC;AACtE,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAGvD,OAAO,EAAE,sBAAsB,EAAE,MAAM,gCAAgC,CAAC;AACxE,OAAO,EAAE,0BAA0B,EAAE,MAAM,oCAAoC,CAAC;AAChF,OAAO,EAAE,sBAAsB,EAAE,MAAM,gCAAgC,CAAC;AA0ExE,SAAS,qBAAqB,CAAC,SAAuC;IACpE,OAAO;QACL,IAAI,EAAE;YACJ,EAAE,EAAE,WAAW;YACf,QAAQ,EAAE,cAAc;YACxB,GAAG,SAAS,EAAE,IAAI;SACnB;QACD,OAAO,EAAE;YACP,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,GAAG,SAAS,EAAE,OAAO;SACtB;QACD,GAAG,EAAE;YACH,EAAE,EAAE,UAAU;YACd,IAAI,EAAE,EAAE;YACR,MAAM,EAAE,KAAK;YACb,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,GAAG,SAAS,EAAE,GAAG;SAClB;QACD,OAAO,EAAE;YACP,IAAI,EAAE,OAAO;YACb,GAAG,EAAE,CAAC;YACN,MAAM,EAAE,GAAG;YACX,UAAU,EAAE,CAAC;YACb,GAAG,SAAS,EAAE,OAAO;SACtB;QACD,KAAK,EAAE;YACL,IAAI,EAAE,YAAY;YAClB,EAAE,EAAE,eAAe;YACnB,GAAG,SAAS,EAAE,KAAK;SACpB;QACD,WAAW,EAAE;YACX,EAAE,EAAE,aAAa;YACjB,IAAI,EAAE,UAAU;YAChB,IAAI,EAAE,aAAa;YACnB,GAAG,SAAS,EAAE,WAAW;SAC1B;QACD,YAAY,EAAE;YACZ,EAAE,EAAE,aAAa;YACjB,IAAI,EAAE,UAAU;YAChB,IAAI,EAAE,UAAU;YAChB,GAAG,SAAS,EAAE,YAAY;SAC3B;QACD,OAAO,EAAE;YACP,EAAE,EAAE,iBAAiB;YACrB,GAAG,EAAE,kBAAkB;YACvB,IAAI,EAAE,cAAc;YACpB,IAAI,EAAE,cAAc;YACpB,GAAG,SAAS,EAAE,OAAO;SACtB;QACD,KAAK,EAAE,SAAS,EAAE,KAAK;KACxB,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,SAA2C;IAChE,OAAO;QACL,EAAE,EAAE,gBAAgB;QACpB,OAAO,EAAE,cAAc;QACvB,WAAW,EAAE,mBAAmB;QAChC,MAAM,EAAE,IAAI;QACZ,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,EAAuD,EACvD,OAAgC;IAEhC,MAAM,GAAG,GAAG,qBAAqB,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IAChD,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAE9C,MAAM,aAAa,GAAG,IAAI,qBAAqB,EAAE,CAAC;IAClD,MAAM,gBAAgB,GAAG,IAAI,6BAA6B,EAAE,CAAC;IAC7D,MAAM,cAAc,GAAG,IAAI,kBAAkB,EAAE,CAAC;IAChD,MAAM,eAAe,GAAG,IAAI,sBAAsB,EAAE,CAAC;IACrD,MAAM,YAAY,GAAG,IAAI,sBAAsB,EAAE,CAAC;IAClD,MAAM,aAAa,GAAG,IAAI,0BAA0B,EAAE,CAAC;IAEvD,sEAAsE;IACtE,sEAAsE;IACtE,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC3B,gBAAgB,CAAC,iBAAiB,CAAC,CAAC;IACpC,gBAAgB,CAAC,SAAS,CAAC,CAAC;IAC5B,gBAAgB,CAAC,cAAc,CAAC,CAAC;IACjC,gBAAgB,CAAC,eAAe,CAAC,CAAC;IAClC,gBAAgB,CAAC,kBAAkB,CAAC,CAAC;IACrC,gBAAgB,CAAC,cAAc,CAAC,CAAC;IAEjC,SAAS,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAAC;IAChD,cAAc,CAAC,8BAA8B,CAAC,gBAAgB,CAAC,CAAC;IAChE,OAAO,CAAC,uBAAuB,CAAC,cAAc,CAAC,CAAC;IAChD,WAAW,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;IAC9C,YAAY,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;IAC5C,eAAe,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;IAChD,WAAW,CAAC,oBAAoB,CAAC;QAC/B,GAAG;QACH,MAAM;QACN,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,KAAK;KAC3C,CAAC,CAAC;IAEH,MAAM,OAAO,GAA2B;QACtC,MAAM,EAAE;YACN,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC;YACrE,KAAK,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,YAAY,CAAC,eAAe,CAAC,QAAQ,CAAC;SAC5D;QACD,OAAO,EAAE;YACP,MAAM,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,aAAa,CAAC,gBAAgB,CAAC,QAAQ,CAAC;YAC9D,GAAG,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,mBAAmB,EAAE;YAC9C,KAAK,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,aAAa,CAAC,eAAe,CAAC,QAAQ,CAAC;YAC5D,OAAO,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,aAAa,CAAC,OAAO,CAAC,QAAQ,CAAC;SACvD;QACD,MAAM,EAAE;YACN,GAAG,EAAE,CAAS,GAAsB,EAAE,EAAE,CAAC,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC;YACpE,GAAG,EAAE,CAAS,GAAsB,EAAE,KAAa,EAAE,EAAE,CACrD,aAAa,CAAC,QAAQ,CAAC,GAAG,EAAE,KAAK,CAAC;SACrC;QACD,GAAG;KACJ,CAAC;IAEF,IAAI,CAAC;QACH,OAAO,MAAM,EAAE,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;YAAS,CAAC;QACT,SAAS,CAAC,OAAO,EAAE,CAAC;QACpB,cAAc,CAAC,OAAO,EAAE,CAAC;QACzB,OAAO,CAAC,OAAO,EAAE,CAAC;QAClB,oEAAoE;QACpE,iEAAiE;QACjE,gBAAgB,CAAC,cAAc,CAAC,CAAC;QACjC,gBAAgB,CAAC,eAAe,CAAC,CAAC;QAClC,gBAAgB,CAAC,kBAAkB,CAAC,CAAC;QACrC,gBAAgB,CAAC,cAAc,CAAC,CAAC;QACjC,aAAa,CAAC,KAAK,EAAE,CAAC;QACtB,YAAY,CAAC,KAAK,EAAE,CAAC;QACrB,aAAa,CAAC,KAAK,EAAE,CAAC;QACtB,eAAe,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;AACH,CAAC"}

package/dist/esm/v3/test/test-input-stream-manager.d.ts

@@ -0,0 +1,46 @@
+import type { InputStreamManager } from "../inputStreams/types.js";
+import { InputStreamOncePromise } from "../inputStreams/types.js";
+import type { InputStreamOnceOptions } from "../realtimeStreams/types.js";
+type Handler = (data: unknown) => void | Promise<void>;
+/**
+ * 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 declare class TestInputStreamManager implements InputStreamManager {
+    private handlers;
+    private onceWaiters;
+    private latest;
+    private lastSeqNums;
+    private pendingSends;
+    setRunId(_runId: string, _streamsVersion?: string): void;
+    on(streamId: string, handler: Handler): {
+        off: () => void;
+    };
+    once(streamId: string, options?: InputStreamOnceOptions): InputStreamOncePromise<unknown>;
+    peek(streamId: string): unknown | undefined;
+    lastSeqNum(streamId: string): number | undefined;
+    setLastSeqNum(streamId: string, seqNum: number): void;
+    shiftBuffer(_streamId: string): boolean;
+    disconnectStream(_streamId: string): void;
+    clearHandlers(): void;
+    reset(): void;
+    disconnect(): void;
+    connectTail(_runId: string, _fromSeq?: number): void;
+    /**
+     * Push data onto an input stream. Resolves pending `once()` waiters
+     * and fires all `on()` handlers (awaiting async handlers).
+     */
+    __sendFromTest(streamId: string, data: unknown): Promise<void>;
+    /**
+     * Immediately resolve every pending `once()` waiter for a stream with a
+     * timeout error. Used to simulate closed streams (e.g. `exitAfterPreloadIdle`).
+     */
+    __closeFromTest(streamId: string): void;
+    private removeWaiter;
+}
+export {};