@trigger.dev/core 0.0.0-prerelease-20260508094307 → 0.0.0-prerelease-deployment-base-image-20260814120901

package/dist/commonjs/v3/test/session-waitpoint-backend.js

@@ -0,0 +1,228 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TestRuntimeManager = exports.SessionWaitpointBackend = void 0;
+exports.installSessionWaitpointBackend = installSessionWaitpointBackend;
+const index_js_1 = require("../apiClient/index.js");
+const friendlyId_js_1 = require("../isomorphic/friendlyId.js");
+const noopRuntimeManager_js_1 = require("../runtime/noopRuntimeManager.js");
+const TIMED_OUT = Symbol("session-waitpoint-timeout");
+function parseTimeoutMs(timeout) {
+    if (!timeout) {
+        return undefined;
+    }
+    const match = /^(\d+)(ms|s|m|h)$/.exec(timeout.trim());
+    if (!match) {
+        return undefined;
+    }
+    const value = Number(match[1]);
+    switch (match[2]) {
+        case "ms":
+            return value;
+        case "s":
+            return value * 1_000;
+        case "m":
+            return value * 60_000;
+        default:
+            return value * 3_600_000;
+    }
+}
+/**
+ * In-process stand-in for the run-engine's session-stream waitpoint machinery.
+ *
+ * `session.in.wait()` suspends a run on a waitpoint; in production the run-engine
+ * completes that waitpoint (and the supervisor resumes the run) when the next
+ * `.in` record arrives. This backend reproduces the task-observable half in one
+ * process: `register()` mints a waitpoint id when the SDK creates the waitpoint,
+ * and `wait()` opens its own short-lived tail on the session channel and resolves
+ * with the next record. That record is handed back through `TestRuntimeManager`
+ * exactly as the real runtime hands back a completed waitpoint's output, so the
+ * same `run()` invocation continues in place (the faithful dev / deployed
+ * task-observable semantics: same promise, same record, no re-invocation).
+ *
+ * It deliberately does not model server-side waitpoint bookkeeping, process
+ * checkpoint/restore, or client-vs-server timeout, none of which are visible to
+ * task code on the resume path.
+ */
+class SessionWaitpointBackend {
+    apiClient;
+    pending = new Map();
+    disabled = false;
+    constructor(apiClient) {
+        this.apiClient = apiClient;
+    }
+    register(body) {
+        const waitpointId = friendlyId_js_1.WaitpointId.generate().friendlyId;
+        this.pending.set(waitpointId, {
+            session: body.session,
+            io: body.io,
+            lastSeqNum: body.lastSeqNum,
+            timeout: body.timeout,
+            abort: new AbortController(),
+        });
+        return { waitpointId, isCached: false };
+    }
+    async wait(waitpointFriendlyId) {
+        if (this.disabled) {
+            return {
+                ok: false,
+                output: JSON.stringify({ message: "Session waitpoint backend disabled" }),
+                outputType: "application/json",
+            };
+        }
+        const pending = this.pending.get(waitpointFriendlyId);
+        if (!pending) {
+            return { ok: true };
+        }
+        const timeoutMs = parseTimeoutMs(pending.timeout);
+        let timer;
+        try {
+            const recordPromise = this.readNextRecord(pending);
+            const result = timeoutMs === undefined
+                ? await recordPromise
+                : await Promise.race([
+                    recordPromise,
+                    new Promise((resolve) => {
+                        timer = setTimeout(() => resolve(TIMED_OUT), timeoutMs);
+                    }),
+                ]);
+            if (result === TIMED_OUT) {
+                pending.abort.abort();
+                return {
+                    ok: false,
+                    output: JSON.stringify({ message: "Timed out" }),
+                    outputType: "application/json",
+                };
+            }
+            const output = typeof result === "string" ? result : JSON.stringify(result);
+            return { ok: true, output, outputType: "application/json" };
+        }
+        catch {
+            return {
+                ok: false,
+                output: JSON.stringify({ message: "Session stream wait ended before a record arrived" }),
+                outputType: "application/json",
+            };
+        }
+        finally {
+            if (timer) {
+                clearTimeout(timer);
+            }
+            this.pending.delete(waitpointFriendlyId);
+        }
+    }
+    disable() {
+        this.disabled = true;
+        for (const pending of this.pending.values()) {
+            pending.abort.abort();
+        }
+        this.pending.clear();
+    }
+    /**
+     * Opens an independent tail on the session channel starting after the
+     * caller's last-seen seq and resolves with the next record. For `.in` the
+     * tail yields the raw appended JSON string (already `JSON.stringify(chunk)`),
+     * which {@link wait} passes straight to the packet parser so it round-trips
+     * to the same object `session.in.once()` returns.
+     */
+    async readNextRecord(pending) {
+        const lastEventId = pending.lastSeqNum !== undefined && pending.lastSeqNum >= 0
+            ? String(pending.lastSeqNum)
+            : undefined;
+        const stream = await this.apiClient.subscribeToSessionStream(pending.session, pending.io, {
+            lastEventId,
+            signal: pending.abort.signal,
+            timeoutInSeconds: 120,
+        });
+        const reader = stream.getReader();
+        try {
+            const { value, done } = await reader.read();
+            if (done) {
+                throw new Error("session stream closed");
+            }
+            return value;
+        }
+        finally {
+            await reader.cancel().catch(() => { });
+            pending.abort.abort();
+        }
+    }
+}
+exports.SessionWaitpointBackend = SessionWaitpointBackend;
+/**
+ * A {@link NoopRuntimeManager} whose `waitForWaitpoint` resolves in place from a
+ * {@link SessionWaitpointBackend} instead of hanging. Install it as the `runtime`
+ * global via `runInMockTaskContext({ runtimeManager })`.
+ */
+class TestRuntimeManager extends noopRuntimeManager_js_1.NoopRuntimeManager {
+    backend;
+    constructor(backend) {
+        super();
+        this.backend = backend;
+    }
+    waitForWaitpoint(params) {
+        return this.backend.wait(params.waitpointFriendlyId);
+    }
+}
+exports.TestRuntimeManager = TestRuntimeManager;
+let activeBackend;
+let patchDepth = 0;
+let originalCreate;
+let originalWait;
+/**
+ * Route the two waitpoint apiClient calls through an in-process backend for the
+ * duration of a harness run, and hand back a {@link TestRuntimeManager} to
+ * install as the `runtime` global.
+ *
+ * `apiClientManager.clientOrThrow()` builds a fresh `ApiClient` per call, so
+ * there is no instance to swap; the two methods are patched on the prototype
+ * (ref-counted, restored by `restore()`). `createSessionStreamWaitpoint`
+ * registers a pending wait and returns a real-shaped `{ waitpointId, isCached }`;
+ * `waitForWaitpointToken` returns `{ success: true }` exactly like the real route,
+ * which only creates the block edge and returns synchronously.
+ */
+function installSessionWaitpointBackend(apiClient) {
+    const backend = new SessionWaitpointBackend(apiClient);
+    const previousBackend = activeBackend;
+    activeBackend = backend;
+    if (patchDepth === 0) {
+        originalCreate = index_js_1.ApiClient.prototype.createSessionStreamWaitpoint;
+        originalWait = index_js_1.ApiClient.prototype.waitForWaitpointToken;
+        index_js_1.ApiClient.prototype.createSessionStreamWaitpoint = function (runFriendlyId, body) {
+            if (activeBackend) {
+                return Promise.resolve(activeBackend.register(body));
+            }
+            return originalCreate.call(this, runFriendlyId, body);
+        };
+        index_js_1.ApiClient.prototype.waitForWaitpointToken = function () {
+            if (activeBackend) {
+                return Promise.resolve({ success: true });
+            }
+            // eslint-disable-next-line prefer-rest-params
+            return originalWait.apply(this, arguments);
+        };
+    }
+    patchDepth += 1;
+    let restored = false;
+    return {
+        backend,
+        runtimeManager: new TestRuntimeManager(backend),
+        restore() {
+            if (restored) {
+                return;
+            }
+            restored = true;
+            backend.disable();
+            if (activeBackend === backend) {
+                activeBackend = previousBackend;
+            }
+            patchDepth -= 1;
+            if (patchDepth === 0 && originalCreate && originalWait) {
+                index_js_1.ApiClient.prototype.createSessionStreamWaitpoint = originalCreate;
+                index_js_1.ApiClient.prototype.waitForWaitpointToken = originalWait;
+                originalCreate = undefined;
+                originalWait = undefined;
+            }
+        },
+    };
+}
+//# sourceMappingURL=session-waitpoint-backend.js.map

