SnowLuma/packages/webui/tests/login-totp.test.tsx

import { renderToStaticMarkup } from 'react-dom/server';
import { describe, expect, it, vi } from 'vitest';
import { LoginPage } from '../src/components/pages/login-page';

vi.mock('@/contexts/ThemeContext', () => ({
    useTheme: () => ({
        appearance: {
            background: { type: 'none' },
            reduceMotion: true,
            disableMotion: true,
            palette: 'default',
        },
        mode: 'light',
        setMode: () => undefined,
    }),
    paletteResolved: () => false,
}));

describe('LoginPage 2FA step', () => {
    it('keeps the password form on its own screen', () => {
        const markup = renderToStaticMarkup(
            <LoginPage onLogin={async () => ({ success: false, error: 'no' })} />,
        );
        expect(markup).toContain('输入访问令牌');
        expect(markup).toContain('进入控制台');
        expect(markup).toContain('max-w-md');
        expect(markup).not.toContain('max-w-2xl');
        expect(markup).not.toContain('character 1 of 6');
        expect(markup).not.toContain('验证并进入');
    });

    it('shows a separate TOTP screen after the password factor succeeds', () => {
        const markup = renderToStaticMarkup(
            <LoginPage
                initialPassword="Correct-Horse-1!"
                initialNeedsTotp
                onLogin={async () => ({ success: false, needsTotp: true })}
            />,
        );
        expect(markup).not.toContain('输入访问令牌');
        expect(markup).toContain('输入 Authenticator 中的 6 位验证码');
        expect(markup).not.toContain('验证并进入');
        expect(markup).toContain('character 1 of 6');
        expect(markup).toContain('可以把完整验证码粘贴到任意一格。');
        expect(markup).toContain('使用恢复码');
        expect(markup).toContain('返回');
        expect(markup).toContain('max-w-md');
        expect(markup).not.toContain('max-w-2xl');
        expect(markup).not.toContain('h-[2px] w-3 rounded-full');
    });

    it('uses the same cell UI for a recovery code instead of a single field', () => {
        const markup = renderToStaticMarkup(
            <LoginPage
                initialPassword="Correct-Horse-1!"
                initialNeedsTotp
                initialUseRecovery
                onLogin={async () => ({ success: false, needsTotp: true })}
            />,
        );
        expect(markup).toContain('输入保存的一次性恢复码');
        expect(markup).toContain('character 1 of 8');
        expect(markup).toContain('character 8 of 8');
        expect(markup).toContain('可以把完整恢复码粘贴到任意一格');
        expect(markup).toContain('改用验证码');
        expect(markup).toContain('max-w-2xl');
        expect(markup).toContain('h-[2px] w-3 rounded-full');
        expect(markup).not.toContain('XXXX-XXXX');
        expect(markup).not.toContain('验证并进入');
        expect(markup).not.toContain('输入访问令牌');
    });
});