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>
83 lines
2.9 KiB
TypeScript
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();
|
|
});
|
|
});
|