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

package/dist/commonjs/v3/telnetLogServer.test.js

@@ -0,0 +1,144 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const node_net_1 = __importDefault(require("node:net"));
+const vitest_1 = require("vitest");
+const telnetLogServer_js_1 = require("./telnetLogServer.js");
+const servers = [];
+(0, vitest_1.afterEach)(() => {
+    while (servers.length) {
+        servers.pop()?.close();
+    }
+});
+/** Grab a port the OS just told us is free, then hand it to a TelnetLogServer. */
+async function startServerOnFreePort(name = "test") {
+    const probe = node_net_1.default.createServer();
+    const port = await listening(probe, 0);
+    await new Promise((resolve) => probe.close(() => resolve()));
+    // A banner lets clients deterministically wait until the server has registered them
+    // (it's written in the connection handler), removing the connect/register race.
+    const server = new telnetLogServer_js_1.TelnetLogServer({ port, name, banner: "ready" });
+    servers.push(server);
+    server.start();
+    await delay(30);
+    return { server, port };
+}
+/** Connects and resolves only once the first bytes (the banner) arrive — server-side socket is registered by then. */
+function connectAndCollect(port) {
+    return new Promise((resolve, reject) => {
+        let buffer = "";
+        const socket = node_net_1.default.connect(port, "127.0.0.1");
+        socket.setEncoding("utf8");
+        socket.on("data", (chunk) => {
+            const first = buffer === "";
+            buffer += chunk;
+            if (first) {
+                resolve({ socket, lines: () => buffer });
+            }
+        });
+        socket.on("error", reject);
+    });
+}
+function listening(server, port, host = "127.0.0.1") {
+    return new Promise((resolve) => {
+        server.listen(port, host, () => {
+            const address = server.address();
+            resolve(typeof address === "object" && address ? address.port : port);
+        });
+    });
+}
+(0, vitest_1.describe)("stripAnsi", () => {
+    (0, vitest_1.test)("removes color escape codes", () => {
+        const colored = "2026-06-11 ERROR boom";
+        (0, vitest_1.expect)((0, telnetLogServer_js_1.stripAnsi)(colored)).toBe("2026-06-11 ERROR boom");
+    });
+});
+(0, vitest_1.describe)("formatLogLine", () => {
+    (0, vitest_1.test)("formats a core Logger-shaped object (level/name)", () => {
+        const line = (0, telnetLogServer_js_1.formatLogLine)({
+            timestamp: new Date("2026-06-11T12:00:00.000Z"),
+            name: "webapp",
+            level: "info",
+            message: "queue drained",
+            count: 3,
+        });
+        (0, vitest_1.expect)(line).toBe("2026-06-11T12:00:00.000Z INFO  [webapp] queue drained {count=3}");
+    });
+    (0, vitest_1.test)("formats a SimpleStructuredLogger-shaped object ($level/$name)", () => {
+        const line = (0, telnetLogServer_js_1.formatLogLine)({
+            timestamp: new Date("2026-06-11T12:00:00.000Z"),
+            $name: "supervisor",
+            $level: "warn",
+            message: "retrying",
+            attempt: 2,
+        });
+        (0, vitest_1.expect)(line).toBe("2026-06-11T12:00:00.000Z WARN  [supervisor] retrying {attempt=2}");
+    });
+    (0, vitest_1.test)("omits empty name and extras", () => {
+        const line = (0, telnetLogServer_js_1.formatLogLine)({
+            timestamp: "2026-06-11T12:00:00.000Z",
+            level: "log",
+            message: "hello",
+        });
+        (0, vitest_1.expect)(line).toBe("2026-06-11T12:00:00.000Z LOG   hello");
+    });
+});
+(0, vitest_1.describe)("formatConsoleLine", () => {
+    (0, vitest_1.test)("pretty-formats a JSON structured-log line", () => {
+        const raw = JSON.stringify({
+            timestamp: "2026-06-11T12:00:00.000Z",
+            name: "sso-plugin",
+            level: "info",
+            message: "sso.webhook.connection.activated: connection marked active",
+            connId: "conn_123",
+        });
+        (0, vitest_1.expect)((0, telnetLogServer_js_1.formatConsoleLine)(raw)).toBe("2026-06-11T12:00:00.000Z INFO  [sso-plugin] sso.webhook.connection.activated: connection marked active {connId=conn_123}");
+    });
+    (0, vitest_1.test)("passes non-JSON console output through unchanged", () => {
+        (0, vitest_1.expect)((0, telnetLogServer_js_1.formatConsoleLine)("GET /healthcheck 200 1.2 ms")).toBe("GET /healthcheck 200 1.2 ms");
+    });
+    (0, vitest_1.test)("passes JSON that isn't a structured log through unchanged", () => {
+        const raw = JSON.stringify({ foo: "bar" });
+        (0, vitest_1.expect)((0, telnetLogServer_js_1.formatConsoleLine)(raw)).toBe(raw);
+    });
+});
+(0, vitest_1.describe)("TelnetLogServer", () => {
+    (0, vitest_1.test)("broadcasts a line to a connected client", async () => {
+        const { server, port } = await startServerOnFreePort();
+        const client = await connectAndCollect(port);
+        server.broadcast("first line");
+        await delay(50);
+        (0, vitest_1.expect)(client.lines()).toContain("first line\r\n");
+        client.socket.destroy();
+    });
+    (0, vitest_1.test)("close() ends connected sockets", async () => {
+        const { server, port } = await startServerOnFreePort();
+        const client = await connectAndCollect(port);
+        server.broadcast("alive");
+        await delay(30);
+        (0, vitest_1.expect)(client.lines()).toContain("alive\r\n");
+        let closed = false;
+        client.socket.on("close", () => {
+            closed = true;
+        });
+        server.close();
+        await delay(30);
+        (0, vitest_1.expect)(closed).toBe(true);
+    });
+    (0, vitest_1.test)("EADDRINUSE does not throw", async () => {
+        const blocker = node_net_1.default.createServer();
+        const port = await listening(blocker, 0);
+        const server = new telnetLogServer_js_1.TelnetLogServer({ port, name: "test" });
+        servers.push(server);
+        // Must not throw even though the port is taken.
+        (0, vitest_1.expect)(() => server.start()).not.toThrow();
+        await delay(30);
+        blocker.close();
+    });
+});
+function delay(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+//# sourceMappingURL=telnetLogServer.test.js.map

package/dist/commonjs/v3/telnetLogServer.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"telnetLogServer.test.js","sourceRoot":"","sources":["../../../src/v3/telnetLogServer.test.ts"],"names":[],"mappings":";;;;;AAAA,wDAA2B;AAC3B,mCAA2D;AAC3D,6DAAoG;AAEpG,MAAM,OAAO,GAAsB,EAAE,CAAC;AAEtC,IAAA,kBAAS,EAAC,GAAG,EAAE;IACb,OAAO,OAAO,CAAC,MAAM,EAAE,CAAC;QACtB,OAAO,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC;IACzB,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,kFAAkF;AAClF,KAAK,UAAU,qBAAqB,CAClC,IAAI,GAAG,MAAM;IAEb,MAAM,KAAK,GAAG,kBAAG,CAAC,YAAY,EAAE,CAAC;IACjC,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IACvC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAEnE,oFAAoF;IACpF,gFAAgF;IAChF,MAAM,MAAM,GAAG,IAAI,oCAAe,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;IACpE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACrB,MAAM,CAAC,KAAK,EAAE,CAAC;IACf,MAAM,KAAK,CAAC,EAAE,CAAC,CAAC;IAChB,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;AAC1B,CAAC;AAED,sHAAsH;AACtH,SAAS,iBAAiB,CAAC,IAAY;IACrC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,MAAM,MAAM,GAAG,kBAAG,CAAC,OAAO,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QAC9C,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC3B,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;YAC1B,MAAM,KAAK,GAAG,MAAM,KAAK,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC;YAChB,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,SAAS,CAAC,MAAkB,EAAE,IAAY,EAAE,IAAI,GAAG,WAAW;IACrE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC7B,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE;YAC7B,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YACjC,OAAO,CAAC,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED,IAAA,iBAAQ,EAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAA,aAAI,EAAC,4BAA4B,EAAE,GAAG,EAAE;QACtC,MAAM,OAAO,GAAG,2CAA2C,CAAC;QAC5D,IAAA,eAAM,EAAC,IAAA,8BAAS,EAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAA,iBAAQ,EAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAA,aAAI,EAAC,kDAAkD,EAAE,GAAG,EAAE;QAC5D,MAAM,IAAI,GAAG,IAAA,kCAAa,EAAC;YACzB,SAAS,EAAE,IAAI,IAAI,CAAC,0BAA0B,CAAC;YAC/C,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE,MAAM;YACb,OAAO,EAAE,eAAe;YACxB,KAAK,EAAE,CAAC;SACT,CAAC,CAAC;QACH,IAAA,eAAM,EAAC,IAAI,CAAC,CAAC,IAAI,CAAC,iEAAiE,CAAC,CAAC;IACvF,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,+DAA+D,EAAE,GAAG,EAAE;QACzE,MAAM,IAAI,GAAG,IAAA,kCAAa,EAAC;YACzB,SAAS,EAAE,IAAI,IAAI,CAAC,0BAA0B,CAAC;YAC/C,KAAK,EAAE,YAAY;YACnB,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,UAAU;YACnB,OAAO,EAAE,CAAC;SACX,CAAC,CAAC;QACH,IAAA,eAAM,EAAC,IAAI,CAAC,CAAC,IAAI,CAAC,kEAAkE,CAAC,CAAC;IACxF,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,6BAA6B,EAAE,GAAG,EAAE;QACvC,MAAM,IAAI,GAAG,IAAA,kCAAa,EAAC;YACzB,SAAS,EAAE,0BAA0B;YACrC,KAAK,EAAE,KAAK;YACZ,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QACH,IAAA,eAAM,EAAC,IAAI,CAAC,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAA,iBAAQ,EAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,IAAA,aAAI,EAAC,2CAA2C,EAAE,GAAG,EAAE;QACrD,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC;YACzB,SAAS,EAAE,0BAA0B;YACrC,IAAI,EAAE,YAAY;YAClB,KAAK,EAAE,MAAM;YACb,OAAO,EAAE,4DAA4D;YACrE,MAAM,EAAE,UAAU;SACnB,CAAC,CAAC;QACH,IAAA,eAAM,EAAC,IAAA,sCAAiB,EAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CACjC,0HAA0H,CAC3H,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,kDAAkD,EAAE,GAAG,EAAE;QAC5D,IAAA,eAAM,EAAC,IAAA,sCAAiB,EAAC,6BAA6B,CAAC,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;IAC/F,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,2DAA2D,EAAE,GAAG,EAAE;QACrE,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3C,IAAA,eAAM,EAAC,IAAA,sCAAiB,EAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAA,iBAAQ,EAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAA,aAAI,EAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,qBAAqB,EAAE,CAAC;QACvD,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAC7C,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAC/B,MAAM,KAAK,CAAC,EAAE,CAAC,CAAC;QAChB,IAAA,eAAM,EAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,qBAAqB,EAAE,CAAC;QACvD,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAC7C,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAC1B,MAAM,KAAK,CAAC,EAAE,CAAC,CAAC;QAChB,IAAA,eAAM,EAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAE9C,IAAI,MAAM,GAAG,KAAK,CAAC;QACnB,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAC7B,MAAM,GAAG,IAAI,CAAC;QAChB,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,KAAK,CAAC,EAAE,CAAC,CAAC;QAChB,IAAA,eAAM,EAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,OAAO,GAAG,kBAAG,CAAC,YAAY,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAEzC,MAAM,MAAM,GAAG,IAAI,oCAAe,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAC3D,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrB,gDAAgD;QAChD,IAAA,eAAM,EAAC,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QAC3C,MAAM,KAAK,CAAC,EAAE,CAAC,CAAC;QAEhB,OAAO,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,SAAS,KAAK,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC"}

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

