55 lines
2.0 KiB
TypeScript
55 lines
2.0 KiB
TypeScript
|
import { test as playwrightTestFn } from '@playwright/test'
|
||
|
|
import { fixtures, Fixtures, AuthenticatedTronApp } from './fixtures/fixtureSetup'
|
||
|
|
export { expect, Page, BrowserContext, TestInfo } from '@playwright/test'
|
||
|
|
|
||
|
|
// Our custom decorated Zoo test object. Makes it easier to add fixtures, and
|
||
|
|
// switch between web and electron if needed.
|
||
|
|
const pwTestFnWithFixtures = playwrightTestFn.extend<Fixtures>(fixtures)
|
||
|
|
|
||
|
|
export function test(desc, objOrFn, fnMaybe) {
|
||
|
|
const hasTestConf = typeof objOrFn === 'object'
|
||
|
|
const fn = hasTestConf ? fnMaybe : objOrFn
|
||
|
|
|
||
|
|
return pwTestFnWithFixtures(desc, hasTestConf ? objOrFn : {}, async ({ page, context, cmdBar, editor, toolbar, scene, homePage }, testInfo) => {
|
||
|
|
// To switch to web, change this to AuthenticatedApp from fixtureSetup.ts
|
||
|
|
const tronApp = new AuthenticatedTronApp(
|
||
|
|
context,
|
||
|
|
page,
|
||
|
|
testInfo
|
||
|
|
)
|
||
|
|
|
||
|
|
const fixtures: Fixtures = { cmdBar, editor, toolbar, scene, homePage }
|
||
|
|
await tronApp.initialise({ fixtures })
|
||
|
|
|
||
|
|
// We need to patch this because addInitScript will bind too late in our
|
||
|
|
// electron tests, never running. We need to call reload() after each call
|
||
|
|
// to guarantee it runs.
|
||
|
|
const oldContextAddInitScript = tronApp.context.addInitScript
|
||
|
|
tronApp.context.addInitScript = async function(a, b) {
|
||
|
|
await oldContextAddInitScript.apply(this, [a, b])
|
||
|
|
await tronApp.page.reload()
|
||
|
|
}
|
||
|
|
|
||
|
|
// No idea why we mix and match page and context's addInitScript but we do
|
||
|
|
const oldPageAddInitScript = tronApp.page.addInitScript
|
||
|
|
tronApp.page.addInitScript = async function(a, b) {
|
||
|
|
await oldPageAddInitScript.apply(this, [a, b])
|
||
|
|
await tronApp.page.reload()
|
||
|
|
}
|
||
|
|
|
||
|
|
await fn({
|
||
|
|
context: tronApp.context,
|
||
|
|
page: tronApp.page,
|
||
|
|
...fixtures
|
||
|
|
}, testInfo)
|
||
|
|
|
||
|
|
testInfo.tronApp = tronApp
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
test.describe = pwTestFnWithFixtures.describe
|
||
|
|
test.beforeEach = pwTestFnWithFixtures.beforeEach
|
||
|
|
test.afterEach = pwTestFnWithFixtures.afterEach
|
||
|
|
test.step = pwTestFnWithFixtures.step
|
||
|
|
test.skip = pwTestFnWithFixtures.skip
|