mostlymatter/e2e-tests/playwright/support/browser_context.ts

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

import {writeFile} from 'node:fs/promises';

import {request, Browser, BrowserContext} from '@playwright/test';

import {UserProfile} from '@mattermost/types/users';
import testConfig from '@e2e-test.config';
import pages from '@e2e-support/ui/pages';

export class TestBrowser {
    readonly browser: Browser;
    context: BrowserContext | null;

    constructor(browser: Browser) {
        this.browser = browser;
        this.context = null;
    }

    async login(user: UserProfile) {
        const options = {storageState: ''};
        if (user) {
            // Log in via API request and save user storage
            const storagePath = await loginByAPI(user.username, user.password);
            options.storageState = storagePath;
        }

        // Sign in a user in new browser context
        const context = await this.browser.newContext(options);
        const page = await context.newPage();

        const channelsPage = new pages.ChannelsPage(page);
        const systemConsolePage = new pages.SystemConsolePage(page);
        const scheduledDraftPage = new pages.ScheduledDraftPage(page);
        const draftPage = new pages.DraftPage(page);

        this.context = context;

        return {context, page, channelsPage, systemConsolePage, scheduledDraftPage, draftPage};
    }

    async close() {
        if (this.context) {
            await this.context.close();
        }
    }
}

export async function loginByAPI(loginId: string, password: string, token = '', ldapOnly = false) {
    const requestContext = await request.newContext();

    const data: any = {
        login_id: loginId,
        password,
        token,
        deviceId: '',
    };

    if (ldapOnly) {
        data.ldap_only = 'true';
    }

    // Log in via API
    await requestContext.post(`${testConfig.baseURL}/api/v4/users/login`, {
        data,
        headers: {'X-Requested-With': 'XMLHttpRequest'},
    });

    // Save signed-in state to a folder
    const storagePath = `storage_state/${Date.now()}_${loginId}_${password}${token ? '_' + token : ''}${
        ldapOnly ? '_ldap' : ''
    }.json`;
    const storageState = await requestContext.storageState({path: storagePath});
    await requestContext.dispose();

    // Append origins to bypass seeing landing page then write to file
    storageState.origins.push({
        origin: testConfig.baseURL,
        localStorage: [{name: '__landingPageSeen__', value: 'true'}],
    });
    await writeFile(storagePath, JSON.stringify(storageState));

    return storagePath;
}