Files
dev/third_party/pi/packages/coding-agent/test/auth-url-copy.test.ts
alexion 6081e3e325 chore(vendor): import pi 1.3.0
Imported as a linear vendor commit so forge rebases do not replay the upstream root tree.

git-subtree-dir: third_party/pi

git-subtree-split: 71023aec1546d6d677a65669c5665c0927ce0dd0
2026-10-15 17:58:48 -04:00

61 lines
2.4 KiB
TypeScript

import { setKeybindings, type TUI } from "@earendil-works/pi-tui";
import { beforeAll, beforeEach, describe, expect, test, vi } from "vitest";
import { KeybindingsManager } from "../src/core/keybindings.ts";
import { McpManagerView } from "../src/extensions/mcp/ui.ts";
import { LoginDialogComponent } from "../src/modes/interactive/components/login-dialog.ts";
import { initTheme, theme } from "../src/modes/interactive/theme/theme.ts";
import { stripAnsi } from "../src/utils/ansi.ts";
const { copyToClipboard } = vi.hoisted(() => ({ copyToClipboard: vi.fn(async (_text: string) => {}) }));
vi.mock("../src/utils/clipboard.ts", () => ({ copyToClipboard }));
vi.mock("../src/utils/open-browser.ts", () => ({ openBrowser: vi.fn() }));
const URL = `https://auth.example.invalid/authorize?${"x".repeat(300)}`;
const CTRL_X = "\x18";
const tui = { requestRender: vi.fn() } as unknown as TUI;
function rendered(component: { render(width: number): string[] }): string {
return stripAnsi(component.render(80).join("\n"));
}
describe("sign-in URL copy key", () => {
beforeAll(() => {
initTheme("dark");
});
beforeEach(() => {
setKeybindings(new KeybindingsManager());
copyToClipboard.mockClear();
});
test("login dialog copies the auth URL instead of typing into the code input", async () => {
const dialog = new LoginDialogComponent(tui, "test", () => {});
dialog.showAuth(URL);
void dialog.showManualInput("Paste the code:");
expect(rendered(dialog)).toContain("ctrl+x to copy");
dialog.handleInput(CTRL_X);
await vi.waitFor(() => expect(rendered(dialog)).toContain("Copied URL to clipboard"));
expect(copyToClipboard).toHaveBeenCalledWith(URL);
});
test("login dialog ignores the copy key without an auth URL", () => {
const dialog = new LoginDialogComponent(tui, "test", () => {});
dialog.showDeviceCode({ userCode: "ABCD", verificationUri: "https://example.invalid/device" });
dialog.handleInput(CTRL_X);
expect(copyToClipboard).not.toHaveBeenCalled();
});
test("MCP sign-in screen copies the authorization URL", async () => {
const view = new McpManagerView(tui, theme, new KeybindingsManager());
void view.redirectUrl("Sign in to issues", URL, new AbortController().signal);
expect(rendered(view)).toContain("ctrl+x to copy");
view.handleInput(CTRL_X);
await vi.waitFor(() => expect(rendered(view)).toContain("Copied URL to clipboard"));
expect(copyToClipboard).toHaveBeenCalledWith(URL);
});
});