518 lines
16 KiB
TypeScript
518 lines
16 KiB
TypeScript
/**
|
|
* EventSource Mock Cleanup Requirements:
|
|
*
|
|
* This test file uses a MockEventSource class that tracks all instances in a static
|
|
* `instances` array. To prevent test isolation issues, we must ensure:
|
|
*
|
|
* 1. `MockEventSource.instances` is reset to empty before each test
|
|
* 2. Any lingering EventSource instances are closed and removed after each test
|
|
* 3. Fake timers are restored to real timers after each test (in case a test failed
|
|
* before it could restore them)
|
|
*
|
|
* Without proper cleanup, fake timers from one test can leak to subsequent tests,
|
|
* causing `waitFor()` calls to hang indefinitely.
|
|
*/
|
|
|
|
import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
|
|
import { renderHook, act, waitFor } from "@testing-library/react";
|
|
import { MAX_LOG_ENTRIES, useMultiAgentLogs } from "../useMultiAgentLogs";
|
|
import { fetchAgentLogs } from "../../api";
|
|
import { MockEventSource } from "../../../vitest.setup";
|
|
|
|
// Mock the api module
|
|
vi.mock("../../api", () => ({
|
|
fetchAgentLogs: vi.fn().mockResolvedValue([]),
|
|
}));
|
|
|
|
const mockFetchAgentLogs = vi.mocked(fetchAgentLogs);
|
|
|
|
// Helper to get the last connection for a specific task ID
|
|
function getConnection(taskId: string): MockEventSource | undefined {
|
|
const url = `/api/tasks/${taskId}/logs/stream`;
|
|
const matching = MockEventSource.instances.filter((e) => e.url === url);
|
|
return matching[matching.length - 1];
|
|
}
|
|
|
|
// Helper to get all connections for a task ID
|
|
function getConnections(taskId: string): MockEventSource[] {
|
|
const url = `/api/tasks/${taskId}/logs/stream`;
|
|
return MockEventSource.instances.filter((e) => e.url === url);
|
|
}
|
|
|
|
beforeEach(() => {
|
|
MockEventSource.instances = [];
|
|
mockFetchAgentLogs.mockReset().mockResolvedValue([]);
|
|
|
|
// Ensure we start with real timers for every test
|
|
vi.useRealTimers();
|
|
});
|
|
|
|
afterEach(() => {
|
|
// Close all lingering EventSource instances to clear reconnect timers
|
|
for (const instance of MockEventSource.instances) {
|
|
instance.close();
|
|
}
|
|
MockEventSource.instances = [];
|
|
|
|
// Safety: ensure real timers are restored even if a test failed
|
|
vi.useRealTimers();
|
|
});
|
|
|
|
describe("useMultiAgentLogs", () => {
|
|
it("initializes with empty entries for all provided task IDs", () => {
|
|
const { result } = renderHook(() => useMultiAgentLogs(["FN-001", "FN-002"]));
|
|
|
|
expect(result.current["FN-001"]).toBeDefined();
|
|
expect(result.current["FN-001"].entries).toEqual([]);
|
|
expect(result.current["FN-001"].loading).toBe(true);
|
|
|
|
expect(result.current["FN-002"]).toBeDefined();
|
|
expect(result.current["FN-002"].entries).toEqual([]);
|
|
expect(result.current["FN-002"].loading).toBe(true);
|
|
});
|
|
|
|
it("returns empty object when no task IDs provided", () => {
|
|
const { result } = renderHook(() => useMultiAgentLogs([]));
|
|
|
|
expect(Object.keys(result.current)).toHaveLength(0);
|
|
});
|
|
|
|
it("fetches historical logs for each task on mount", async () => {
|
|
const logs1 = [
|
|
{ timestamp: "2026-01-01T00:00:00Z", taskId: "FN-001", text: "log1", type: "text" as const },
|
|
];
|
|
const logs2 = [
|
|
{ timestamp: "2026-01-01T00:00:00Z", taskId: "FN-002", text: "log2", type: "text" as const },
|
|
];
|
|
|
|
mockFetchAgentLogs.mockImplementation((taskId) => {
|
|
if (taskId === "FN-001") return Promise.resolve(logs1);
|
|
if (taskId === "FN-002") return Promise.resolve(logs2);
|
|
return Promise.resolve([]);
|
|
});
|
|
|
|
const { result } = renderHook(() => useMultiAgentLogs(["FN-001", "FN-002"]));
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].entries).toEqual(logs1);
|
|
expect(result.current["FN-002"].entries).toEqual(logs2);
|
|
});
|
|
|
|
expect(mockFetchAgentLogs).toHaveBeenCalledWith("FN-001", undefined, { limit: 500 });
|
|
expect(mockFetchAgentLogs).toHaveBeenCalledWith("FN-002", undefined, { limit: 500 });
|
|
});
|
|
|
|
it("opens SSE EventSource for each task ID", async () => {
|
|
mockFetchAgentLogs.mockResolvedValue([]);
|
|
|
|
renderHook(() => useMultiAgentLogs(["FN-001", "FN-002"]));
|
|
|
|
await waitFor(() => {
|
|
// Filter to unique URLs (Strict Mode may create duplicates)
|
|
const urls = [...new Set(MockEventSource.instances.map((es) => es.url))];
|
|
expect(urls).toContain("/api/tasks/FN-001/logs/stream");
|
|
expect(urls).toContain("/api/tasks/FN-002/logs/stream");
|
|
});
|
|
});
|
|
|
|
it("merges live SSE events with historical entries", async () => {
|
|
const historical = [
|
|
{ timestamp: "2026-01-01T00:00:00Z", taskId: "FN-001", text: "old", type: "text" as const },
|
|
];
|
|
// Use mockResolvedValue (not Once) to handle Strict Mode double-run
|
|
mockFetchAgentLogs.mockResolvedValue(historical);
|
|
|
|
const { result } = renderHook(() => useMultiAgentLogs(["FN-001"]));
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].entries).toHaveLength(1);
|
|
});
|
|
|
|
const es = getConnection("FN-001");
|
|
expect(es).toBeDefined();
|
|
|
|
act(() => {
|
|
es!._emit("agent:log", {
|
|
timestamp: "2026-01-01T00:01:00Z",
|
|
taskId: "FN-001",
|
|
text: "new",
|
|
type: "text",
|
|
});
|
|
});
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].entries).toHaveLength(2);
|
|
});
|
|
|
|
expect(result.current["FN-001"].entries[1].text).toBe("new");
|
|
});
|
|
|
|
it("closes all SSE connections on unmount (memory leak prevention)", async () => {
|
|
mockFetchAgentLogs.mockResolvedValue([]);
|
|
|
|
const { unmount } = renderHook(() => useMultiAgentLogs(["FN-001", "FN-002"]));
|
|
|
|
// Wait for connections to be established
|
|
await waitFor(() => {
|
|
expect(MockEventSource.instances.length).toBeGreaterThanOrEqual(2);
|
|
});
|
|
|
|
// Get unique instances by URL (handling Strict Mode duplicates)
|
|
const uniqueByUrl = new Map<string, MockEventSource>();
|
|
for (const es of MockEventSource.instances) {
|
|
if (!uniqueByUrl.has(es.url) || !es.close.mock?.calls?.length) {
|
|
uniqueByUrl.set(es.url, es);
|
|
}
|
|
}
|
|
const finalInstances = Array.from(uniqueByUrl.values());
|
|
|
|
unmount();
|
|
|
|
// Verify all final connections are closed
|
|
for (const es of finalInstances) {
|
|
expect(es.close).toHaveBeenCalled();
|
|
}
|
|
});
|
|
|
|
it("closes specific connection when task ID removed from array", async () => {
|
|
mockFetchAgentLogs.mockResolvedValue([]);
|
|
|
|
const { rerender } = renderHook(
|
|
({ taskIds }: { taskIds: string[] }) => useMultiAgentLogs(taskIds),
|
|
{ initialProps: { taskIds: ["FN-001", "FN-002"] } },
|
|
);
|
|
|
|
await waitFor(() => {
|
|
expect(MockEventSource.instances.length).toBeGreaterThanOrEqual(2);
|
|
});
|
|
|
|
const es1 = getConnection("FN-001");
|
|
const es2 = getConnection("FN-002");
|
|
|
|
rerender({ taskIds: ["FN-001"] });
|
|
|
|
await waitFor(() => {
|
|
expect(es2!.close).toHaveBeenCalled();
|
|
});
|
|
|
|
expect(es1!.close).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("opens new connection when task ID added to array", async () => {
|
|
mockFetchAgentLogs.mockResolvedValue([]);
|
|
|
|
const { rerender } = renderHook(
|
|
({ taskIds }: { taskIds: string[] }) => useMultiAgentLogs(taskIds),
|
|
{ initialProps: { taskIds: ["FN-001"] } },
|
|
);
|
|
|
|
await waitFor(() => {
|
|
expect(MockEventSource.instances.length).toBeGreaterThanOrEqual(1);
|
|
});
|
|
|
|
rerender({ taskIds: ["FN-001", "FN-002"] });
|
|
|
|
await waitFor(() => {
|
|
const urls = [...new Set(MockEventSource.instances.map((es) => es.url))];
|
|
expect(urls).toContain("/api/tasks/FN-002/logs/stream");
|
|
});
|
|
});
|
|
|
|
it("provides per-task clear function that resets entries", async () => {
|
|
const logs = [
|
|
{ timestamp: "2026-01-01T00:00:00Z", taskId: "FN-001", text: "log1", type: "text" as const },
|
|
{ timestamp: "2026-01-01T00:01:00Z", taskId: "FN-001", text: "log2", type: "text" as const },
|
|
];
|
|
// Use mockResolvedValue (not Once) to handle Strict Mode double-run
|
|
mockFetchAgentLogs.mockResolvedValue(logs);
|
|
|
|
const { result } = renderHook(() => useMultiAgentLogs(["FN-001", "FN-002"]));
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].entries).toHaveLength(2);
|
|
});
|
|
|
|
// Clear only FN-001
|
|
act(() => {
|
|
result.current["FN-001"].clear();
|
|
});
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].entries).toHaveLength(0);
|
|
});
|
|
});
|
|
|
|
it("handles errors gracefully when fetching historical logs", async () => {
|
|
mockFetchAgentLogs.mockRejectedValue(new Error("Network error"));
|
|
|
|
const { result } = renderHook(() => useMultiAgentLogs(["FN-001"]));
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].loading).toBe(false);
|
|
});
|
|
|
|
expect(result.current["FN-001"].entries).toEqual([]);
|
|
});
|
|
|
|
it("does not create duplicate connections while historical fetch is still pending", async () => {
|
|
let resolveFetch: ((value: never[]) => void) | undefined;
|
|
mockFetchAgentLogs.mockImplementation(
|
|
() => new Promise((resolve) => {
|
|
resolveFetch = resolve;
|
|
}),
|
|
);
|
|
|
|
const { rerender } = renderHook(
|
|
({ taskIds }: { taskIds: string[] }) => useMultiAgentLogs(taskIds),
|
|
{ initialProps: { taskIds: ["FN-001"] } },
|
|
);
|
|
|
|
await waitFor(() => {
|
|
// Allow for Strict Mode double-rendering
|
|
expect(getConnections("FN-001").length).toBeGreaterThanOrEqual(1);
|
|
});
|
|
|
|
const initialCount = getConnections("FN-001").length;
|
|
|
|
rerender({ taskIds: ["FN-001"] });
|
|
|
|
await new Promise((resolve) => setTimeout(resolve, 10));
|
|
|
|
// Should not create additional connections on rerender with same IDs
|
|
expect(getConnections("FN-001").length).toBe(initialCount);
|
|
|
|
resolveFetch?.([]);
|
|
|
|
await waitFor(() => {
|
|
expect(mockFetchAgentLogs).toHaveBeenCalledTimes(1);
|
|
});
|
|
});
|
|
|
|
it("closes a task connection when its stream emits an error", async () => {
|
|
mockFetchAgentLogs.mockResolvedValue([]);
|
|
|
|
renderHook(() => useMultiAgentLogs(["FN-001"]));
|
|
|
|
await waitFor(() => {
|
|
expect(MockEventSource.instances.length).toBeGreaterThanOrEqual(1);
|
|
});
|
|
|
|
const es = getConnection("FN-001");
|
|
expect(es).toBeDefined();
|
|
|
|
act(() => {
|
|
es!._emit("error");
|
|
});
|
|
|
|
expect(es!.close).toHaveBeenCalledTimes(1);
|
|
});
|
|
|
|
it("truncates oversized historical logs per task to the most recent entries", async () => {
|
|
const oversized = Array.from({ length: MAX_LOG_ENTRIES + 10 }, (_, index) => ({
|
|
timestamp: `2026-01-01T00:${String(index).padStart(2, "0")}:00Z`,
|
|
taskId: "FN-001",
|
|
text: `entry-${index}`,
|
|
type: "text" as const,
|
|
}));
|
|
|
|
mockFetchAgentLogs.mockResolvedValue(oversized);
|
|
|
|
const { result } = renderHook(() => useMultiAgentLogs(["FN-001"]));
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].entries).toHaveLength(MAX_LOG_ENTRIES);
|
|
});
|
|
|
|
expect(result.current["FN-001"].entries[0].text).toBe("entry-10");
|
|
expect(result.current["FN-001"].entries.at(-1)?.text).toBe(`entry-${MAX_LOG_ENTRIES + 9}`);
|
|
});
|
|
|
|
it("preserves streamed entries that arrive before historical fetch resolves", async () => {
|
|
let resolveFetch: ((value: Array<{ timestamp: string; taskId: string; text: string; type: "text" }>) => void) | undefined;
|
|
mockFetchAgentLogs.mockImplementation(
|
|
() => new Promise((resolve) => {
|
|
resolveFetch = resolve;
|
|
}),
|
|
);
|
|
|
|
const { result } = renderHook(() => useMultiAgentLogs(["FN-001"]));
|
|
|
|
await waitFor(() => {
|
|
expect(MockEventSource.instances.length).toBeGreaterThanOrEqual(1);
|
|
});
|
|
|
|
const es = getConnection("FN-001");
|
|
expect(es).toBeDefined();
|
|
|
|
act(() => {
|
|
es!._emit("agent:log", {
|
|
timestamp: "2026-01-01T00:01:00Z",
|
|
taskId: "FN-001",
|
|
text: "live-before-history",
|
|
type: "text",
|
|
});
|
|
});
|
|
|
|
act(() => {
|
|
resolveFetch?.([
|
|
{
|
|
timestamp: "2026-01-01T00:00:00Z",
|
|
taskId: "FN-001",
|
|
text: "historical",
|
|
type: "text",
|
|
},
|
|
]);
|
|
});
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].entries).toHaveLength(2);
|
|
});
|
|
|
|
expect(result.current["FN-001"].entries[0].text).toBe("historical");
|
|
expect(result.current["FN-001"].entries[1].text).toBe("live-before-history");
|
|
});
|
|
|
|
it("truncates live SSE entries per task to the most recent entries", async () => {
|
|
mockFetchAgentLogs.mockResolvedValue([]);
|
|
|
|
const { result } = renderHook(() => useMultiAgentLogs(["FN-001"]));
|
|
|
|
await waitFor(() => {
|
|
expect(MockEventSource.instances.length).toBeGreaterThanOrEqual(1);
|
|
});
|
|
|
|
const es = getConnection("FN-001");
|
|
expect(es).toBeDefined();
|
|
|
|
act(() => {
|
|
for (let index = 0; index < MAX_LOG_ENTRIES + 15; index++) {
|
|
es!._emit("agent:log", {
|
|
timestamp: `2026-01-01T00:${String(index).padStart(2, "0")}:00Z`,
|
|
taskId: "FN-001",
|
|
text: `live-${index}`,
|
|
type: "text",
|
|
});
|
|
}
|
|
});
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].entries).toHaveLength(MAX_LOG_ENTRIES);
|
|
});
|
|
|
|
expect(result.current["FN-001"].entries[0].text).toBe("live-15");
|
|
expect(result.current["FN-001"].entries.at(-1)?.text).toBe(`live-${MAX_LOG_ENTRIES + 14}`);
|
|
});
|
|
|
|
it("handles SSE events for multiple tasks independently", async () => {
|
|
const logs1 = [
|
|
{ timestamp: "2026-01-01T00:00:00Z", taskId: "FN-001", text: "task1-old", type: "text" as const },
|
|
];
|
|
const logs2 = [
|
|
{ timestamp: "2026-01-01T00:00:00Z", taskId: "FN-002", text: "task2-old", type: "text" as const },
|
|
];
|
|
|
|
mockFetchAgentLogs.mockImplementation((taskId) => {
|
|
if (taskId === "FN-001") return Promise.resolve(logs1);
|
|
if (taskId === "FN-002") return Promise.resolve(logs2);
|
|
return Promise.resolve([]);
|
|
});
|
|
|
|
const { result } = renderHook(() => useMultiAgentLogs(["FN-001", "FN-002"]));
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].entries).toHaveLength(1);
|
|
expect(result.current["FN-002"].entries).toHaveLength(1);
|
|
});
|
|
|
|
const es1 = getConnection("FN-001");
|
|
const es2 = getConnection("FN-002");
|
|
expect(es1).toBeDefined();
|
|
expect(es2).toBeDefined();
|
|
|
|
act(() => {
|
|
es1!._emit("agent:log", {
|
|
timestamp: "2026-01-01T00:01:00Z",
|
|
taskId: "FN-001",
|
|
text: "task1-new",
|
|
type: "text",
|
|
});
|
|
});
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].entries).toHaveLength(2);
|
|
expect(result.current["FN-002"].entries).toHaveLength(1);
|
|
});
|
|
|
|
act(() => {
|
|
es2!._emit("agent:log", {
|
|
timestamp: "2026-01-01T00:01:00Z",
|
|
taskId: "FN-002",
|
|
text: "task2-new",
|
|
type: "text",
|
|
});
|
|
});
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].entries).toHaveLength(2);
|
|
expect(result.current["FN-002"].entries).toHaveLength(2);
|
|
});
|
|
|
|
expect(result.current["FN-001"].entries[1].text).toBe("task1-new");
|
|
expect(result.current["FN-002"].entries[1].text).toBe("task2-new");
|
|
});
|
|
|
|
it("preserves long text and detail in historical log entries without truncation", async () => {
|
|
const longText = "A".repeat(5000);
|
|
const longDetail = "B".repeat(5000);
|
|
mockFetchAgentLogs.mockResolvedValue([
|
|
{ timestamp: "2026-01-01T00:00:00Z", taskId: "FN-001", text: longText, type: "text" as const },
|
|
{ timestamp: "2026-01-01T00:00:01Z", taskId: "FN-001", text: "Read", type: "tool" as const, detail: longDetail },
|
|
]);
|
|
|
|
const { result } = renderHook(() => useMultiAgentLogs(["FN-001"]));
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].entries).toHaveLength(2);
|
|
});
|
|
|
|
expect(result.current["FN-001"].entries[0].text).toBe(longText);
|
|
expect(result.current["FN-001"].entries[0].text.length).toBe(5000);
|
|
expect(result.current["FN-001"].entries[1].detail).toBe(longDetail);
|
|
expect(result.current["FN-001"].entries[1].detail!.length).toBe(5000);
|
|
});
|
|
|
|
it("preserves long text and detail in live SSE entries without truncation", async () => {
|
|
mockFetchAgentLogs.mockResolvedValue([]);
|
|
|
|
const { result } = renderHook(() => useMultiAgentLogs(["FN-001"]));
|
|
|
|
await waitFor(() => {
|
|
expect(MockEventSource.instances.length).toBeGreaterThanOrEqual(1);
|
|
});
|
|
|
|
const es = getConnection("FN-001");
|
|
expect(es).toBeDefined();
|
|
|
|
const longText = "X".repeat(5000);
|
|
const longDetail = "Y".repeat(5000);
|
|
act(() => {
|
|
es!._emit("agent:log", {
|
|
timestamp: "2026-01-01T00:01:00Z",
|
|
taskId: "FN-001",
|
|
text: longText,
|
|
type: "text",
|
|
detail: longDetail,
|
|
});
|
|
});
|
|
|
|
await waitFor(() => {
|
|
expect(result.current["FN-001"].entries).toHaveLength(1);
|
|
});
|
|
|
|
expect(result.current["FN-001"].entries[0].text).toBe(longText);
|
|
expect(result.current["FN-001"].entries[0].text.length).toBe(5000);
|
|
expect(result.current["FN-001"].entries[0].detail).toBe(longDetail);
|
|
expect(result.current["FN-001"].entries[0].detail!.length).toBe(5000);
|
|
});
|
|
});
|