1 files changed, 139 insertions, 0 deletions
diff --git a/remote/test/puppeteer/test/src/CDPSession.spec.ts b/remote/test/puppeteer/test/src/CDPSession.spec.ts
new file mode 100644
index 0000000000..b059e72723
--- /dev/null
+++ b/remote/test/puppeteer/test/src/CDPSession.spec.ts
@@ -0,0 +1,139 @@
+/**
+ * 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 expect from 'expect';
+import {isErrorLike} from 'puppeteer-core/internal/util/ErrorLike.js';
+
+import {
+  getTestState,
+  setupTestBrowserHooks,
+  setupTestPageAndContextHooks,
+} from './mocha-utils.js';
+import {waitEvent} from './utils.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: unknown[] = [];
+    client.on('Network.requestWillBeSent', event => {
+      return events.push(event);
+    });
+    await Promise.all([
+      waitEvent(client, 'Network.requestWillBeSent'),
+      page.goto(server.EMPTY_PAGE),
+    ]);
+    expect(events).toHaveLength(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();
+  });
+});