@wppconnect/wa-js/tests/wpp-test.ts

WPPConnect/WA-JS is an open-source project with the aim of exporting functions from WhatsApp Web

/*!
 * Copyright 2021 WPPConnect Team
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import {
  BrowserContext,
  chromium,
  firefox,
  Page,
  test as base,
  webkit,
} from '@playwright/test';
import * as os from 'os';
import * as path from 'path';

export { expect } from '@playwright/test';

export type TestOptions = {
  loggedContext: BrowserContext;
  loggedPage: Page;
};

const prepareWhatsAppPage = async (page: Page) => {
  page.on('domcontentloaded', async () => {
    await page.addScriptTag({
      path: path.join(__dirname, '../dist/wppconnect-wa.js'),
    });
  });

  await page.goto('https://web.whatsapp.com/', {
    waitUntil: 'domcontentloaded',
  });

  await page.waitForFunction(() => WPP && WPP.isReady);
};

export const test = base.extend<TestOptions>({
  page: async ({ page }, use) => {
    await prepareWhatsAppPage(page);

    await use(page);
  },
  loggedContext: async ({ browserName, contextOptions }, use) => {
    const userDataDir = path.join(os.tmpdir(), `wa-js-test-${browserName}`);

    const browsers = { chromium, firefox, webkit };

    const loggedContext = await browsers[browserName].launchPersistentContext(
      userDataDir,
      contextOptions
    );

    await use(loggedContext);
    await loggedContext.close();
  },
  loggedPage: async ({ loggedContext: cleanContext }, use) => {
    const cleanPage = cleanContext.pages().length
      ? cleanContext.pages()[0]
      : await cleanContext.newPage();

    await prepareWhatsAppPage(cleanPage);

    await use(cleanPage);
    await cleanPage.close();
  },
});