Carlos Florêncio
53 lines
1.6 KiB
TypeScript
53 lines
1.6 KiB
TypeScript
import type { Page } from "@playwright/test";
|
|
import type { AgentRuntimeAvailability } from "@/lib/types/agent-runtime";
|
|
|
|
export type AgentRuntimeAvailabilitySnapshot = AgentRuntimeAvailability;
|
|
|
|
type E2EStoreWindow = Window & {
|
|
__KANDEV_E2E_STORE__?: {
|
|
getState: () => {
|
|
setAgentRuntime: (snapshot: AgentRuntimeAvailabilitySnapshot | null) => void;
|
|
};
|
|
};
|
|
};
|
|
|
|
export async function setAgentRuntimeAvailability(
|
|
page: Page,
|
|
snapshot: AgentRuntimeAvailabilitySnapshot | null,
|
|
): Promise<void> {
|
|
await page.evaluate((nextSnapshot) => {
|
|
const store = (window as E2EStoreWindow).__KANDEV_E2E_STORE__;
|
|
if (!store) throw new Error("E2E store bridge missing");
|
|
store.getState().setAgentRuntime(nextSnapshot);
|
|
}, snapshot);
|
|
}
|
|
|
|
export async function stubAgentRuntimeRestart(page: Page): Promise<() => number> {
|
|
let restartRequests = 0;
|
|
|
|
await page.route("**/api/v1/system/restart-capability", (route) =>
|
|
route.fulfill({
|
|
status: 200,
|
|
contentType: "application/json",
|
|
body: JSON.stringify({ supported: true, mode: "supervisor", adapter: "supervisor" }),
|
|
}),
|
|
);
|
|
await page.route("**/api/v1/system/info", (route) =>
|
|
route.fulfill({
|
|
status: 200,
|
|
contentType: "application/json",
|
|
body: JSON.stringify({ boot_id: "e2e-stable-boot" }),
|
|
}),
|
|
);
|
|
await page.route("**/api/v1/system/restart", (route) => {
|
|
restartRequests += 1;
|
|
return route.fulfill({
|
|
status: 200,
|
|
contentType: "application/json",
|
|
body: JSON.stringify({ accepted: true, message: "accepted" }),
|
|
});
|
|
});
|
|
|
|
return () => restartRequests;
|
|
}
|