point-source
70 lines
2.5 KiB
TypeScript
70 lines
2.5 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { aggregateTaskPendingInput, isInputCapableSessionState } from "./task-pending-input";
|
|
|
|
describe("isInputCapableSessionState", () => {
|
|
it.each([
|
|
["RUNNING", true],
|
|
["WAITING_FOR_INPUT", true],
|
|
["COMPLETED", false],
|
|
["FAILED", false],
|
|
["CANCELLED", false],
|
|
["STARTING", false],
|
|
["CREATED", false],
|
|
] as const)("%s -> %s", (state, expected) => {
|
|
expect(isInputCapableSessionState(state)).toBe(expected);
|
|
});
|
|
});
|
|
|
|
describe("aggregateTaskPendingInput", () => {
|
|
it("ignores sessions that are not input-capable", () => {
|
|
const sessions = [{ id: "s1", state: "COMPLETED" }];
|
|
const result = aggregateTaskPendingInput(
|
|
sessions,
|
|
() => ({
|
|
clarification: true,
|
|
permission: true,
|
|
}),
|
|
undefined,
|
|
);
|
|
expect(result).toEqual({ clarification: false, permission: false, hasUnloadedMessages: false });
|
|
});
|
|
|
|
it("ORs flags across every input-capable session", () => {
|
|
const sessions = [
|
|
{ id: "s1", state: "RUNNING" },
|
|
{ id: "s2", state: "WAITING_FOR_INPUT" },
|
|
];
|
|
const flagsBySession: Record<string, { clarification: boolean; permission: boolean }> = {
|
|
s1: { clarification: true, permission: false },
|
|
s2: { clarification: false, permission: true },
|
|
};
|
|
const result = aggregateTaskPendingInput(
|
|
sessions,
|
|
(session) => flagsBySession[session.id],
|
|
undefined,
|
|
);
|
|
expect(result).toEqual({ clarification: true, permission: true, hasUnloadedMessages: false });
|
|
});
|
|
|
|
it("falls back to taskPendingAction when a session's messages are not loaded", () => {
|
|
const sessions = [{ id: "s1", state: "RUNNING" }];
|
|
const result = aggregateTaskPendingInput(sessions, () => undefined, "clarification");
|
|
expect(result).toEqual({ clarification: true, permission: false, hasUnloadedMessages: true });
|
|
});
|
|
|
|
it("falls back to taskPendingAction when there are no sessions", () => {
|
|
const result = aggregateTaskPendingInput([], () => undefined, "permission");
|
|
expect(result).toEqual({ clarification: false, permission: true, hasUnloadedMessages: false });
|
|
});
|
|
|
|
it("does not apply the taskPendingAction fallback once every session's messages are loaded", () => {
|
|
const sessions = [{ id: "s1", state: "RUNNING" }];
|
|
const result = aggregateTaskPendingInput(
|
|
sessions,
|
|
() => ({ clarification: false, permission: false }),
|
|
"clarification",
|
|
);
|
|
expect(result).toEqual({ clarification: false, permission: false, hasUnloadedMessages: false });
|
|
});
|
|
});
|