run-time.test.ts

import { launchBrowser, MOCK_HOST, MOCK_URL, sleep, type LaunchContext } from "../common/base"
import { parseTime2Sec, readRecordsOfFirstPage } from "../common/record"
import { createWhitelist } from "../common/whitelist"

let context: LaunchContext

async function clickRunTimeChange(siteHost: string): Promise<void> {
    const sitePage = await context.openAppPage("/additional/site-manage")
    await sitePage.focus('input[placeholder]')
    await sitePage.keyboard.type(siteHost)
    await sitePage.keyboard.press('Enter')
    await sleep(.1)
    await sitePage.evaluate(() => {
        const runTimeSwitch = document.querySelector<HTMLDivElement>('table > tbody > tr > td.el-table_1_column_7 .el-switch')
        runTimeSwitch?.click()
    })
}

describe('Run time tracking', () => {
    beforeEach(async () => context = await launchBrowser())

    afterEach(() => context.close())

    test('Basically track', async () => {
        await context.newPageAndWaitCsInjected(MOCK_URL)
        await sleep(1.1)
        let records = await readRecordsOfFirstPage(context)
        let record = records[0]
        expect(parseTime2Sec(record.time)).toBeGreaterThanOrEqual(1)
        expect(record.runTime).toBeFalsy()

        // 1. Enable run time tracking
        const enableTs = Date.now()
        await clickRunTimeChange(MOCK_HOST)

        // 2. Sleep
        const emptyPage = await context.newPage()
        await sleep(1.1)

        records = await readRecordsOfFirstPage(context)
        const runTime1 = parseTime2Sec(records[0]?.runTime) ?? 0
        expect(runTime1).toBeGreaterThanOrEqual(1)

        // 3. Add another page sharing the same run time with old page
        await context.newPageAndWaitCsInjected(MOCK_URL)
        // jump to new page
        await emptyPage.bringToFront()
        await sleep(1)

        records = await readRecordsOfFirstPage(context)
        const runTime2 = parseTime2Sec(records[0].runTime)
        expect(runTime2).toBeGreaterThanOrEqual(runTime1 + 1)
        expect(runTime2).toBeLessThan((Date.now() - enableTs) / 1000)

        // 3. Disable run time tracking
        await clickRunTimeChange(MOCK_HOST)
        const disableTs = Date.now()
        await emptyPage.bringToFront()
        await sleep(4)
        records = await readRecordsOfFirstPage(context)
        const runTime3 = parseTime2Sec(records[0].runTime)
        expect(runTime3).toBeLessThanOrEqual(Math.round((disableTs - enableTs) / 1000))
    }, 60000)

    test('white list', async () => {
        await context.newPage(MOCK_URL)

        // 1. Enable
        await clickRunTimeChange(MOCK_HOST)
        const enableTs = Date.now()
        await sleep(4)

        let records = await readRecordsOfFirstPage(context)
        const runTime = parseTime2Sec(records[0].runTime)
        expect(runTime).toBeTruthy()
        expect(runTime).toBeLessThanOrEqual((Date.now() - enableTs) / 1000 + 1)

        // 2. Add whitelist
        await createWhitelist(context, MOCK_HOST)
        const disableTs = Date.now()

        await sleep(2)

        records = await readRecordsOfFirstPage(context)
        const runTime1 = parseTime2Sec(records[0].runTime)
        expect(runTime1).toBeLessThan((disableTs - enableTs) / 1000 + 1)
    }, 60000)
})