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