From 590474a9a49def533c1264db8ac82427bd507520 Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Sun, 19 Apr 2026 04:19:35 +0100 Subject: [PATCH] test: share compact session fixture --- .../compact.hooks.harness.ts | 42 +++++++------------ 1 file changed, 16 insertions(+), 26 deletions(-) diff --git a/src/agents/pi-embedded-runner/compact.hooks.harness.ts b/src/agents/pi-embedded-runner/compact.hooks.harness.ts index fa2b54d6cea..b6ce3eb0720 100644 --- a/src/agents/pi-embedded-runner/compact.hooks.harness.ts +++ b/src/agents/pi-embedded-runner/compact.hooks.harness.ts @@ -71,18 +71,21 @@ export const resolveMemorySearchConfigMock = vi.fn(() => ({ })); export const resolveSessionAgentIdMock = vi.fn(() => "main"); export const estimateTokensMock = vi.fn((_message?: unknown) => 10); -export const sessionMessages: unknown[] = [ - { role: "user", content: "hello", timestamp: 1 }, - { role: "assistant", content: [{ type: "text", text: "hi" }], timestamp: 2 }, - { - role: "toolResult", - toolCallId: "t1", - toolName: "exec", - content: [{ type: "text", text: "output" }], - isError: false, - timestamp: 3, - }, -]; +function createDefaultSessionMessages(): unknown[] { + return [ + { role: "user", content: "hello", timestamp: 1 }, + { role: "assistant", content: [{ type: "text", text: "hi" }], timestamp: 2 }, + { + role: "toolResult", + toolCallId: "t1", + toolName: "exec", + content: [{ type: "text", text: "output" }], + isError: false, + timestamp: 3, + }, + ]; +} +export const sessionMessages: unknown[] = createDefaultSessionMessages(); export const sessionAbortCompactionMock: Mock<(reason?: unknown) => void> = vi.fn(); export const createOpenClawCodingToolsMock = vi.fn(() => []); export const resolveEmbeddedAgentStreamFnMock: Mock< @@ -118,20 +121,7 @@ export function resetCompactSessionStateMocks(): void { resolveSessionAgentIdMock.mockReturnValue("main"); estimateTokensMock.mockReset(); estimateTokensMock.mockReturnValue(10); - sessionMessages.splice( - 0, - sessionMessages.length, - { role: "user", content: "hello", timestamp: 1 }, - { role: "assistant", content: [{ type: "text", text: "hi" }], timestamp: 2 }, - { - role: "toolResult", - toolCallId: "t1", - toolName: "exec", - content: [{ type: "text", text: "output" }], - isError: false, - timestamp: 3, - }, - ); + sessionMessages.splice(0, sessionMessages.length, ...createDefaultSessionMessages()); sessionAbortCompactionMock.mockReset(); resolveEmbeddedAgentStreamFnMock.mockReset(); resolveEmbeddedAgentStreamFnMock.mockImplementation((_params?: unknown) => vi.fn());