import path from 'node:path'
import { expect, test, beforeEach, vi } from 'vitest'
import puppeteer from 'puppeteer-core'
import { remote } from '../../../src/index.js'
vi.mock('fetch')
vi.mock('puppeteer-core')
* Given that Puppeteer is not a direct dependency of this package, we can't mock
* it and dynamically import it. Instead, we mock the "userImport" helper and make
* it resolve to the mocked Puppeteer.
*/
vi.mock('@wdio/utils', async (origMod) => {
const orig = await origMod() as any
const puppeteer = await import('puppeteer-core')
return {
...orig,
userImport: vi.fn().mockResolvedValue(puppeteer.default)
}
})
vi.mock('@wdio/logger', () => import(path.join(process.cwd(), '__mocks__', '@wdio/logger')))
const cdpSession = new puppeteer.CDPSessionMock()
beforeEach(() => {
vi.mocked(fetch).mockClear()
cdpSession.send.mockClear()
})
test('should fail if wrong params applied', async () => {
const browser = await remote({
capabilities: {
browserName: 'devtools'
}
})
const err: Error = await browser.throttleCPU().catch((err: Error) => err)
expect(err.message).toContain('Invalid factor for "throttleCPU". Expected it to be a number (int)')
})
test('should slowdown the CPU by a factor of 2', async () => {
const browser = await remote({
capabilities: {
browserName: 'devtools'
}
})
await browser.throttleCPU(2)
expect(cdpSession.send.mock.calls).toMatchSnapshot()
})
