mirror of
https://github.com/badlogic/pi-mono.git
synced 2026-11-05 23:37:25 +00:00
ced72c2f09
Login and MCP sign-in screens show a copy hint and copy the URL to the clipboard (OSC 52 over SSH). /mcp login in the TUI now uses the MCP sign-in screen instead of a notification.
60 lines
2.4 KiB
TypeScript
60 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);
|
|
});
|
|
});
|