package/dist/commonjs/v3/test/session-waitpoint-backend.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-waitpoint-backend.js","sourceRoot":"","sources":["../../../../src/v3/test/session-waitpoint-backend.ts"],"names":[],"mappings":";;;;AAAA,oDAAkD;AAClD,+DAA0D;AAC1D,4EAAsE;AAgBtE,MAAM,SAAS,GAAG,MAAM,CAAC,2BAA2B,CAAC,CAAC;AAEtD,SAAS,cAAc,CAAC,OAA2B;IACjD,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,KAAK,GAAG,mBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IACvD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC/B,QAAQ,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QACjB,KAAK,IAAI;YACP,OAAO,KAAK,CAAC;QACf,KAAK,GAAG;YACN,OAAO,KAAK,GAAG,KAAK,CAAC;QACvB,KAAK,GAAG;YACN,OAAO,KAAK,GAAG,MAAM,CAAC;QACxB;YACE,OAAO,KAAK,GAAG,SAAS,CAAC;IAC7B,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;;;;GAgBG;AACH;IAI+B,SAAS;IAHrB,OAAO,GAAG,IAAI,GAAG,EAAuB,CAAC;IAClD,QAAQ,GAAG,KAAK,CAAC;IAEzB,YAA6B,SAAoB;yBAApB,SAAS;IAAc,CAAC;IAErD,QAAQ,CACN,IAA6C;QAE7C,MAAM,WAAW,GAAG,2BAAW,CAAC,QAAQ,EAAE,CAAC,UAAU,CAAC;QACtD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE;YAC5B,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,KAAK,EAAE,IAAI,eAAe,EAAE;SAC7B,CAAC,CAAC;QACH,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,mBAA2B;QACpC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,oCAAoC,EAAE,CAAC;gBACzE,UAAU,EAAE,kBAAkB;aAC/B,CAAC;QACJ,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QACtD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;QACtB,CAAC;QAED,MAAM,SAAS,GAAG,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAClD,IAAI,KAAgD,CAAC;QAErD,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YACnD,MAAM,MAAM,GACV,SAAS,KAAK,SAAS;gBACrB,CAAC,CAAC,MAAM,aAAa;gBACrB,CAAC,CAAC,MAAM,OAAO,CAAC,IAAI,CAAC;oBACjB,aAAa;oBACb,IAAI,OAAO,CAAmB,CAAC,OAAO,EAAE,EAAE;wBACxC,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC,CAAC;oBAC1D,CAAC,CAAC;iBACH,CAAC,CAAC;YAET,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBACzB,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;gBACtB,OAAO;oBACL,EAAE,EAAE,KAAK;oBACT,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;oBAChD,UAAU,EAAE,kBAAkB;iBAC/B,CAAC;YACJ,CAAC;YAED,MAAM,MAAM,GAAG,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAC5E,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,kBAAkB,EAAE,CAAC;QAC9D,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,mDAAmD,EAAE,CAAC;gBACxF,UAAU,EAAE,kBAAkB;aAC/B,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,IAAI,KAAK,EAAE,CAAC;gBACV,YAAY,CAAC,KAAK,CAAC,CAAC;YACtB,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,OAAO;QACL,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC5C,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACxB,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED;;;;;;OAMG;IACK,KAAK,CAAC,cAAc,CAAC,OAAoB;QAC/C,MAAM,WAAW,GACf,OAAO,CAAC,UAAU,KAAK,SAAS,IAAI,OAAO,CAAC,UAAU,IAAI,CAAC;YACzD,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC;YAC5B,CAAC,CAAC,SAAS,CAAC;QAEhB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,wBAAwB,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,EAAE;YACxF,WAAW;YACX,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,MAAM;YAC5B,gBAAgB,EAAE,GAAG;SACtB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;QAClC,IAAI,CAAC;YACH,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAC3C,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC;gBAAS,CAAC;YACT,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACtC,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACxB,CAAC;IACH,CAAC;CACF;;AAED;;;;GAIG;AACH,wBAAgC,SAAQ,0CAAkB;IAC3B,OAAO;IAApC,YAA6B,OAAgC;QAC3D,KAAK,EAAE,CAAC;uBADmB,OAAO;IAEpC,CAAC;IAEQ,gBAAgB,CAAC,MAGzB;QACC,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;IACvD,CAAC;CACF;;AAED,IAAI,aAAkD,CAAC;AACvD,IAAI,UAAU,GAAG,CAAC,CAAC;AACnB,IAAI,cAAqE,CAAC;AAC1E,IAAI,YAA4D,CAAC;AAEjE;;;;;;;;;;;GAWG;AACH,wCAA+C,SAAoB;IAKjE,MAAM,OAAO,GAAG,IAAI,uBAAuB,CAAC,SAAS,CAAC,CAAC;IACvD,MAAM,eAAe,GAAG,aAAa,CAAC;IACtC,aAAa,GAAG,OAAO,CAAC;IAExB,IAAI,UAAU,KAAK,CAAC,EAAE,CAAC;QACrB,cAAc,GAAG,oBAAS,CAAC,SAAS,CAAC,4BAA4B,CAAC;QAClE,YAAY,GAAG,oBAAS,CAAC,SAAS,CAAC,qBAAqB,CAAC;QAEzD,oBAAS,CAAC,SAAS,CAAC,4BAA4B,GAAG,UAEjD,aAAqB,EACrB,IAA6C;YAE7C,IAAI,aAAa,EAAE,CAAC;gBAClB,OAAO,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;YACvD,CAAC;YACD,OAAO,cAAe,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,EAAE,IAAI,CAAC,CAAC;QACzD,CAAyD,CAAC;QAE1D,oBAAS,CAAC,SAAS,CAAC,qBAAqB,GAAG;YAG1C,IAAI,aAAa,EAAE,CAAC;gBAClB,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5C,CAAC;YACD,8CAA8C;YAC9C,OAAQ,YAAmD,CAAC,KAAK,CAAC,IAAI,EAAE,SAAkB,CAAC,CAAC;QAC9F,CAAkD,CAAC;IACrD,CAAC;IAED,UAAU,IAAI,CAAC,CAAC;IAChB,IAAI,QAAQ,GAAG,KAAK,CAAC;IAErB,OAAO;QACL,OAAO;QACP,cAAc,EAAE,IAAI,kBAAkB,CAAC,OAAO,CAAC;QAC/C,OAAO;YACL,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO;YACT,CAAC;YACD,QAAQ,GAAG,IAAI,CAAC;YAChB,OAAO,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,aAAa,KAAK,OAAO,EAAE,CAAC;gBAC9B,aAAa,GAAG,eAAe,CAAC;YAClC,CAAC;YACD,UAAU,IAAI,CAAC,CAAC;YAChB,IAAI,UAAU,KAAK,CAAC,IAAI,cAAc,IAAI,YAAY,EAAE,CAAC;gBACvD,oBAAS,CAAC,SAAS,CAAC,4BAA4B,GAAG,cAAc,CAAC;gBAClE,oBAAS,CAAC,SAAS,CAAC,qBAAqB,GAAG,YAAY,CAAC;gBACzD,cAAc,GAAG,SAAS,CAAC;gBAC3B,YAAY,GAAG,SAAS,CAAC;YAC3B,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/commonjs/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 {};

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

@@ -0,0 +1,192 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TestInputStreamManager = void 0;
+const types_js_1 = require("../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.
+ */
+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 types_js_1.InputStreamOncePromise((resolve) => {
+            if (options?.signal?.aborted) {
+                resolve({
+                    ok: false,
+                    error: new types_js_1.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 types_js_1.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 types_js_1.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 types_js_1.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);
+    }
+}
+exports.TestInputStreamManager = TestInputStreamManager;
+//# sourceMappingURL=test-input-stream-manager.js.map

package/dist/commonjs/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,uDAA2F;AAY3F;;;;;;;;GAQG;AACH;IACU,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,iCAAsB,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,kCAAuB,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,kCAAuB,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,kCAAuB,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,CAAC,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,CAAC,CAAC;QAC7F,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,kCAAuB,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/commonjs/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/commonjs/v3/test/test-realtime-streams-manager.js

@@ -0,0 +1,141 @@
+"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);
+        // eslint-disable-next-line no-this-alias
+        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();
+        this.writeListeners.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