chore: Fix types in tests 26/N.

This commit is contained in:
cpojer
2026-02-17 14:31:40 +09:00
parent 6e5df1dc0f
commit 4235435309
14 changed files with 57 additions and 45 deletions

View File

@@ -17,8 +17,9 @@ describe("injectHistoryImagesIntoMessages", () => {
const didMutate = injectHistoryImagesIntoMessages(messages, new Map([[0, [image]]]));
expect(didMutate).toBe(true);
expect(Array.isArray(messages[0]?.content)).toBe(true);
const content = messages[0]?.content as Array<{ type: string; text?: string; data?: string }>;
const firstUser = messages[0] as Extract<AgentMessage, { role: "user" }> | undefined;
expect(Array.isArray(firstUser?.content)).toBe(true);
const content = firstUser?.content as Array<{ type: string; text?: string; data?: string }>;
expect(content).toHaveLength(2);
expect(content[0]?.type).toBe("text");
expect(content[1]).toMatchObject({ type: "image", data: "abc" });
@@ -35,7 +36,7 @@ describe("injectHistoryImagesIntoMessages", () => {
const didMutate = injectHistoryImagesIntoMessages(messages, new Map([[0, [image]]]));
expect(didMutate).toBe(false);
const first = messages[0];
const first = messages[0] as Extract<AgentMessage, { role: "user" }> | undefined;
if (!first || !Array.isArray(first.content)) {
throw new Error("expected array content");
}
@@ -47,12 +48,13 @@ describe("injectHistoryImagesIntoMessages", () => {
{
role: "assistant",
content: "noop",
} as AgentMessage,
} as unknown as AgentMessage,
];
const didMutate = injectHistoryImagesIntoMessages(messages, new Map([[1, [image]]]));
expect(didMutate).toBe(false);
expect(messages[0]?.content).toBe("noop");
const firstAssistant = messages[0] as Extract<AgentMessage, { role: "assistant" }> | undefined;
expect(firstAssistant?.content).toBe("noop");
});
});

View File

@@ -1,3 +1,4 @@
import type { AgentMessage } from "@mariozechner/pi-agent-core";
import { describe, expect, it } from "vitest";
import {
selectCompactionTimeoutSnapshot,
@@ -31,8 +32,8 @@ describe("compaction-timeout helpers", () => {
});
it("uses pre-compaction snapshot when compaction timeout occurs", () => {
const pre = [{ role: "assistant", content: "pre" }] as const;
const current = [{ role: "assistant", content: "current" }] as const;
const pre = [{ role: "assistant", content: "pre" } as unknown as AgentMessage] as const;
const current = [{ role: "assistant", content: "current" } as unknown as AgentMessage] as const;
const selected = selectCompactionTimeoutSnapshot({
timedOutDuringCompaction: true,
preCompactionSnapshot: [...pre],
@@ -46,7 +47,7 @@ describe("compaction-timeout helpers", () => {
});
it("falls back to current snapshot when pre-compaction snapshot is unavailable", () => {
const current = [{ role: "assistant", content: "current" }] as const;
const current = [{ role: "assistant", content: "current" } as unknown as AgentMessage] as const;
const selected = selectCompactionTimeoutSnapshot({
timedOutDuringCompaction: true,
preCompactionSnapshot: null,

View File

@@ -274,7 +274,7 @@ describe("buildEmbeddedRunPayloads", () => {
it("shows mutating tool errors even when assistant output exists", () => {
const payloads = buildPayloads({
assistantTexts: ["Done."],
lastAssistant: { stopReason: "end_turn" } as AssistantMessage,
lastAssistant: { stopReason: "end_turn" } as unknown as AssistantMessage,
lastToolError: { toolName: "write", error: "file missing" },
});
@@ -287,7 +287,7 @@ describe("buildEmbeddedRunPayloads", () => {
it("does not treat session_status read failures as mutating when explicitly flagged", () => {
const payloads = buildPayloads({
assistantTexts: ["Status loaded."],
lastAssistant: { stopReason: "end_turn" } as AssistantMessage,
lastAssistant: { stopReason: "end_turn" } as unknown as AssistantMessage,
lastToolError: {
toolName: "session_status",
error: "model required",
@@ -312,7 +312,7 @@ describe("buildEmbeddedRunPayloads", () => {
const payloads = buildPayloads({
assistantTexts: [warningText ?? ""],
lastAssistant: { stopReason: "end_turn" } as AssistantMessage,
lastAssistant: { stopReason: "end_turn" } as unknown as AssistantMessage,
lastToolError: {
toolName: "write",
error: "file missing",