@@ -0,0 +1,8 @@
+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";
+export { TestSessionStreamManager } from "./test-session-stream-manager.js";
+export { StandardSessionStreamManager } from "../sessionStreams/manager.js";
+export type { SessionStreamManager } from "../sessionStreams/types.js";
+export { SessionWaitpointBackend, TestRuntimeManager, installSessionWaitpointBackend, } from "./session-waitpoint-backend.js";

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

@@ -0,0 +1,20 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.installSessionWaitpointBackend = exports.TestRuntimeManager = exports.SessionWaitpointBackend = exports.StandardSessionStreamManager = exports.TestSessionStreamManager = exports.TestRunMetadataManager = exports.TestRealtimeStreamsManager = exports.TestInputStreamManager = exports.runInMockTaskContext = void 0;
+var mock_task_context_js_1 = require("./mock-task-context.js");
+Object.defineProperty(exports, "runInMockTaskContext", { enumerable: true, get: function () { return mock_task_context_js_1.runInMockTaskContext; } });
+var test_input_stream_manager_js_1 = require("./test-input-stream-manager.js");
+Object.defineProperty(exports, "TestInputStreamManager", { enumerable: true, get: function () { return test_input_stream_manager_js_1.TestInputStreamManager; } });
+var test_realtime_streams_manager_js_1 = require("./test-realtime-streams-manager.js");
+Object.defineProperty(exports, "TestRealtimeStreamsManager", { enumerable: true, get: function () { return test_realtime_streams_manager_js_1.TestRealtimeStreamsManager; } });
+var test_run_metadata_manager_js_1 = require("./test-run-metadata-manager.js");
+Object.defineProperty(exports, "TestRunMetadataManager", { enumerable: true, get: function () { return test_run_metadata_manager_js_1.TestRunMetadataManager; } });
+var test_session_stream_manager_js_1 = require("./test-session-stream-manager.js");
+Object.defineProperty(exports, "TestSessionStreamManager", { enumerable: true, get: function () { return test_session_stream_manager_js_1.TestSessionStreamManager; } });
+var manager_js_1 = require("../sessionStreams/manager.js");
+Object.defineProperty(exports, "StandardSessionStreamManager", { enumerable: true, get: function () { return manager_js_1.StandardSessionStreamManager; } });
+var session_waitpoint_backend_js_1 = require("./session-waitpoint-backend.js");
+Object.defineProperty(exports, "SessionWaitpointBackend", { enumerable: true, get: function () { return session_waitpoint_backend_js_1.SessionWaitpointBackend; } });
+Object.defineProperty(exports, "TestRuntimeManager", { enumerable: true, get: function () { return session_waitpoint_backend_js_1.TestRuntimeManager; } });
+Object.defineProperty(exports, "installSessionWaitpointBackend", { enumerable: true, get: function () { return session_waitpoint_backend_js_1.installSessionWaitpointBackend; } });
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/v3/test/index.ts"],"names":[],"mappings":";;;AAAA,+DAIgC;AAH9B,4HAAA,oBAAoB,OAAA;AAItB,+EAAwE;AAA/D,sIAAA,sBAAsB,OAAA;AAC/B,uFAAgF;AAAvE,8IAAA,0BAA0B,OAAA;AACnC,+EAAwE;AAA/D,sIAAA,sBAAsB,OAAA;AAC/B,mFAA4E;AAAnE,0IAAA,wBAAwB,OAAA;AACjC,2DAA4E;AAAnE,0HAAA,4BAA4B,OAAA;AAErC,+EAIwC;AAHtC,uIAAA,uBAAuB,OAAA;AACvB,kIAAA,kBAAkB,OAAA;AAClB,8IAAA,8BAA8B,OAAA"}

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

