/** * Copyright 2018 Google Inc. All rights reserved. * * 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 {waitEvent} from './utils.js'; import expect from 'expect'; import { getTestState, setupTestBrowserHooks, setupTestPageAndContextHooks, } from './mocha-utils.js'; import {isErrorLike} from 'puppeteer-core/internal/util/ErrorLike.js'; describe('Target.createCDPSession', function () { setupTestBrowserHooks(); setupTestPageAndContextHooks(); it('should work', async () => { const {page} = getTestState(); const client = await page.target().createCDPSession(); await Promise.all([ client.send('Runtime.enable'), client.send('Runtime.evaluate', {expression: 'window.foo = "bar"'}), ]); const foo = await page.evaluate(() => { return (globalThis as any).foo; }); expect(foo).toBe('bar'); }); it('should not report created targets for custom CDP sessions', async () => { const {browser} = getTestState(); let called = 0; browser.browserContexts()[0]!.on('targetcreated', async target => { called++; if (called > 1) { throw new Error('Too many targets created'); } await target.createCDPSession(); }); await browser.newPage(); }); it('should send events', async () => { const {page, server} = getTestState(); const client = await page.target().createCDPSession(); await client.send('Network.enable'); const events = []; client.on('Network.requestWillBeSent', event => { return events.push(event); }); await page.goto(server.EMPTY_PAGE); expect(events.length).toBe(1); }); it('should enable and disable domains independently', async () => { const {page} = getTestState(); const client = await page.target().createCDPSession(); await client.send('Runtime.enable'); await client.send('Debugger.enable'); // JS coverage enables and then disables Debugger domain. await page.coverage.startJSCoverage(); await page.coverage.stopJSCoverage(); // generate a script in page and wait for the event. const [event] = await Promise.all([ waitEvent(client, 'Debugger.scriptParsed'), page.evaluate('//# sourceURL=foo.js'), ]); // expect events to be dispatched. expect(event.url).toBe('foo.js'); }); it('should be able to detach session', async () => { const {page} = getTestState(); const client = await page.target().createCDPSession(); await client.send('Runtime.enable'); const evalResponse = await client.send('Runtime.evaluate', { expression: '1 + 2', returnByValue: true, }); expect(evalResponse.result.value).toBe(3); await client.detach(); let error!: Error; try { await client.send('Runtime.evaluate', { expression: '3 + 1', returnByValue: true, }); } catch (error_) { if (isErrorLike(error_)) { error = error_ as Error; } } expect(error.message).toContain('Session closed.'); }); it('should throw nice errors', async () => { const {page} = getTestState(); const client = await page.target().createCDPSession(); const error = await theSourceOfTheProblems().catch(error => { return error; }); expect(error.stack).toContain('theSourceOfTheProblems'); expect(error.message).toContain('ThisCommand.DoesNotExist'); async function theSourceOfTheProblems() { // @ts-expect-error This fails in TS as it knows that command does not // exist but we want to have this tests for our users who consume in JS // not TS. await client.send('ThisCommand.DoesNotExist'); } }); it('should expose the underlying connection', async () => { const {page} = getTestState(); const client = await page.target().createCDPSession(); expect(client.connection()).toBeTruthy(); }); });