/** * Copyright (c) Microsoft Corporation. * * 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 { test } from '@playwright/test'; import type { TestModeName } from './testMode'; import { DefaultTestMode, DriverTestMode } from './testMode'; import * as playwrightLibrary from 'playwright-core'; export type TestModeWorkerOptions = { mode: TestModeName; }; export type TestModeTestFixtures = { toImpl: (rpcObject?: any) => any; }; export type TestModeWorkerFixtures = { toImplInWorkerScope: (rpcObject?: any) => any; playwright: typeof import('@playwright/test'); _playwrightImpl: typeof import('@playwright/test'); }; export const testModeTest = test.extend({ mode: ['default', { scope: 'worker', option: true }], _playwrightImpl: [async ({ mode }, run) => { const testMode = { 'default': new DefaultTestMode(), 'service': new DefaultTestMode(), 'service2': new DefaultTestMode(), 'service-grid': new DefaultTestMode(), 'driver': new DriverTestMode(), }[mode]; require('playwright-core/lib/utils').setUnderTest(); const playwright = await testMode.setup(); playwright._setSelectors(playwrightLibrary.selectors); await run(playwright); await testMode.teardown(); }, { scope: 'worker' }], toImplInWorkerScope: [async ({ playwright }, use) => { await use((playwright as any)._toImpl); }, { scope: 'worker' }], toImpl: async ({ toImplInWorkerScope: toImplWorker, mode }, use, testInfo) => { if (mode !== 'default' || process.env.PW_TEST_REUSE_CONTEXT) testInfo.skip(); await use(toImplWorker); }, });