Luke The Dev
121 lines
3.7 KiB
TypeScript
121 lines
3.7 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
|
|
import type { TranscriptEntry } from "@/features/agents/state/transcript";
|
|
import {
|
|
reduceOfficeDeskHoldState,
|
|
reduceOfficeQaHoldState,
|
|
resolveOfficeIntentSnapshot,
|
|
resolveOfficeDeskDirective,
|
|
resolveOfficeQaDirective,
|
|
} from "@/lib/office/deskDirectives";
|
|
|
|
const createUserEntry = (
|
|
text: string,
|
|
sequenceKey: number,
|
|
): TranscriptEntry => ({
|
|
entryId: `entry-${sequenceKey}`,
|
|
role: "user",
|
|
kind: "user",
|
|
text,
|
|
sessionKey: "agent:agent-1:studio:test",
|
|
runId: null,
|
|
source: "history",
|
|
timestampMs: sequenceKey * 1_000,
|
|
sequenceKey,
|
|
confirmed: true,
|
|
fingerprint: `fp-${sequenceKey}`,
|
|
});
|
|
|
|
describe("deskDirectives", () => {
|
|
it("recognizes desk and release directives", () => {
|
|
expect(resolveOfficeDeskDirective("Go to your desk.")).toBe("desk");
|
|
expect(resolveOfficeDeskDirective("Please head to your desk now.")).toBe("desk");
|
|
expect(resolveOfficeDeskDirective("Go back to the desk.")).toBe("desk");
|
|
expect(resolveOfficeDeskDirective("> ok you can leave the desk")).toBe("release");
|
|
expect(resolveOfficeDeskDirective("go on a walk")).toBe("release");
|
|
expect(resolveOfficeDeskDirective("you can leave your desk now")).toBe("release");
|
|
expect(resolveOfficeDeskDirective("go to walk")).toBe("release");
|
|
});
|
|
|
|
it("builds a unified office intent snapshot", () => {
|
|
expect(resolveOfficeIntentSnapshot("Let's go to the gym.")).toMatchObject({
|
|
gym: { directive: "gym", source: "manual" },
|
|
desk: null,
|
|
standup: null,
|
|
});
|
|
expect(resolveOfficeIntentSnapshot("lets have a scrum meeting")).toMatchObject({
|
|
standup: "standup",
|
|
gym: null,
|
|
qa: null,
|
|
});
|
|
});
|
|
|
|
it("keeps the desk hold through unrelated messages", () => {
|
|
expect(
|
|
reduceOfficeDeskHoldState({
|
|
currentHeld: true,
|
|
lastUserMessage: "Can you summarize that for me?",
|
|
transcriptEntries: undefined,
|
|
}),
|
|
).toBe(true);
|
|
});
|
|
|
|
it("rebuilds the desk hold from transcript history", () => {
|
|
expect(
|
|
reduceOfficeDeskHoldState({
|
|
currentHeld: false,
|
|
lastUserMessage: "What did you finish?",
|
|
transcriptEntries: [
|
|
createUserEntry("> Go to your desk.", 1),
|
|
createUserEntry("> What did you finish?", 2),
|
|
],
|
|
}),
|
|
).toBe(true);
|
|
});
|
|
|
|
it("clears the desk hold when a release directive arrives", () => {
|
|
expect(
|
|
reduceOfficeDeskHoldState({
|
|
currentHeld: true,
|
|
lastUserMessage: "ok you can leave the desk",
|
|
transcriptEntries: [createUserEntry("> Go to your desk.", 1)],
|
|
}),
|
|
).toBe(false);
|
|
});
|
|
|
|
it("recognizes QA lab directives and releases", () => {
|
|
expect(resolveOfficeQaDirective("Please write tests for this flow.")).toBe("qa_lab");
|
|
expect(resolveOfficeQaDirective("Can you run tests and verify it?")).toBe(
|
|
"qa_lab",
|
|
);
|
|
expect(resolveOfficeQaDirective("Reproduce this bug in the QA room.")).toBe(
|
|
"qa_lab",
|
|
);
|
|
expect(resolveOfficeQaDirective("done testing")).toBe("release");
|
|
expect(resolveOfficeQaDirective("leave the QA lab")).toBe("release");
|
|
});
|
|
|
|
it("rebuilds the QA lab hold from transcript history", () => {
|
|
expect(
|
|
reduceOfficeQaHoldState({
|
|
currentHeld: false,
|
|
lastUserMessage: "What failed?",
|
|
transcriptEntries: [
|
|
createUserEntry("> Run tests in the QA lab.", 1),
|
|
createUserEntry("> What failed?", 2),
|
|
],
|
|
}),
|
|
).toBe(true);
|
|
});
|
|
|
|
it("clears the QA lab hold when a release directive arrives", () => {
|
|
expect(
|
|
reduceOfficeQaHoldState({
|
|
currentHeld: true,
|
|
lastUserMessage: "stop testing",
|
|
transcriptEntries: [createUserEntry("> Run tests in the QA lab.", 1)],
|
|
}),
|
|
).toBe(false);
|
|
});
|
|
});
|