Carlos Florêncio
52 lines
1.7 KiB
TypeScript
52 lines
1.7 KiB
TypeScript
import { act, renderHook } from "@testing-library/react";
|
|
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
import { useSummarizeSession } from "./use-summarize-session";
|
|
|
|
const mockListMessages = vi.fn();
|
|
const mockExecuteUtilityPrompt = vi.fn();
|
|
|
|
vi.mock("@/lib/api/domains/session-api", () => ({
|
|
listTaskSessionMessages: (...args: unknown[]) => mockListMessages(...args),
|
|
}));
|
|
|
|
vi.mock("@/lib/api/domains/utility-api", () => ({
|
|
executeUtilityPrompt: (...args: unknown[]) => mockExecuteUtilityPrompt(...args),
|
|
}));
|
|
|
|
describe("useSummarizeSession", () => {
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
});
|
|
|
|
it("returns the generated summary", async () => {
|
|
mockListMessages.mockResolvedValue({
|
|
messages: [{ type: "message", author_type: "user", content: "hello" }],
|
|
});
|
|
mockExecuteUtilityPrompt.mockResolvedValue({ success: true, response: "summary" });
|
|
const { result } = renderHook(() => useSummarizeSession());
|
|
|
|
let summary;
|
|
await act(async () => {
|
|
summary = await result.current.summarize("session-1");
|
|
});
|
|
|
|
expect(summary).toEqual({ summary: "summary" });
|
|
});
|
|
|
|
it("returns backend execution errors instead of swallowing them", async () => {
|
|
mockListMessages.mockResolvedValue({
|
|
messages: [{ type: "message", author_type: "user", content: "hello" }],
|
|
});
|
|
mockExecuteUtilityPrompt.mockRejectedValue(new Error("connection refused"));
|
|
const { result } = renderHook(() => useSummarizeSession());
|
|
|
|
let summary;
|
|
await act(async () => {
|
|
summary = await result.current.summarize("session-1");
|
|
});
|
|
|
|
expect(summary).toEqual({ summary: null, error: "connection refused" });
|
|
expect(result.current.isSummarizing).toBe(false);
|
|
});
|
|
});
|