40 lines
2.4 KiB
TypeScript
40 lines
2.4 KiB
TypeScript
import { expect, test } from "@playwright/test";
|
||
import { findImageGenerationFailure } from "@/lib/canvas/image-generation-status";
|
||
import { getRequestFailureDetails } from "@/lib/request-error";
|
||
import { serverLoader } from "../helpers/canvas-download-server-loader";
|
||
|
||
test("a terminal image error reaches the stream and persists for disconnected clients", async () => {
|
||
const chunks: any[] = [];
|
||
const saved: any[] = [];
|
||
const load = serverLoader({ "@/lib/db/queries": { saveMessages: async ({ messages }: any) => saved.push(...messages) } });
|
||
const { reportImageGenerationFailure } = load("app/(chat)/api/chat/_lib/image-generation-failure.ts");
|
||
await reportImageGenerationFailure({
|
||
writer: { write: (chunk: any) => chunks.push(chunk) },
|
||
chatId: "chat", userMessageId: "original-user", messageId: "request-id",
|
||
error: new Error("get image url failed"), label: "test",
|
||
});
|
||
const streamed = JSON.parse(chunks.find(chunk => chunk.type === "data-imageGenerationStatus").data);
|
||
const persisted = findImageGenerationFailure(saved, "original-user");
|
||
expect(persisted).toEqual(streamed);
|
||
expect(persisted).toMatchObject({ phase: "failed", requestId: "request-id", failure: { code: "SOURCE_IMAGE_UNAVAILABLE", outcome: "failed" } });
|
||
const text = saved[0].parts[0].text;
|
||
expect(text).toContain("重新上传原图");
|
||
expect(text).toContain("请求编号：request-id");
|
||
expect(JSON.parse(chunks.at(-1).data).parts[0].text).toContain("IMAGE_GENERATION_END");
|
||
|
||
await reportImageGenerationFailure({
|
||
writer: { write: () => { throw new Error("browser disconnected"); } },
|
||
chatId: "chat", userMessageId: "next-user", messageId: "next-request",
|
||
error: new Error("get image url failed"), label: "test",
|
||
});
|
||
expect(findImageGenerationFailure(saved, "next-user")?.requestId).toBe("next-request");
|
||
});
|
||
|
||
test("late failures remain associated with their original user request", () => {
|
||
const status = { phase: "failed" as const, userMessageId: "first", failure: getRequestFailureDetails(new Error("HTTP 524")) };
|
||
const messages = [{ role: "user", parts: [] }, { role: "assistant", parts: [{ type: "data-imageGenerationStatus", data: status }] }];
|
||
expect(findImageGenerationFailure(messages, "first")).toEqual(status);
|
||
expect(findImageGenerationFailure(messages, "second")).toBeNull();
|
||
expect(findImageGenerationFailure([{ role: "assistant", parts: [{ type: "text", text: "请分析失败原因" }] }], "second")).toBeNull();
|
||
});
|