mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-07-27 08:28:00 +00:00
f881996e1a
* fix(auth): recover from setup status timeouts * test(auth): cover setup status recovery flows --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
103 lines
2.9 KiB
TypeScript
103 lines
2.9 KiB
TypeScript
import { expect, test, type Page } from "@playwright/test";
|
|
|
|
const SETUP_STATUS_URL = "**/api/v1/auth/setup-status";
|
|
const SERVICE_UNAVAILABLE_TITLE = "Service temporarily unavailable";
|
|
|
|
async function mockSetupStatusRecovery(
|
|
page: Page,
|
|
recoveredStatus: {
|
|
needs_setup: boolean;
|
|
registration_enabled?: boolean;
|
|
},
|
|
) {
|
|
let requestCount = 0;
|
|
|
|
await page.route(SETUP_STATUS_URL, (route) => {
|
|
if (route.request().method() !== "GET") {
|
|
return route.fallback();
|
|
}
|
|
|
|
requestCount += 1;
|
|
if (requestCount === 1) {
|
|
return route.fulfill({
|
|
status: 503,
|
|
contentType: "application/json",
|
|
body: JSON.stringify({ detail: "Gateway unavailable" }),
|
|
});
|
|
}
|
|
|
|
return route.fulfill({
|
|
status: 200,
|
|
contentType: "application/json",
|
|
body: JSON.stringify(recoveredStatus),
|
|
});
|
|
});
|
|
|
|
return () => requestCount;
|
|
}
|
|
|
|
test.describe("auth setup-status recovery", () => {
|
|
test.beforeEach(async ({ page }) => {
|
|
await page.route("**/api/v1/auth/providers", (route) =>
|
|
route.fulfill({
|
|
status: 200,
|
|
contentType: "application/json",
|
|
body: JSON.stringify({ providers: [] }),
|
|
}),
|
|
);
|
|
});
|
|
|
|
test("login restores registration after setup-status retry", async ({
|
|
page,
|
|
}) => {
|
|
const getRequestCount = await mockSetupStatusRecovery(page, {
|
|
needs_setup: false,
|
|
registration_enabled: true,
|
|
});
|
|
|
|
await page.goto("/login");
|
|
|
|
await expect(page.getByText(SERVICE_UNAVAILABLE_TITLE)).toBeVisible();
|
|
await expect(page.getByRole("button", { name: "Sign In" })).toBeEnabled();
|
|
await expect(
|
|
page.getByRole("button", { name: /Don't have an account/i }),
|
|
).toHaveCount(0);
|
|
expect(getRequestCount()).toBe(1);
|
|
|
|
await page.getByRole("button", { name: "Try again" }).click();
|
|
|
|
await expect
|
|
.poll(getRequestCount, { message: "setup-status should be retried" })
|
|
.toBe(2);
|
|
await expect(page.getByText(SERVICE_UNAVAILABLE_TITLE)).toBeHidden();
|
|
await expect(
|
|
page.getByRole("button", { name: /Don't have an account/i }),
|
|
).toBeVisible();
|
|
});
|
|
|
|
test("setup restores the administrator form after setup-status retry", async ({
|
|
page,
|
|
}) => {
|
|
const getRequestCount = await mockSetupStatusRecovery(page, {
|
|
needs_setup: true,
|
|
});
|
|
|
|
await page.goto("/setup");
|
|
|
|
await expect(page.getByText(SERVICE_UNAVAILABLE_TITLE)).toBeVisible();
|
|
await expect(page.getByRole("button", { name: "Try again" })).toBeVisible();
|
|
expect(getRequestCount()).toBe(1);
|
|
|
|
await page.getByRole("button", { name: "Try again" }).click();
|
|
|
|
await expect
|
|
.poll(getRequestCount, { message: "setup-status should be retried" })
|
|
.toBe(2);
|
|
await expect(page.getByText(SERVICE_UNAVAILABLE_TITLE)).toBeHidden();
|
|
await expect(
|
|
page.getByRole("button", { name: "Create Admin Account" }),
|
|
).toBeVisible();
|
|
await expect(page.getByRole("textbox", { name: "Email" })).toBeVisible();
|
|
});
|
|
});
|