Files
openworker/surfaces/gui/e2e/composer.spec.ts
T
Rohit C PrasadandDevika 2b45018ffa OpenWorker: initial import
Imported from andrewyng/aisuite@1b4bbf303e
(contents of its platform/ directory, hoisted to the repo root).
Development history prior to this commit lives in that repository.

Co-authored-by: Devika <devikaverma11@gmail.com>
2026-07-21 11:09:41 -07:00

98 lines
4.6 KiB
TypeScript

import { test, expect } from "./fixtures";
// Guards the three-control composer row (§22): send-gating (accent only with content), the "+"
// attach menu, and the Mode menu (permission options + the folded-in Send-to-Inbox toggle).
test("composer: send-gating, + attach menu, Mode menu", async ({ page }) => {
await page.goto("/");
await page.getByText("Draft the launch note").first().click();
const box = page.getByPlaceholder(/Ask the coworker/);
const send = page.getByRole("button", { name: "Send" });
// Send is subtle grey when empty, accent once there's content, grey again when cleared.
await expect(send).not.toHaveClass(/bg-accent/);
await box.fill("hello there");
await expect(send).toHaveClass(/bg-accent/);
await box.fill("");
await expect(send).not.toHaveClass(/bg-accent/);
// "+" attach menu offers the three typed shortcuts.
await page.getByRole("button", { name: "Attach" }).click();
await expect(page.getByRole("button", { name: "Photo or image" })).toBeVisible();
await expect(page.getByRole("button", { name: "PDF", exact: true })).toBeVisible();
await expect(page.getByRole("button", { name: "Other files" })).toBeVisible();
// Clicking the backdrop closes it.
await page.locator(".fixed.inset-0.z-30").click();
await expect(page.getByRole("button", { name: "Photo or image" })).toHaveCount(0);
// Mode menu (workspace personas only): the five permission options with the current one
// marked, plus the Unattended/send-to-Inbox toggle at the bottom (§22).
await page.getByRole("button", { name: "Mode", exact: true }).click();
const menu = page.getByTestId("mode-menu");
await expect(menu.getByText("Discuss")).toBeVisible();
await expect(menu.getByText("Explore read-only, propose a plan")).toBeVisible();
// The current mode is marked with a ✓.
await expect(menu.locator("button").filter({ hasText: "Ask for approval" })).toContainText("✓");
await expect(menu.getByRole("switch", { name: "Send approvals to the Inbox" })).toBeVisible();
// Picking an option closes the menu (and would flip the live engine's mode).
await menu.getByText("Full access").click();
await expect(page.getByTestId("mode-menu")).toHaveCount(0);
});
// PDFs read as data URLs and show a named chip (DMG #29 walkthrough catch: PDFs silently
// no-op'd because readFile only handled images and text).
test("composer: picking a PDF shows an attachment chip and arms send", async ({ page }) => {
await page.goto("/");
await page.getByText("Draft the launch note").first().click();
const send = page.getByRole("button", { name: "Send" });
await expect(send).not.toHaveClass(/bg-accent/);
await page.locator('input[type="file"]').setInputFiles({
name: "report.pdf",
mimeType: "application/pdf",
buffer: Buffer.from("%PDF-1.4\n1 0 obj\n<<>>\nendobj\ntrailer\n<<>>\n%%EOF"),
});
const chip = page.locator(".attach-chip");
await expect(chip).toContainText("report.pdf");
await expect(send).toHaveClass(/bg-accent/); // attachment alone arms send
// Removing the chip disarms send again.
await chip.locator(".attach-x").click();
await expect(page.locator(".attach-chip")).toHaveCount(0);
await expect(send).not.toHaveClass(/bg-accent/);
});
// Token-savings threshold (owner ask, 2026-07-17): a PDF over the user's page limit is
// REJECTED with a visible notice — no chip, send stays disarmed. Fixture limit: 2 pages;
// the mock inspect endpoint reads the page count from a "%%pages=N" marker in the body.
test("composer: PDF over the page threshold is rejected with a notice", async ({ page }) => {
await page.goto("/");
await page.getByText("Draft the launch note").first().click();
await page.locator('input[type="file"]').setInputFiles({
name: "big-report.pdf",
mimeType: "application/pdf",
buffer: Buffer.from("%PDF-1.4\n%%pages=34\ntrailer\n<<>>\n%%EOF"),
});
const notice = page.getByTestId("attach-notice");
await expect(notice).toContainText("big-report.pdf skipped");
await expect(notice).toContainText("34 pages is over your 2-page limit");
await expect(page.locator(".attach-chip")).toHaveCount(0);
await expect(page.getByRole("button", { name: "Send" })).not.toHaveClass(/bg-accent/);
// The ✕ dismisses the notice.
await notice.getByRole("button").click();
await expect(page.getByTestId("attach-notice")).toHaveCount(0);
// A small PDF (1 page per the mock) still attaches fine after a rejection.
await page.locator('input[type="file"]').setInputFiles({
name: "small.pdf",
mimeType: "application/pdf",
buffer: Buffer.from("%PDF-1.4\n%%pages=1\ntrailer\n<<>>\n%%EOF"),
});
await expect(page.locator(".attach-chip")).toContainText("small.pdf");
});