José Almeida
120 lines
3.8 KiB
TypeScript
120 lines
3.8 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { buildSubmitMessage } from "./chat-input-area";
|
|
import { resolveStatusRowTaskId, shouldRenderChatStatusBar } from "./chat-status-bar";
|
|
import { hasPendingClarification, shouldShowProceed } from "./types";
|
|
import type { AgentMessageComment } from "@/lib/state/slices/comments";
|
|
|
|
const messageComment: AgentMessageComment = {
|
|
id: "message-comment-1",
|
|
sessionId: "session-1",
|
|
source: "agent-message",
|
|
messageId: "reply-1",
|
|
selectedText: "answer",
|
|
anchor: {
|
|
messageId: "reply-1",
|
|
start: 0,
|
|
end: 6,
|
|
selectedText: "answer",
|
|
prefix: "",
|
|
suffix: "",
|
|
},
|
|
text: "Please make this more precise.",
|
|
createdAt: "2026-07-20T00:00:00Z",
|
|
status: "pending",
|
|
};
|
|
|
|
describe("buildSubmitMessage agent message comments", () => {
|
|
it("includes selected response context while preserving ordinary prose", () => {
|
|
const result = buildSubmitMessage({
|
|
message: "Continue from here.",
|
|
pendingPRFeedback: [],
|
|
planComments: [],
|
|
messageComments: [messageComment],
|
|
});
|
|
|
|
expect(result).toContain("### Agent Message Comments");
|
|
expect(result).toContain("> answer");
|
|
expect(result).toContain("Continue from here.");
|
|
});
|
|
});
|
|
|
|
describe("buildSubmitMessage plan comment ownership", () => {
|
|
it("leaves task-owned plan comments out of client-formatted Markdown", () => {
|
|
const result = buildSubmitMessage({
|
|
message: "Please continue.",
|
|
pendingPRFeedback: [],
|
|
planComments: [
|
|
{
|
|
id: "plan-comment-1",
|
|
sessionId: "",
|
|
taskId: "task-1",
|
|
planId: "plan-1",
|
|
version: 2,
|
|
source: "plan",
|
|
text: "Split this step.",
|
|
selectedText: "Large step",
|
|
createdAt: "2026-09-02T00:00:00Z",
|
|
status: "pending",
|
|
},
|
|
],
|
|
});
|
|
|
|
expect(result).toBe("Please continue.");
|
|
});
|
|
});
|
|
|
|
describe("shouldRenderChatStatusBar", () => {
|
|
it("removes empty taskless status chrome when its auto-scroll control is hidden", () => {
|
|
expect(
|
|
shouldRenderChatStatusBar({
|
|
hasTask: false,
|
|
hasTodos: false,
|
|
hasQueueChip: false,
|
|
showRightControls: false,
|
|
showProceed: false,
|
|
}),
|
|
).toBe(false);
|
|
});
|
|
});
|
|
|
|
describe("shouldShowProceed", () => {
|
|
it.each([
|
|
["not busy without a clarification", false, false, true],
|
|
["busy without a clarification", true, false, false],
|
|
["waiting for input with a pending clarification", false, true, false],
|
|
])("%s", (_state, isAgentBusy, hasPendingClarification, expected) => {
|
|
expect(shouldShowProceed("Review", isAgentBusy, hasPendingClarification)).toBe(expected);
|
|
});
|
|
});
|
|
|
|
describe("hasPendingClarification", () => {
|
|
it("retains the message-derived fallback", () => {
|
|
expect(hasPendingClarification(true, null)).toBe(true);
|
|
});
|
|
|
|
it("uses the durable session projection while messages hydrate", () => {
|
|
expect(hasPendingClarification(false, "clarification")).toBe(true);
|
|
});
|
|
|
|
it("does not treat a durable permission request as a clarification", () => {
|
|
expect(hasPendingClarification(false, "permission")).toBe(false);
|
|
});
|
|
});
|
|
|
|
describe("resolveStatusRowTaskId", () => {
|
|
it("prefers the session-derived task so existing hosts are unchanged", () => {
|
|
expect(resolveStatusRowTaskId("from-session", "from-host")).toBe("from-session");
|
|
});
|
|
|
|
// The regression: a blocked chain step has no session, so the session-derived
|
|
// id is null and the status row (and with it the dependency chip) disappeared
|
|
// on exactly the tasks the chip exists to describe.
|
|
it("falls back to the host's task when the task has no session yet", () => {
|
|
expect(resolveStatusRowTaskId(null, "from-host")).toBe("from-host");
|
|
});
|
|
|
|
it("stays null when neither side knows a task", () => {
|
|
expect(resolveStatusRowTaskId(null, null)).toBeNull();
|
|
});
|
|
});
|