wehub-resource-sync
107 lines
3.7 KiB
TypeScript
107 lines
3.7 KiB
TypeScript
import { afterEach, beforeAll, beforeEach, describe, expect, it, vi } from "bun:test";
|
|
import { resetSettingsForTest, Settings } from "@oh-my-pi/pi-coding-agent/config/settings";
|
|
import { EventController } from "@oh-my-pi/pi-coding-agent/modes/controllers/event-controller";
|
|
import { initTheme } from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
|
|
import type { InteractiveModeContext } from "@oh-my-pi/pi-coding-agent/modes/types";
|
|
import type { AgentSessionEvent } from "@oh-my-pi/pi-coding-agent/session/agent-session";
|
|
|
|
function createContext() {
|
|
const setWorkingMessage = vi.fn();
|
|
const ensureLoadingAnimation = vi.fn();
|
|
const pendingTools = new Map<string, unknown>();
|
|
const session = {
|
|
getToolByName: () => undefined,
|
|
isAborting: false,
|
|
};
|
|
const ctx = {
|
|
isInitialized: true,
|
|
settings: { get: () => false },
|
|
statusLine: { invalidate: vi.fn(), markActivityStart: vi.fn(), markActivityEnd: vi.fn() },
|
|
updateEditorTopBorder: vi.fn(),
|
|
pendingTools,
|
|
hideThinkingBlock: false,
|
|
setWorkingMessage,
|
|
clearPinnedError: vi.fn(),
|
|
ensureLoadingAnimation,
|
|
ui: { requestRender: vi.fn() },
|
|
session,
|
|
viewSession: session,
|
|
} as unknown as InteractiveModeContext;
|
|
return { ctx, pendingTools, setWorkingMessage, session };
|
|
}
|
|
|
|
const AGENT_START = { type: "agent_start" } as unknown as AgentSessionEvent;
|
|
|
|
/** A `tool_execution_start` whose toolCallId is pre-seeded into `pendingTools`,
|
|
* so the handler only runs the intent->working-message path and skips component
|
|
* construction (which needs far heavier mocks). */
|
|
function toolStartWithIntent(toolCallId: string, intent: string): AgentSessionEvent {
|
|
return {
|
|
type: "tool_execution_start",
|
|
toolCallId,
|
|
toolName: "grep",
|
|
args: {},
|
|
intent,
|
|
} as unknown as AgentSessionEvent;
|
|
}
|
|
|
|
describe("EventController aborted-turn working messages", () => {
|
|
beforeAll(async () => {
|
|
await initTheme(false);
|
|
});
|
|
|
|
beforeEach(async () => {
|
|
resetSettingsForTest();
|
|
await Settings.init({ inMemory: true });
|
|
});
|
|
|
|
afterEach(() => {
|
|
vi.restoreAllMocks();
|
|
resetSettingsForTest();
|
|
});
|
|
|
|
it("suppresses late intent-driven working-message updates while aborting", async () => {
|
|
const { ctx, pendingTools, setWorkingMessage, session } = createContext();
|
|
const controller = new EventController(ctx);
|
|
await controller.handleEvent(AGENT_START);
|
|
setWorkingMessage.mockClear();
|
|
session.isAborting = true;
|
|
|
|
pendingTools.set("late-call", {});
|
|
await controller.handleEvent(toolStartWithIntent("late-call", "Reticulating splines"));
|
|
|
|
expect(setWorkingMessage).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("lets intent updates drive the loader when not aborting", async () => {
|
|
const { ctx, pendingTools, setWorkingMessage } = createContext();
|
|
const controller = new EventController(ctx);
|
|
await controller.handleEvent(AGENT_START);
|
|
setWorkingMessage.mockClear();
|
|
|
|
pendingTools.set("call-1", {});
|
|
await controller.handleEvent(toolStartWithIntent("call-1", "Searching files"));
|
|
|
|
expect(setWorkingMessage).toHaveBeenCalledTimes(1);
|
|
expect(setWorkingMessage.mock.calls[0]?.[0]).toContain("Searching files");
|
|
});
|
|
|
|
it("resumes intent updates once aborting clears", async () => {
|
|
const { ctx, pendingTools, setWorkingMessage, session } = createContext();
|
|
const controller = new EventController(ctx);
|
|
await controller.handleEvent(AGENT_START);
|
|
session.isAborting = true;
|
|
|
|
pendingTools.set("late-call", {});
|
|
await controller.handleEvent(toolStartWithIntent("late-call", "Reticulating splines"));
|
|
setWorkingMessage.mockClear();
|
|
session.isAborting = false;
|
|
|
|
pendingTools.set("call-2", {});
|
|
await controller.handleEvent(toolStartWithIntent("call-2", "Editing module"));
|
|
|
|
expect(setWorkingMessage).toHaveBeenCalledTimes(1);
|
|
expect(setWorkingMessage.mock.calls[0]?.[0]).toContain("Editing module");
|
|
});
|
|
});
|