@@ -0,0 +1,138 @@
+import type { RuntimeManager } from "../runtime/manager.js";
+import type { ServerBackgroundWorker, TaskRunContext } from "../schemas/index.js";
+import type { LocalsKey } from "../locals/types.js";
+import type { SessionChannelIO, SessionStreamManager } from "../sessionStreams/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;
+    /**
+     * Session-streams manager installed as the `session-streams` global. Defaults
+     * to an in-memory {@link TestSessionStreamManager}. Pass a real
+     * `StandardSessionStreamManager` (with an ApiClient pointed at a running
+     * webapp) to drive the task's `.in`/`.out` against real streams.
+     */
+    sessionStreamManager?: SessionStreamManager;
+    /**
+     * Runtime manager installed as the `runtime` global. Defaults to a
+     * {@link NoopRuntimeManager}. Pass a `TestRuntimeManager` (wired to a
+     * {@link SessionWaitpointBackend}) to make `session.in.wait()` suspend and
+     * resume in place against real streams, without the run-engine.
+     */
+    runtimeManager?: RuntimeManager;
+};
+/**
+ * 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;
+    };
+    /**
+     * Session-scoped channel drivers. The `.in` side is backed by a
+     * {@link TestSessionStreamManager} installed as the `sessionStreams`
+     * global — so the task's `session.in.on/once/peek/waitWithIdleTimeout`
+     * calls receive records sent through this driver.
+     */
+    sessions: {
+        in: {
+            /**
+             * Send a record onto `session.in` for the given session. Resolves
+             * pending `once()` waiters and fires all `on()` handlers.
+             */
+            send(sessionId: string, data: unknown, io?: SessionChannelIO): Promise<void>;
+            /** Close pending `once()` waiters with a timeout error. */
+            close(sessionId: string, io?: SessionChannelIO): 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/commonjs/v3/test/mock-task-context.js

@@ -0,0 +1,201 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runInMockTaskContext = runInMockTaskContext;
+const input_streams_api_js_1 = require("../input-streams-api.js");
+const realtime_streams_api_js_1 = require("../realtime-streams-api.js");
+const session_streams_api_js_1 = require("../session-streams-api.js");
+const locals_api_js_1 = require("../locals-api.js");
+const run_metadata_api_js_1 = require("../run-metadata-api.js");
+const task_context_api_js_1 = require("../task-context-api.js");
+const lifecycle_hooks_api_js_1 = require("../lifecycle-hooks-api.js");
+const runtime_api_js_1 = require("../runtime-api.js");
+const manager_js_1 = require("../locals/manager.js");
+const manager_js_2 = require("../lifecycleHooks/manager.js");
+const noopRuntimeManager_js_1 = require("../runtime/noopRuntimeManager.js");
+const globals_js_1 = require("../utils/globals.js");
+const test_input_stream_manager_js_1 = require("./test-input-stream-manager.js");
+const test_realtime_streams_manager_js_1 = require("./test-realtime-streams-manager.js");
+const test_run_metadata_manager_js_1 = require("./test-run-metadata-manager.js");
+const test_session_stream_manager_js_1 = require("./test-session-stream-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,
+            isReplay: 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" } } }
+ * );
+ * ```
+ */
+async function runInMockTaskContext(fn, options) {
+    const ctx = defaultTaskRunContext(options?.ctx);
+    const worker = defaultWorker(options?.worker);
+    const localsManager = new manager_js_1.StandardLocalsManager();
+    const lifecycleManager = new manager_js_2.StandardLifecycleHooksManager();
+    const runtimeManager = options?.runtimeManager ?? new noopRuntimeManager_js_1.NoopRuntimeManager();
+    const metadataManager = new test_run_metadata_manager_js_1.TestRunMetadataManager();
+    const inputManager = new test_input_stream_manager_js_1.TestInputStreamManager();
+    const outputManager = new test_realtime_streams_manager_js_1.TestRealtimeStreamsManager();
+    const sessionStreamManager = options?.sessionStreamManager ?? new test_session_stream_manager_js_1.TestSessionStreamManager();
+    // Unregister any previously-installed managers so `setGlobal*` wins —
+    // `registerGlobal` returns false silently if an entry already exists.
+    (0, globals_js_1.unregisterGlobal)("locals");
+    (0, globals_js_1.unregisterGlobal)("lifecycle-hooks");
+    (0, globals_js_1.unregisterGlobal)("runtime");
+    (0, globals_js_1.unregisterGlobal)("run-metadata");
+    (0, globals_js_1.unregisterGlobal)("input-streams");
+    (0, globals_js_1.unregisterGlobal)("realtime-streams");
+    (0, globals_js_1.unregisterGlobal)("session-streams");
+    (0, globals_js_1.unregisterGlobal)("task-context");
+    locals_api_js_1.localsAPI.setGlobalLocalsManager(localsManager);
+    lifecycle_hooks_api_js_1.lifecycleHooks.setGlobalLifecycleHooksManager(lifecycleManager);
+    runtime_api_js_1.runtime.setGlobalRuntimeManager(runtimeManager);
+    run_metadata_api_js_1.runMetadata.setGlobalManager(metadataManager);
+    input_streams_api_js_1.inputStreams.setGlobalManager(inputManager);
+    realtime_streams_api_js_1.realtimeStreams.setGlobalManager(outputManager);
+    session_streams_api_js_1.sessionStreams.setGlobalManager(sessionStreamManager);
+    task_context_api_js_1.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),
+        },
+        sessions: {
+            in: {
+                send: (sessionId, data, io = "in") => sessionStreamManager instanceof test_session_stream_manager_js_1.TestSessionStreamManager
+                    ? sessionStreamManager.__sendFromTest(sessionId, io, data)
+                    : Promise.reject(new Error("drivers.sessions.in.send requires the default TestSessionStreamManager")),
+                close: (sessionId, io = "in") => {
+                    if (sessionStreamManager instanceof test_session_stream_manager_js_1.TestSessionStreamManager) {
+                        sessionStreamManager.__closeFromTest(sessionId, io);
+                    }
+                },
+            },
+        },
+        ctx,
+    };
+    try {
+        return await fn(drivers);
+    }
+    finally {
+        locals_api_js_1.localsAPI.disable();
+        lifecycle_hooks_api_js_1.lifecycleHooks.disable();
+        runtime_api_js_1.runtime.disable();
+        // taskContext.disable() only sets a flag — unregister the global so
+        // `taskContext.ctx` returns undefined after the harness returns.
+        (0, globals_js_1.unregisterGlobal)("task-context");
+        (0, globals_js_1.unregisterGlobal)("input-streams");
+        (0, globals_js_1.unregisterGlobal)("realtime-streams");
+        (0, globals_js_1.unregisterGlobal)("session-streams");
+        (0, globals_js_1.unregisterGlobal)("run-metadata");
+        localsManager.reset();
+        inputManager.reset();
+        outputManager.reset();
+        if (sessionStreamManager instanceof test_session_stream_manager_js_1.TestSessionStreamManager) {
+            sessionStreamManager.reset();
+        }
+        metadataManager.reset();
+    }
+}
+//# sourceMappingURL=mock-task-context.js.map

package/dist/commonjs/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,kEAAuD;AACvD,wEAA6D;AAC7D,sEAA2D;AAC3D,oDAA6C;AAC7C,gEAAqD;AACrD,gEAAqD;AACrD,sEAA2D;AAC3D,sDAA4C;AAC5C,qDAA6D;AAC7D,6DAA6E;AAC7E,4EAAsE;AAEtE,oDAAuD;AAIvD,iFAAwE;AACxE,yFAAgF;AAChF,iFAAwE;AACxE,qFAA4E;AAyG5E,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,QAAQ,EAAE,KAAK;YACf,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;AACI,KAAK,+BACV,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,kCAAqB,EAAE,CAAC;IAClD,MAAM,gBAAgB,GAAG,IAAI,0CAA6B,EAAE,CAAC;IAC7D,MAAM,cAAc,GAAG,OAAO,EAAE,cAAc,IAAI,IAAI,0CAAkB,EAAE,CAAC;IAC3E,MAAM,eAAe,GAAG,IAAI,qDAAsB,EAAE,CAAC;IACrD,MAAM,YAAY,GAAG,IAAI,qDAAsB,EAAE,CAAC;IAClD,MAAM,aAAa,GAAG,IAAI,6DAA0B,EAAE,CAAC;IACvD,MAAM,oBAAoB,GAAG,OAAO,EAAE,oBAAoB,IAAI,IAAI,yDAAwB,EAAE,CAAC;IAE7F,sEAAsE;IACtE,sEAAsE;IACtE,IAAA,6BAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3B,IAAA,6BAAgB,EAAC,iBAAiB,CAAC,CAAC;IACpC,IAAA,6BAAgB,EAAC,SAAS,CAAC,CAAC;IAC5B,IAAA,6BAAgB,EAAC,cAAc,CAAC,CAAC;IACjC,IAAA,6BAAgB,EAAC,eAAe,CAAC,CAAC;IAClC,IAAA,6BAAgB,EAAC,kBAAkB,CAAC,CAAC;IACrC,IAAA,6BAAgB,EAAC,iBAAiB,CAAC,CAAC;IACpC,IAAA,6BAAgB,EAAC,cAAc,CAAC,CAAC;IAEjC,yBAAS,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAAC;IAChD,uCAAc,CAAC,8BAA8B,CAAC,gBAAgB,CAAC,CAAC;IAChE,wBAAO,CAAC,uBAAuB,CAAC,cAAc,CAAC,CAAC;IAChD,iCAAW,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;IAC9C,mCAAY,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;IAC5C,yCAAe,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;IAChD,uCAAc,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,CAAC;IACtD,iCAAW,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,CAAC,aAAa,CAAC,QAAQ,CAAC,GAAG,EAAE,KAAK,CAAC;SAC3F;QACD,QAAQ,EAAE;YACR,EAAE,EAAE;gBACF,IAAI,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,CACnC,oBAAoB,YAAY,yDAAwB;oBACtD,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,SAAS,EAAE,EAAE,EAAE,IAAI,CAAC;oBAC1D,CAAC,CAAC,OAAO,CAAC,MAAM,CACZ,IAAI,KAAK,CAAC,wEAAwE,CAAC,CACpF;gBACP,KAAK,EAAE,CAAC,SAAS,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE;oBAC9B,IAAI,oBAAoB,YAAY,yDAAwB,EAAE,CAAC;wBAC7D,oBAAoB,CAAC,eAAe,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;oBACtD,CAAC;gBACH,CAAC;aACF;SACF;QACD,GAAG;KACJ,CAAC;IAEF,IAAI,CAAC;QACH,OAAO,MAAM,EAAE,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;YAAS,CAAC;QACT,yBAAS,CAAC,OAAO,EAAE,CAAC;QACpB,uCAAc,CAAC,OAAO,EAAE,CAAC;QACzB,wBAAO,CAAC,OAAO,EAAE,CAAC;QAClB,oEAAoE;QACpE,iEAAiE;QACjE,IAAA,6BAAgB,EAAC,cAAc,CAAC,CAAC;QACjC,IAAA,6BAAgB,EAAC,eAAe,CAAC,CAAC;QAClC,IAAA,6BAAgB,EAAC,kBAAkB,CAAC,CAAC;QACrC,IAAA,6BAAgB,EAAC,iBAAiB,CAAC,CAAC;QACpC,IAAA,6BAAgB,EAAC,cAAc,CAAC,CAAC;QACjC,aAAa,CAAC,KAAK,EAAE,CAAC;QACtB,YAAY,CAAC,KAAK,EAAE,CAAC;QACrB,aAAa,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,oBAAoB,YAAY,yDAAwB,EAAE,CAAC;YAC7D,oBAAoB,CAAC,KAAK,EAAE,CAAC;QAC/B,CAAC;QACD,eAAe,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;AACH,CAAC"}

package/dist/commonjs/v3/test/session-waitpoint-backend.d.ts

@@ -0,0 +1,68 @@
+import { ApiClient } from "../apiClient/index.js";
+import { NoopRuntimeManager } from "../runtime/noopRuntimeManager.js";
+import type { CreateSessionStreamWaitpointRequestBody, CreateSessionStreamWaitpointResponseBody } from "../schemas/api.js";
+import type { WaitpointTokenResult } from "../schemas/common.js";
+/**
+ * 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.
+ */
+export declare class SessionWaitpointBackend {
+    private readonly apiClient;
+    private readonly pending;
+    private disabled;
+    constructor(apiClient: ApiClient);
+    register(body: CreateSessionStreamWaitpointRequestBody): CreateSessionStreamWaitpointResponseBody;
+    wait(waitpointFriendlyId: string): Promise<WaitpointTokenResult>;
+    disable(): void;
+    /**
+     * 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.
+     */
+    private readNextRecord;
+}
+/**
+ * A {@link NoopRuntimeManager} whose `waitForWaitpoint` resolves in place from a
+ * {@link SessionWaitpointBackend} instead of hanging. Install it as the `runtime`
+ * global via `runInMockTaskContext({ runtimeManager })`.
+ */
+export declare class TestRuntimeManager extends NoopRuntimeManager {
+    private readonly backend;
+    constructor(backend: SessionWaitpointBackend);
+    waitForWaitpoint(params: {
+        waitpointFriendlyId: string;
+        finishDate?: Date;
+    }): Promise<WaitpointTokenResult>;
+}
+/**
+ * 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.
+ */
+export declare function installSessionWaitpointBackend(apiClient: ApiClient): {
+    backend: SessionWaitpointBackend;
+    runtimeManager: TestRuntimeManager;
+    restore: () => void;
+};