Carlos Florêncio
104 lines
2.9 KiB
TypeScript
104 lines
2.9 KiB
TypeScript
import { expect, type Page } from "@playwright/test";
|
|
import type { SeedData } from "../fixtures/test-base";
|
|
import type { ApiClient } from "./api-client";
|
|
import { SessionPage } from "../pages/session-page";
|
|
|
|
const DONE_STATES = ["COMPLETED", "WAITING_FOR_INPUT"];
|
|
|
|
export async function waitForLatestSessionDone(
|
|
apiClient: ApiClient,
|
|
taskId: string,
|
|
expectedCount: number,
|
|
message: string,
|
|
timeout = 120_000,
|
|
): Promise<void> {
|
|
await expect
|
|
.poll(
|
|
async () => {
|
|
const { sessions } = await apiClient.listTaskSessions(taskId);
|
|
if (sessions.length < expectedCount) return false;
|
|
// API returns sessions newest-first.
|
|
const latest = sessions[0];
|
|
return DONE_STATES.includes(latest?.state ?? "");
|
|
},
|
|
{ timeout, message },
|
|
)
|
|
.toBe(true);
|
|
}
|
|
|
|
export async function waitForSessionDone(
|
|
apiClient: ApiClient,
|
|
taskId: string,
|
|
sessionId: string,
|
|
message: string,
|
|
timeout = 120_000,
|
|
): Promise<void> {
|
|
await expect
|
|
.poll(
|
|
async () => {
|
|
const { sessions } = await apiClient.listTaskSessions(taskId);
|
|
const session = sessions.find((s) => s.id === sessionId);
|
|
return DONE_STATES.includes(session?.state ?? "");
|
|
},
|
|
{ timeout, message },
|
|
)
|
|
.toBe(true);
|
|
}
|
|
|
|
export async function waitForSessionEnvironment(
|
|
apiClient: ApiClient,
|
|
options: {
|
|
taskId: string;
|
|
sessionId: string;
|
|
expectedEnvironmentId: string;
|
|
message: string;
|
|
timeout?: number;
|
|
},
|
|
): Promise<void> {
|
|
await expect
|
|
.poll(
|
|
async () => {
|
|
const { sessions } = await apiClient.listTaskSessions(options.taskId);
|
|
const session = sessions.find((s) => s.id === options.sessionId);
|
|
return session?.task_environment_id ?? "";
|
|
},
|
|
{ timeout: options.timeout ?? 60_000, message: options.message },
|
|
)
|
|
.toBe(options.expectedEnvironmentId);
|
|
}
|
|
|
|
export async function openTaskSession(page: Page, taskId: string): Promise<SessionPage> {
|
|
await page.goto(`/t/${taskId}`);
|
|
const session = new SessionPage(page);
|
|
await session.waitForLoad();
|
|
return session;
|
|
}
|
|
|
|
/**
|
|
* Seed a task + session and navigate to it, waiting for the first (normal)
|
|
* turn to complete. Follow-up prompts can then exercise retry flows from a
|
|
* clean idle state.
|
|
*/
|
|
export async function seedIdleSession(
|
|
testPage: Page,
|
|
apiClient: ApiClient,
|
|
seedData: SeedData,
|
|
title: string,
|
|
): Promise<SessionPage> {
|
|
const task = await apiClient.createTaskWithAgent(
|
|
seedData.workspaceId,
|
|
title,
|
|
seedData.agentProfileId,
|
|
{
|
|
description: "/e2e:simple-message",
|
|
workflow_id: seedData.workflowId,
|
|
workflow_step_id: seedData.startStepId,
|
|
repository_ids: [seedData.repositoryId],
|
|
},
|
|
);
|
|
if (!task.session_id) throw new Error("createTaskWithAgent did not return a session_id");
|
|
const session = await openTaskSession(testPage, task.id);
|
|
await session.waitForChatIdle({ timeout: 30_000 });
|
|
return session;
|
|
}
|