agent-orchestrator/frontend/src/renderer/hooks/useBrowserView.test.tsx

266 lines
9.0 KiB
TypeScript

import { act, renderHook, waitFor } from "@testing-library/react";
import { afterEach, describe, expect, it, vi } from "vitest";
import { useBrowserView, type BrowserNavState } from "./useBrowserView";
type Listener = (state: BrowserNavState) => void;
function createSlot(rect: Partial<DOMRect> = {}) {
const slot = document.createElement("div");
document.body.appendChild(slot);
slot.getBoundingClientRect = vi.fn(() => ({
x: 12,
y: 34,
width: 320,
height: 240,
top: 34,
right: 332,
bottom: 274,
left: 12,
toJSON: () => ({}),
...rect,
}));
return slot;
}
function setupBridge() {
const listeners = new Set<Listener>();
const bridge = {
stateFor(viewId: string): BrowserNavState {
return {
viewId,
url: "",
title: "",
canGoBack: false,
canGoForward: false,
isLoading: false,
};
},
ensure: vi.fn(async (sessionId: string): Promise<BrowserNavState> => ({
viewId: `42:${sessionId}`,
url: "",
title: "",
canGoBack: false,
canGoForward: false,
isLoading: false,
})),
setBounds: vi.fn(),
navigate: vi.fn(async ({ viewId }: { viewId: string }) => bridge.stateFor(viewId)),
clear: vi.fn(async (viewId: string) => bridge.stateFor(viewId)),
goBack: vi.fn(async (viewId: string) => bridge.stateFor(viewId)),
goForward: vi.fn(async (viewId: string) => bridge.stateFor(viewId)),
reload: vi.fn(async (viewId: string) => bridge.stateFor(viewId)),
stop: vi.fn(async (viewId: string) => bridge.stateFor(viewId)),
destroy: vi.fn(),
onNavState: vi.fn((listener: Listener) => {
listeners.add(listener);
return () => listeners.delete(listener);
}),
emit(state: BrowserNavState) {
listeners.forEach((listener) => listener(state));
},
};
window.ao = { ...window.ao!, browser: bridge };
return bridge;
}
describe("useBrowserView", () => {
afterEach(() => {
vi.restoreAllMocks();
document.body.replaceChildren();
});
it("ensures a scoped browser view and reports the measured slot bounds", async () => {
const bridge = setupBridge();
const slot = createSlot();
const { result } = renderHook(() => useBrowserView({ sessionId: "sess-1", active: true, poppedOut: false }));
await waitFor(() => expect(bridge.ensure).toHaveBeenCalledWith("sess-1"));
act(() => result.current.slotRef(slot));
await waitFor(() =>
expect(bridge.setBounds).toHaveBeenCalledWith({
viewId: "42:sess-1",
rect: { x: 12, y: 34, width: 320, height: 240 },
visible: true,
}),
);
expect(result.current.viewId).toBe("42:sess-1");
});
it("clamps the native view to its resizable-panel column when the slot overspills", async () => {
const bridge = setupBridge();
// The slot is wider than its column (e.g. the `min-w-[280px]` wrapper on a
// narrower inspector panel). The native overlay isn't clipped by DOM
// overflow, so the reported bounds must be intersected with the column.
const column = document.createElement("div");
column.setAttribute("data-panel", "");
column.getBoundingClientRect = vi.fn(() => ({
x: 100,
y: 0,
width: 150,
height: 600,
top: 0,
right: 250,
bottom: 600,
left: 100,
toJSON: () => ({}),
}));
const slot = createSlot();
column.appendChild(slot);
document.body.appendChild(column);
const { result } = renderHook(() => useBrowserView({ sessionId: "sess-1", active: true, poppedOut: false }));
await waitFor(() => expect(bridge.ensure).toHaveBeenCalledWith("sess-1"));
act(() => result.current.slotRef(slot));
await waitFor(() =>
expect(bridge.setBounds).toHaveBeenCalledWith({
viewId: "42:sess-1",
rect: { x: 100, y: 34, width: 150, height: 240 },
visible: true,
}),
);
});
it("hides the native view when inactive and on unmount without destroying session state", async () => {
const bridge = setupBridge();
const slot = createSlot();
const { result, rerender, unmount } = renderHook(
({ active }) => useBrowserView({ sessionId: "sess-1", active, poppedOut: false }),
{ initialProps: { active: true } },
);
await waitFor(() => expect(result.current.viewId).toBe("42:sess-1"));
act(() => result.current.slotRef(slot));
rerender({ active: false });
await waitFor(() =>
expect(bridge.setBounds).toHaveBeenLastCalledWith({
viewId: "42:sess-1",
rect: { x: 0, y: 0, width: 0, height: 0 },
visible: false,
}),
);
unmount();
expect(bridge.setBounds).toHaveBeenLastCalledWith({
viewId: "42:sess-1",
rect: { x: 0, y: 0, width: 0, height: 0 },
visible: false,
});
expect(bridge.destroy).not.toHaveBeenCalled();
});
it("updates nav state only for the current view", async () => {
const bridge = setupBridge();
const { result } = renderHook(() => useBrowserView({ sessionId: "sess-1", active: true, poppedOut: false }));
await waitFor(() => expect(result.current.viewId).toBe("42:sess-1"));
act(() =>
bridge.emit({
viewId: "other:sess-1",
url: "https://ignored.test/",
title: "Ignored",
canGoBack: true,
canGoForward: true,
isLoading: true,
}),
);
expect(result.current.navState.url).toBe("");
act(() =>
bridge.emit({
viewId: "42:sess-1",
url: "http://localhost:5173/",
title: "Local app",
canGoBack: false,
canGoForward: true,
isLoading: false,
}),
);
expect(result.current.navState.url).toBe("http://localhost:5173/");
expect(result.current.navState.title).toBe("Local app");
});
it("navigates on each preview revision, including a same-URL re-run, and ignores replays", async () => {
const bridge = setupBridge();
const { rerender } = renderHook(
({ previewUrl, previewRevision }) =>
useBrowserView({ sessionId: "sess-1", active: true, poppedOut: false, previewUrl, previewRevision }),
{ initialProps: { previewUrl: "http://localhost:5173/", previewRevision: 1 } },
);
await waitFor(() =>
expect(bridge.navigate).toHaveBeenCalledWith({ viewId: "42:sess-1", url: "http://localhost:5173/" }),
);
expect(bridge.navigate).toHaveBeenCalledTimes(1);
// CDC replays the session payload on an unrelated update (revision
// unchanged) — the panel must not reload.
rerender({ previewUrl: "http://localhost:5173/", previewRevision: 1 });
expect(bridge.navigate).toHaveBeenCalledTimes(1);
// Re-running `ao preview` with the SAME url bumps the revision and must
// re-navigate (refresh) — the regression this issue fixes.
rerender({ previewUrl: "http://localhost:5173/", previewRevision: 2 });
await waitFor(() => expect(bridge.navigate).toHaveBeenCalledTimes(2));
// A changed target with a fresh revision navigates to the new URL.
rerender({ previewUrl: "file:///tmp/preview/index.html", previewRevision: 3 });
await waitFor(() =>
expect(bridge.navigate).toHaveBeenCalledWith({ viewId: "42:sess-1", url: "file:///tmp/preview/index.html" }),
);
expect(bridge.navigate).toHaveBeenCalledTimes(3);
});
it("navigates legacy preview URLs when the daemon omits preview revisions", async () => {
const bridge = setupBridge();
const { result, rerender } = renderHook(
({ previewUrl }) => useBrowserView({ sessionId: "sess-1", active: true, poppedOut: false, previewUrl }),
{ initialProps: { previewUrl: undefined as string | undefined } },
);
await waitFor(() => expect(result.current.viewId).toBe("42:sess-1"));
expect(bridge.navigate).not.toHaveBeenCalled();
rerender({ previewUrl: "http://localhost:5173/" });
await waitFor(() =>
expect(bridge.navigate).toHaveBeenCalledWith({ viewId: "42:sess-1", url: "http://localhost:5173/" }),
);
expect(bridge.navigate).toHaveBeenCalledTimes(1);
rerender({ previewUrl: "http://localhost:5173/" });
expect(bridge.navigate).toHaveBeenCalledTimes(1);
rerender({ previewUrl: "C:\\Users\\Lenovo\\Downloads\\sm5\\paper_explainer.html" });
await waitFor(() =>
expect(bridge.navigate).toHaveBeenCalledWith({
viewId: "42:sess-1",
url: "C:\\Users\\Lenovo\\Downloads\\sm5\\paper_explainer.html",
}),
);
expect(bridge.navigate).toHaveBeenCalledTimes(2);
});
it("clears the view when the preview is reset (ao preview clear) and does not navigate", async () => {
const bridge = setupBridge();
const { rerender } = renderHook(
({ previewUrl, previewRevision }) =>
useBrowserView({ sessionId: "sess-1", active: true, poppedOut: false, previewUrl, previewRevision }),
{ initialProps: { previewUrl: "http://localhost:5173/" as string | undefined, previewRevision: 1 } },
);
await waitFor(() => expect(bridge.navigate).toHaveBeenCalledTimes(1));
// `ao preview clear` empties previewUrl and bumps the revision.
rerender({ previewUrl: undefined, previewRevision: 2 });
await waitFor(() => expect(bridge.clear).toHaveBeenCalledWith("42:sess-1"));
expect(bridge.navigate).toHaveBeenCalledTimes(1);
});
it("does not navigate or clear without a preview URL at revision zero", async () => {
const bridge = setupBridge();
const { result } = renderHook(() => useBrowserView({ sessionId: "sess-1", active: true, poppedOut: false }));
await waitFor(() => expect(result.current.viewId).toBe("42:sess-1"));
expect(bridge.navigate).not.toHaveBeenCalled();
expect(bridge.clear).not.toHaveBeenCalled();
});
});