AdoptVaultView.test.tsx

7.6 KB · raw

1// @vitest-environment jsdom23import type { JSX } from "solid-js";4import { afterEach, describe, expect, it, vi } from "vitest";5import { Onboarding } from "./Onboarding";6import {7  AdoptVaultView,8  type AdoptVaultViewProps,9} from "./AdoptVaultView";10import { flushView, inputText, renderView } from "./view-test-utils";11import { createPairingInvitation } from "../../shared/pairing-invitation";1213const INVITATION = createPairingInvitation(14  "wss://relay.peachpasswords.com",15  "AAECAwQFBgcICQoLDA0ODw",16);1718vi.mock("@zxcvbn-ts/core", () => ({19  zxcvbn: (value: string) => ({ score: value.length >= 16 ? 4 : 1 }),20  zxcvbnOptions: { setOptions: vi.fn() },21}));22vi.mock("@zxcvbn-ts/language-common", () => ({23  adjacencyGraphs: {},24  dictionary: {},25}));26vi.mock("@zxcvbn-ts/language-en", () => ({27  translations: {},28  dictionary: {},29}));3031function button(container: HTMLElement, text: string): HTMLButtonElement {32  const found = Array.from(container.querySelectorAll("button")).find((b) =>33    (b.textContent ?? "").includes(text),34  );35  if (!found) throw new Error(`button not found: ${text}`);36  return found as HTMLButtonElement;37}3839async function enterInvitation(container: HTMLElement) {40  button(container, "Enter invitation manually").click();41  await flushView();42  inputText(43    container.querySelector("#adopt-relay-invitation") as HTMLTextAreaElement,44    INVITATION,45  );46  await flushView();47}4849async function enterStrongPassword(container: HTMLElement) {50  inputText(container.querySelector("#adopt-password") as HTMLInputElement, "9xQ!vZ2pLm74Tr#Ke8Wd5Bn");51  inputText(container.querySelector("#adopt-confirm-password") as HTMLInputElement, "9xQ!vZ2pLm74Tr#Ke8Wd5Bn");52  await flushView();53}5455describe("vault adoption onboarding", () => {56  const cleanups: Array<() => void> = [];5758  afterEach(() => {59    vi.useRealTimers();60    while (cleanups.length) cleanups.pop()!();61  });6263  function mounted(view: () => JSX.Element) {64    const rendered = renderView(view);65    cleanups.push(rendered.dispose);66    return rendered;67  }6869  it("T-A21: Onboarding offers Create vault, Restore from backup, and I already have Peach", () => {70    vi.useFakeTimers();71    const onComplete = vi.fn();72    const onRestore = vi.fn();73    const onAdopt = vi.fn();74    const view = mounted(() => <Onboarding onComplete={onComplete} onRestore={onRestore} onAdopt={onAdopt} />);7576    button(view.container, "next").click();77    vi.advanceTimersByTime(220);78    button(view.container, "next").click();79    vi.advanceTimersByTime(220);8081    expect(view.container.textContent).toContain("Create vault");82    expect(view.container.textContent).toContain("Restore from backup");83    expect(view.container.textContent).toContain("I already have Peach");84    button(view.container, "I already have Peach").click();85    expect(onAdopt).toHaveBeenCalledTimes(1);86  });8788  it("T-A22: password screen uses the same minimum-length gate as create-vault", () => {89    const view = mounted(() => <AdoptVaultView onCancel={() => {}} onDone={() => {}} startAdoption={vi.fn()} />);90    const continueButton = button(view.container, "Continue");91    expect(continueButton.disabled).toBe(true);92    inputText(view.container.querySelector("#adopt-password") as HTMLInputElement, "short");93    inputText(view.container.querySelector("#adopt-confirm-password") as HTMLInputElement, "short");94    expect(view.container.textContent).toContain("need 7 more characters");95    expect(continueButton.disabled).toBe(true);96  });9798  it("T-A26: relay adoption cannot start until a valid invitation is entered", async () => {99    const startAdoption = vi.fn(100      async (_args: Parameters<AdoptVaultViewProps["startAdoption"]>[0]) => {},101    );102    const view = mounted(() => <AdoptVaultView onCancel={() => {}} onDone={() => {}} startAdoption={startAdoption} />);103    await enterStrongPassword(view.container);104    button(view.container, "Continue").click();105    await flushView();106107    expect(view.container.textContent).toContain("Connecting via secure relay");108    button(view.container, "Enter invitation manually").click();109    await flushView();110    expect(button(view.container, "Connect securely").disabled).toBe(true);111    inputText(112      view.container.querySelector("#adopt-relay-invitation") as HTMLTextAreaElement,113      "peach-pair-v1:not-valid",114    );115    await flushView();116    expect(button(view.container, "Connect securely").disabled).toBe(true);117118    inputText(119      view.container.querySelector("#adopt-relay-invitation") as HTMLTextAreaElement,120      INVITATION,121    );122    await flushView();123    expect(button(view.container, "Connect securely").disabled).toBe(false);124    expect(startAdoption).not.toHaveBeenCalled();125  });126127  it("T-A27: the relay invitation is handed to the adoption transport", async () => {128    const startAdoption = vi.fn(129      async (_args: Parameters<AdoptVaultViewProps["startAdoption"]>[0]) => {},130    );131    const view = mounted(() => <AdoptVaultView onCancel={() => {}} onDone={() => {}} startAdoption={startAdoption} />);132    await enterStrongPassword(view.container);133    button(view.container, "Continue").click();134    await enterInvitation(view.container);135    button(view.container, "Connect securely").click();136    await flushView();137138    expect(startAdoption).toHaveBeenCalledTimes(1);139    expect(startAdoption.mock.calls[0][0]).toMatchObject({140      connection: { invitation: INVITATION },141    });142  });143144  it("T-A23: adoption flow reaches the SAS screen", async () => {145    const startAdoption = vi.fn(async ({ confirmSAS }: { confirmSAS(sas: string): Promise<boolean> }) => {146      await confirmSAS("895245");147    });148    const view = mounted(() => <AdoptVaultView onCancel={() => {}} onDone={() => {}} startAdoption={startAdoption} />);149    await enterStrongPassword(view.container);150    button(view.container, "Continue").click();151    await enterInvitation(view.container);152    button(view.container, "Connect securely").click();153    await flushView();154155    expect(view.container.textContent).toContain("Confirm this code matches");156    expect(view.container.textContent).toContain("8 9 5 2 4 5");157  });158159  it("T-A24: success leaves a vault and opens Peach", async () => {160    const onDone = vi.fn();161    const view = mounted(162      () => (163        <AdoptVaultView164          onCancel={() => {}}165          onDone={onDone}166          startAdoption={vi.fn(async () => {})}167        />168      ),169    );170    await enterStrongPassword(view.container);171    button(view.container, "Continue").click();172    await enterInvitation(view.container);173    button(view.container, "Connect securely").click();174    await flushView();175176    expect(view.container.textContent).toContain("Joined your vault.");177    button(view.container, "Open Peach").click();178    expect(onDone).toHaveBeenCalledTimes(1);179  });180181  it("T-A25: SAS cancel keeps receiver vault-less and returns to the password step", async () => {182    const createVault = vi.fn();183    const view = mounted(184      () => (185        <AdoptVaultView186          onCancel={() => {}}187          onDone={() => {}}188          startAdoption={vi.fn(async ({ confirmSAS }) => {189            const ok = await confirmSAS("895245");190            if (ok) await createVault();191            else throw new Error("sas_rejected");192          })}193        />194      ),195    );196    await enterStrongPassword(view.container);197    button(view.container, "Continue").click();198    await enterInvitation(view.container);199    button(view.container, "Connect securely").click();200    await flushView();201    button(view.container, "Cancel").click();202    await flushView();203204    expect(createVault).not.toHaveBeenCalled();205    expect(view.container.textContent).toContain("Set a password for this device.");206  });207});