mattermost/e2e-tests/cypress/tests/support/ui/mfa.js

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.

import authenticator from 'authenticator';

import * as TIMEOUTS from '../../fixtures/timeouts';

Cypress.Commands.add('uiGetMFASecret', (userId) => {
    return cy.url().then((url) => {
        if (url.includes('mfa/setup')) {
            // # Complete MFA setup if we are on token setup page /mfa/setup
            return cy.get('#mfa').wait(TIMEOUTS.HALF_SEC).find('.col-sm-12').then((p) => {
                const secretp = p.text();
                const secret = secretp.split(' ')[1];

                const token = authenticator.generateToken(secret);
                cy.findByPlaceholderText('MFA Code').type(token);
                cy.findByText('Save').click();

                cy.wait(TIMEOUTS.HALF_SEC);
                cy.findByText('Okay').click();

                return cy.wrap(secret);
            });
        }

        // # If the user already has MFA enabled, reset the secret.
        return cy.apiGenerateMfaSecret(userId).then((res) => {
            return cy.wrap(res.code.secret);
        });
    });
});