Carlos Florêncio
74 lines
1.8 KiB
TypeScript
74 lines
1.8 KiB
TypeScript
import type { Page } from "@playwright/test";
|
|
|
|
const RATE_LIMIT_RESET_AT = "2030-01-01T00:00:00Z";
|
|
|
|
const REFRESHED_RATE_LIMIT = {
|
|
core: {
|
|
resource: "core",
|
|
remaining: 3210,
|
|
limit: 5000,
|
|
reset_at: RATE_LIMIT_RESET_AT,
|
|
updated_at: "2029-12-31T23:59:00Z",
|
|
},
|
|
graphql: {
|
|
resource: "graphql",
|
|
remaining: 4789,
|
|
limit: 5000,
|
|
reset_at: RATE_LIMIT_RESET_AT,
|
|
updated_at: "2029-12-31T23:59:00Z",
|
|
},
|
|
search: {
|
|
resource: "search",
|
|
remaining: 24,
|
|
limit: 30,
|
|
reset_at: RATE_LIMIT_RESET_AT,
|
|
updated_at: "2029-12-31T23:59:00Z",
|
|
},
|
|
};
|
|
|
|
const DISCOVERY_FAILURE = {
|
|
state: "degraded",
|
|
failed_target_count: 1,
|
|
category: "invalid_query",
|
|
last_failure_at: "2029-12-31T23:58:00Z",
|
|
retry_at: "2030-01-01T00:01:00Z",
|
|
revision: 4,
|
|
credential_generation: 1,
|
|
};
|
|
|
|
const DISCOVERY_RECOVERED = {
|
|
state: "healthy",
|
|
failed_target_count: 0,
|
|
revision: 5,
|
|
credential_generation: 1,
|
|
};
|
|
|
|
export async function stubGitHubRateLimits(page: Page, workspaceId: string) {
|
|
let recovered = false;
|
|
await page.route("**/api/v1/github/status?*", async (route) => {
|
|
const url = new URL(route.request().url());
|
|
if (url.searchParams.get("workspace_id") !== workspaceId) {
|
|
await route.continue();
|
|
return;
|
|
}
|
|
const response = await route.fetch();
|
|
const body = (await response.json()) as Record<string, unknown>;
|
|
const refreshed = url.searchParams.get("refresh_rate_limit") === "true";
|
|
body.pr_discovery_health = recovered ? DISCOVERY_RECOVERED : DISCOVERY_FAILURE;
|
|
if (refreshed) {
|
|
body.rate_limit = REFRESHED_RATE_LIMIT;
|
|
} else {
|
|
delete body.rate_limit;
|
|
}
|
|
await route.fulfill({
|
|
response,
|
|
json: body,
|
|
});
|
|
});
|
|
return {
|
|
recoverPRDiscovery() {
|
|
recovered = true;
|
|
},
|
|
};
|
|
}
|