Carlos Florêncio
110 lines
4.2 KiB
TypeScript
110 lines
4.2 KiB
TypeScript
import { expect, type Page } from "@playwright/test";
|
|
import type { SeedData } from "../../fixtures/test-base";
|
|
import type { ApiClient } from "../../helpers/api-client";
|
|
import { waitForSessionState } from "../../helpers/session";
|
|
import { SessionPage } from "../../pages/session-page";
|
|
import { SidebarFilterPopoverPage } from "../../pages/sidebar-filter-popover";
|
|
|
|
// @covers AC-TASKS-RUNTIME-STATE-PUBLICATION-ORDER-001.5
|
|
// @covers AC-TASKS-RUNTIME-STATE-PUBLICATION-ORDER-001.6
|
|
export async function expectSendNowWorkflowRunning(
|
|
page: Page,
|
|
api: ApiClient,
|
|
seed: SeedData,
|
|
mobile: boolean,
|
|
): Promise<void> {
|
|
const workflow = await api.createWorkflow(seed.workspaceId, "Queued turn state");
|
|
const review = await api.createWorkflowStep(workflow.id, "Review", 0, { is_start_step: true });
|
|
const working = await api.createWorkflowStep(workflow.id, "In Progress", 1);
|
|
await api.createWorkflowStep(workflow.id, "Done", 2);
|
|
const task = await api.createTaskWithAgent(
|
|
seed.workspaceId,
|
|
"Queued workflow state",
|
|
seed.agentProfileId,
|
|
{
|
|
description: "/e2e:simple-message",
|
|
workflow_id: workflow.id,
|
|
workflow_step_id: review.id,
|
|
repository_ids: [seed.repositoryId],
|
|
},
|
|
);
|
|
if (!task.session_id) throw new Error("Task did not return a session ID");
|
|
const sessionId = task.session_id;
|
|
await page.goto(`/t/${task.id}`);
|
|
const session = new SessionPage(page);
|
|
await session.waitForLoad();
|
|
await session.waitForChatIdle();
|
|
await waitForSessionState(api, {
|
|
taskId: task.id,
|
|
sessionId,
|
|
expectedState: "WAITING_FOR_INPUT",
|
|
message: "queued workflow session should settle before queue setup",
|
|
});
|
|
await api.updateWorkflowStep(review.id, {
|
|
events: { on_turn_start: [{ type: "move_to_step", config: { step_id: working.id } }] },
|
|
});
|
|
const identity = await api.getQueueSessionIdentity(task.id, sessionId);
|
|
await api.setQueueAutoRun(identity, false);
|
|
await api.queueMessage(
|
|
identity,
|
|
[
|
|
'e2e:message("replacement is active")',
|
|
// Flush the text buffer before holding the mock turn open.
|
|
'e2e:tool_use("Check", {})',
|
|
'e2e:tool_result("checked")',
|
|
"e2e:delay(15000)",
|
|
'e2e:message("replacement finished")',
|
|
].join("\n"),
|
|
);
|
|
const chat = session.activeChat();
|
|
const chip = chat.getByTestId("queue-chip");
|
|
if (mobile) await chip.tap();
|
|
else await chip.click();
|
|
const row = chat.getByTestId("queue-entry");
|
|
await expect(row).toHaveCount(1);
|
|
if (!mobile) await row.hover();
|
|
const sendNow = row.getByTestId("queue-entry-send-now");
|
|
if (mobile) await sendNow.tap();
|
|
else await sendNow.click();
|
|
|
|
const agentBodies = chat.locator("[data-agent-message-body][data-message-id]");
|
|
await expect(agentBodies.filter({ hasText: "replacement is active" })).toBeVisible();
|
|
await waitForSessionState(api, {
|
|
taskId: task.id,
|
|
sessionId,
|
|
expectedState: "RUNNING",
|
|
message: "Send Now should keep the queued workflow turn running",
|
|
});
|
|
const runningTask = await api.getTask(task.id);
|
|
expect(runningTask.state).toBe("IN_PROGRESS");
|
|
expect(runningTask.workflow_step_id).toBe(working.id);
|
|
await expect(session.cancelAgentButton()).toBeVisible();
|
|
|
|
if (mobile) await page.getByTestId("mobile-session-menu").tap();
|
|
const surface = mobile
|
|
? page.getByRole("dialog", { name: "Tasks", exact: true })
|
|
: session.sidebar;
|
|
const filters = new SidebarFilterPopoverPage(page);
|
|
if (mobile) await surface.getByTestId("sidebar-filter-gear").tap();
|
|
else await filters.open();
|
|
await filters.setGroup("State");
|
|
await filters.close();
|
|
await expect(
|
|
surface.locator('[data-testid="sidebar-group-header"][data-group-key="IN_PROGRESS"]'),
|
|
).toBeVisible();
|
|
await expect(
|
|
surface.locator(`[data-task-row-id="${task.id}"]`).getByTestId("task-state-running"),
|
|
).toBeVisible();
|
|
if (mobile) await page.keyboard.press("Escape");
|
|
|
|
await waitForSessionState(api, {
|
|
taskId: task.id,
|
|
sessionId,
|
|
expectedState: "WAITING_FOR_INPUT",
|
|
message: "queued workflow turn should settle after its replacement prompt",
|
|
});
|
|
await expect(agentBodies.filter({ hasText: "replacement finished" })).toBeVisible();
|
|
await expect.poll(async () => (await api.getTask(task.id)).state).toBe("REVIEW");
|
|
await expect(session.cancelAgentButton()).toBeHidden();
|
|
}
|