80 lines
2.2 KiB
TypeScript
80 lines
2.2 KiB
TypeScript
import { act, renderHook } from "@testing-library/react";
|
|
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
import type { AgentSummary } from "@/api/types/agents";
|
|
import { useAgentStatusPolling } from "./useAgentStatusPolling";
|
|
|
|
const agent = (startup_status: AgentSummary["startup_status"]) => ({
|
|
id: "agent",
|
|
name: "Agent",
|
|
description: "",
|
|
workspace_dir: "",
|
|
enabled: startup_status !== "disabled",
|
|
backend: "qwenpaw" as const,
|
|
startup_status,
|
|
});
|
|
|
|
describe("useAgentStatusPolling", () => {
|
|
afterEach(() => {
|
|
vi.useRealTimers();
|
|
});
|
|
|
|
it("keeps refreshing while an agent remains starting", async () => {
|
|
vi.useFakeTimers();
|
|
const refresh = vi.fn().mockResolvedValue(undefined);
|
|
renderHook(() => useAgentStatusPolling([agent("starting")], refresh));
|
|
|
|
await act(async () => {
|
|
await vi.advanceTimersByTimeAsync(3000);
|
|
});
|
|
|
|
expect(refresh).toHaveBeenCalledTimes(2);
|
|
});
|
|
|
|
it("retries after a refresh failure", async () => {
|
|
vi.useFakeTimers();
|
|
const refresh = vi.fn().mockRejectedValue(new Error("offline"));
|
|
renderHook(() => useAgentStatusPolling([agent("starting")], refresh));
|
|
|
|
await act(async () => {
|
|
await vi.advanceTimersByTimeAsync(3000);
|
|
});
|
|
|
|
expect(refresh).toHaveBeenCalledTimes(2);
|
|
});
|
|
|
|
it("does not refresh terminal statuses", async () => {
|
|
vi.useFakeTimers();
|
|
const refresh = vi.fn().mockResolvedValue(undefined);
|
|
renderHook(() => useAgentStatusPolling([agent("running")], refresh));
|
|
|
|
await act(async () => {
|
|
await vi.advanceTimersByTimeAsync(3000);
|
|
});
|
|
|
|
expect(refresh).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("stops polling when agents reach terminal statuses", async () => {
|
|
vi.useFakeTimers();
|
|
const refresh = vi.fn().mockResolvedValue(undefined);
|
|
const { rerender } = renderHook(
|
|
({ status }) => useAgentStatusPolling([agent(status)], refresh),
|
|
{
|
|
initialProps: {
|
|
status: "starting" as AgentSummary["startup_status"],
|
|
},
|
|
},
|
|
);
|
|
|
|
await act(async () => {
|
|
await vi.advanceTimersByTimeAsync(1500);
|
|
});
|
|
rerender({ status: "running" });
|
|
await act(async () => {
|
|
await vi.advanceTimersByTimeAsync(3000);
|
|
});
|
|
|
|
expect(refresh).toHaveBeenCalledOnce();
|
|
});
|
|
});
|