Function mountHook

site/src/modules/resources/useAgentLogs.test.ts:56–86

(options: MountHookOptions)

}>;

function mountHook(options: MountHookOptions): MountHookResult {
	const { initialAgentId, enabled = true } = options;
	const serverResult: ServerResult = { current: undefined };

	vi.spyOn(apiModule, "watchWorkspaceAgentLogs").mockImplementation(
		(agentId, params) => {
			return new OneWayWebSocket({
				apiRoute: `/api/v2/workspaceagents/${agentId}/logs`,
				searchParams: new URLSearchParams({
					follow: "true",
					after: params?.after?.toString() || "0",
				}),
				websocketInit: (url) => {
					const [mockSocket, mockServer] = createMockWebSocket(url);
					serverResult.current = mockServer;
					return mockSocket;
				},
			});
		},
	);

	void vi.spyOn(console, "error").mockImplementation(() => {});
	const toastError = vi.spyOn(toast, "error");

	const { result: hookResult, rerender } = renderHook(
		(props) => useAgentLogs(props),
		{ initialProps: { enabled, agentId: initialAgentId } },
	);

	return { rerender, serverResult, hookResult, toastError };
}

describe("useAgentLogs", () => {
	it("Automatically sorts logs that are received out of order", async () => {