accounts-frontend/packages/app/services/authFlow/PasswordState.test.ts

import expect from 'app/test/unexpected';
import sinon, { SinonMock } from 'sinon';

import PasswordState from 'app/services/authFlow/PasswordState';
import CompleteState from 'app/services/authFlow/CompleteState';
import MfaState from 'app/services/authFlow/MfaState';
import LoginState from 'app/services/authFlow/LoginState';
import ForgotPasswordState from 'app/services/authFlow/ForgotPasswordState';
import ChooseAccountState from 'app/services/authFlow/ChooseAccountState';

import { bootstrap, expectState, expectNavigate, expectRun, MockedAuthContext } from './helpers';

describe('PasswordState', () => {
    let state: PasswordState;
    let context: MockedAuthContext;
    let mock: SinonMock;

    beforeEach(() => {
        state = new PasswordState();

        const data = bootstrap();
        context = data.context;
        mock = data.mock;
    });

    afterEach(() => {
        mock.verify();
    });

    describe('#enter', () => {
        it('should navigate to /password', () => {
            context.getState.returns({
                user: { isGuest: true },
                auth: {
                    credentials: { login: 'foo' },
                },
            });

            expectNavigate(mock, '/password');

            state.enter(context);
        });

        it('should transition to complete if not guest', () => {
            context.getState.returns({
                user: { isGuest: false },
                auth: {
                    credentials: { login: null },
                },
            });

            expectState(mock, CompleteState);

            state.enter(context);
        });
    });

    describe('#resolve', () => {
        it('should call login with login and password', () => {
            const expectedLogin = 'foo';
            const expectedPassword = 'bar';
            const expectedRememberMe = true;

            context.getState.returns({
                auth: {
                    credentials: {
                        login: expectedLogin,
                    },
                },
            });

            expectRun(mock, 'setAccountSwitcher', false);
            expectRun(
                mock,
                'login',
                sinon.match({
                    login: expectedLogin,
                    password: expectedPassword,
                    rememberMe: expectedRememberMe,
                }),
            ).returns(Promise.resolve());
            expectState(mock, CompleteState);

            const payload = {
                password: expectedPassword,
                rememberMe: expectedRememberMe,
            };

            return expect(state.resolve(context, payload), 'to be fulfilled');
        });

        it('should go to MfaState if totp required', () => {
            const expectedLogin = 'foo';
            const expectedPassword = 'bar';
            const expectedRememberMe = true;

            context.getState.returns({
                auth: {
                    credentials: {
                        login: expectedLogin,
                        isTotpRequired: true,
                    },
                },
            });

            // Should not run "setAccountSwitcher"

            expectRun(
                mock,
                'login',
                sinon.match({
                    login: expectedLogin,
                    password: expectedPassword,
                    rememberMe: expectedRememberMe,
                }),
            ).returns(Promise.resolve());
            expectState(mock, MfaState);

            const payload = {
                password: expectedPassword,
                rememberMe: expectedRememberMe,
            };

            return expect(state.resolve(context, payload), 'to be fulfilled');
        });

        it('should navigate to returnUrl if any', () => {
            const expectedLogin = 'foo';
            const expectedPassword = 'bar';
            const expectedReturnUrl = '/returnUrl';
            const expectedRememberMe = true;

            context.getState.returns({
                auth: {
                    credentials: {
                        login: expectedLogin,
                        returnUrl: expectedReturnUrl,
                    },
                },
            });

            expectRun(mock, 'setAccountSwitcher', false);
            expectRun(
                mock,
                'login',
                sinon.match({
                    login: expectedLogin,
                    password: expectedPassword,
                    rememberMe: expectedRememberMe,
                }),
            ).returns(Promise.resolve());
            expectNavigate(mock, expectedReturnUrl);

            const payload = {
                password: expectedPassword,
                rememberMe: expectedRememberMe,
            };

            return expect(state.resolve(context, payload), 'to be fulfilled');
        });
    });

    describe('#reject', () => {
        it('should transition to forgot password state', () => {
            expectState(mock, ForgotPasswordState);

            state.reject(context);
        });
    });

    describe('#goBack', () => {
        it('should transition to login state', () => {
            context.getState.returns({
                auth: {
                    credentials: {
                        login: 'foo',
                    },
                },
            });

            expectRun(mock, 'setLogin', null);
            expectState(mock, LoginState);

            state.goBack(context);
        });

        it('should transition to ChooseAccountState if this is relogin', () => {
            context.getState.returns({
                accounts: {
                    active: 1,
                    available: [{ id: 1 }, { id: 2 }],
                },
                auth: {
                    credentials: {
                        login: 'foo',
                        isRelogin: true,
                    },
                },
            });

            // Should not run "setAccountSwitcher"
            expectRun(mock, 'activateAccount', { id: 2 });
            expectRun(mock, 'removeAccount', { id: 1 });
            expectState(mock, ChooseAccountState);

            state.goBack(context);
        });

        it('should transition to LoginState if this is relogin and only one account available', () => {
            context.getState.returns({
                accounts: {
                    active: 1,
                    available: [{ id: 1 }],
                },
                auth: {
                    credentials: {
                        login: 'foo',
                        isRelogin: true,
                    },
                },
            });

            expectRun(mock, 'logout');
            expectRun(mock, 'setLogin', null);
            expectState(mock, LoginState);

            state.goBack(context);
        });
    });
});