Files
relicario/extension/src/setup/__tests__/setup.test.ts
adlee-was-taken 8044310fba test(ext/setup): cover SetupStep registry shape + clearWizardState (Plan C Phase 3)
Asserts STEPS has the six steps in canonical order, each renders non-empty
HTML and returns a teardown from attach, and clearWizardState zero-fills the
reachable Uint8Array fields before resetting state. Keeps the existing
finishSetup tests.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-05-31 18:11:53 -04:00

83 lines
2.9 KiB
TypeScript

import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
import { finishSetup, STEPS } from '../setup';
import { state, clearWizardState } from '../setup-steps';
describe('finishSetup', () => {
beforeEach(() => {
(global as any).chrome = {
tabs: {
create: vi.fn(() => Promise.resolve({ id: 999 })),
getCurrent: vi.fn(() => Promise.resolve({ id: 42 })),
remove: vi.fn(() => Promise.resolve()),
},
runtime: {
getURL: vi.fn((p: string) => `chrome-extension://abc/${p}`),
},
};
});
it('opens vault.html in a new tab', async () => {
await finishSetup();
expect(chrome.runtime.getURL).toHaveBeenCalledWith('vault.html');
expect(chrome.tabs.create).toHaveBeenCalledWith({
url: 'chrome-extension://abc/vault.html',
});
});
it('closes the current setup tab after opening the vault tab', async () => {
await finishSetup();
expect(chrome.tabs.getCurrent).toHaveBeenCalled();
expect(chrome.tabs.remove).toHaveBeenCalledWith(42);
});
it('still opens the vault tab even if closing the setup tab fails', async () => {
(chrome.tabs.remove as any).mockRejectedValueOnce(new Error('no permission'));
await expect(finishSetup()).resolves.not.toThrow();
expect(chrome.tabs.create).toHaveBeenCalled();
});
});
describe('setup step registry', () => {
it('has the six steps in canonical order', () => {
expect(STEPS.map((s) => s.id)).toEqual(['mode', 'host', 'connection', 'vault', 'device', 'done']);
});
it('each step renders non-empty HTML and attach returns a teardown', () => {
const ctx = { state: {} as never, rerender: vi.fn(), goto: vi.fn() };
for (const step of STEPS) {
const html = step.render(ctx as never);
expect(typeof html).toBe('string');
expect(html.length).toBeGreaterThan(0);
// render output must be in the DOM before attach (attach wires getElementById listeners)
document.body.innerHTML = `<div id="app">${html}</div>`;
const teardown = step.attach(document.body, ctx as never);
expect(typeof teardown).toBe('function');
teardown(); // must not throw
}
});
});
describe('clearWizardState', () => {
afterEach(() => {
clearWizardState();
});
it('zero-fills the reachable Uint8Array fields and resets state', () => {
const carrier = new Uint8Array([1, 2, 3, 4]);
const ref = new Uint8Array([5, 6, 7, 8]);
state.carrierImageBytes = carrier;
state.referenceImageBytes = ref;
state.passphrase = 'secret';
state.mode = 'new';
clearWizardState();
expect(Array.from(carrier)).toEqual([0, 0, 0, 0]); // fill(0) ran on the captured ref
expect(Array.from(ref)).toEqual([0, 0, 0, 0]);
expect(state.carrierImageBytes).toBeNull(); // field reset
expect(state.referenceImageBytes).toBeNull();
expect(state.passphrase).toBe('');
expect(state.mode).toBeNull();
});
});