add method which return current cdp

DevExpress · Mar 28, 2024 · 167fe59 · 167fe59
1 parent 4a30f1c
commit 167fe59
Show file tree

Hide file tree

Showing 16 changed files with 122 additions and 4 deletions.
diff --git a/src/api/test-controller/index.js b/src/api/test-controller/index.js
@@ -35,6 +35,7 @@ import {
     OpenWindowCommand,
     CloseWindowCommand,
     GetCurrentWindowCommand,
+    GetCurrentCDPSessionCommand,
     SwitchToWindowCommand,
     SwitchToWindowByPredicateCommand,
     SwitchToParentWindowCommand,
@@ -518,6 +519,13 @@ export default class TestController {
         return this.enqueueCommand(GetCurrentWindowCommand);
     }
 
+    [delegatedAPI(GetCurrentCDPSessionCommand.methodName)] () {
+        const callsite = getCallsiteForMethod(GetCurrentCDPSessionCommand.methodName);
+        const command  = this._createCommand(GetCurrentCDPSessionCommand, {}, callsite);
+
+        return this.testRun.executeCommand(command, callsite);
+    }
+
     [delegatedAPI(SwitchToWindowCommand.methodName)] (windowSelector) {
         this._validateMultipleWindowCommand(SwitchToWindowCommand.methodName);
 

diff --git a/src/browser/connection/index.ts b/src/browser/connection/index.ts
@@ -36,6 +36,7 @@ import { Proxy } from 'testcafe-hammerhead';
 import { NextTestRunInfo, OpenBrowserAdditionalOptions } from '../../shared/types';
 import { EventType } from '../../native-automation/types';
 import { NativeAutomationBase } from '../../native-automation';
+import remoteChrome from 'chrome-remote-interface';
 
 const getBrowserConnectionDebugScope = (id: string): string => `testcafe:browser:connection:${id}`;
 
@@ -660,6 +661,10 @@ export default class BrowserConnection extends EventEmitter {
         return this.provider.getNewWindowIdInNativeAutomation(this.id, windowId);
     }
 
+    public async getCurrentCDPSession (): Promise<remoteChrome.ProtocolApi | null> {
+        return this.provider.getCurrentCDPSession(this.id);
+    }
+
     public resetActiveWindowId (): void {
         this.provider.resetActiveWindowId(this.id);
     }

diff --git a/src/browser/provider/built-in/dedicated/chrome/index.js b/src/browser/provider/built-in/dedicated/chrome/index.js
@@ -25,7 +25,7 @@ export default {
         return getConfig(name);
     },
 
-    async _getActiveCDPClient (browserId) {
+    async getCurrentCDPSession (browserId) {
         const { browserClient } = this.openedBrowsers[browserId];
         const cdpClient         = await browserClient.getActiveClient();
 
@@ -187,19 +187,19 @@ export default {
     },
 
     async openFileProtocol (browserId, url) {
-        const cdpClient = await this._getActiveCDPClient(browserId);
+        const cdpClient = await this.getCurrentCDPSession(browserId);
 
         await navigateTo(cdpClient, url);
     },
 
     async dispatchNativeAutomationEvent (browserId, type, options) {
-        const cdpClient = await this._getActiveCDPClient(browserId);
+        const cdpClient = await this.getCurrentCDPSession(browserId);
 
         await dispatchNativeAutomationEvent(cdpClient, type, options);
     },
 
     async dispatchNativeAutomationEventSequence (browserId, eventSequence) {
-        const cdpClient = await this._getActiveCDPClient(browserId);
+        const cdpClient = await this.getCurrentCDPSession(browserId);
 
         for (const event of eventSequence) {
             if (event.type === EventType.Delay)

diff --git a/src/browser/provider/index.ts b/src/browser/provider/index.ts
@@ -17,6 +17,7 @@ import getLocalOSInfo, { OSInfo } from 'get-os-info';
 import { OpenBrowserAdditionalOptions } from '../../shared/types';
 import { EventType } from '../../native-automation/types';
 import { NativeAutomationBase } from '../../native-automation';
+import remoteChrome from 'chrome-remote-interface';
 
 
 const DEBUG_LOGGER = debug('testcafe:browser:provider');
@@ -479,6 +480,10 @@ export default class BrowserProvider {
         return this.plugin.getNativeAutomation(browserId);
     }
 
+    public async getCurrentCDPSession (browserId: string): Promise<remoteChrome.ProtocolApi | null> {
+        return this.plugin.getCurrentCDPSession(browserId);
+    }
+
     public getNewWindowIdInNativeAutomation (browserId: string, windowId: string): Promise<void> {
         return this.plugin.getNewWindowIdInNativeAutomation(browserId, windowId);
     }

diff --git a/src/browser/provider/plugin-host.js b/src/browser/provider/plugin-host.js
@@ -178,4 +178,8 @@ export default class BrowserProviderPluginHost {
     getNewWindowIdInNativeAutomation (/*browserId, windowId*/) {
         return Promise.resolve();
     }
+
+    async getCurrentCDPSession (/*browserId*/) {
+        return Promise.resolve();
+    }
 }
diff --git a/src/test-run/commands/actions.d.ts b/src/test-run/commands/actions.d.ts
@@ -67,6 +67,10 @@ export class GetCurrentWindowsCommand extends ActionCommandBase {
     public constructor(obj: object, testRun: TestRun, validateProperties?: boolean);
 }
 
+export class GetCurrentCDPSessionCommand extends ActionCommandBase {
+    public constructor(obj: object, testRun: TestRun, validateProperties?: boolean);
+}
+
 export class SwitchToWindowCommand extends ActionCommandBase {
     public constructor(obj: object, testRun: TestRun, validateProperties?: boolean);
     public windowId: string;

diff --git a/src/test-run/commands/actions.js b/src/test-run/commands/actions.js
@@ -524,6 +524,14 @@ export class GetCurrentWindowsCommand extends ActionCommandBase {
     }
 }
 
+export class GetCurrentCDPSessionCommand extends ActionCommandBase {
+    static methodName = camelCase(TYPE.getCurrentCDPSession);
+
+    constructor (obj, testRun, validateProperties) {
+        super(obj, testRun, TYPE.getCurrentCDPSession, validateProperties);
+    }
+}
+
 export class SwitchToWindowCommand extends ActionCommandBase {
     static methodName = camelCase(TYPE.switchToWindow);
 

diff --git a/src/test-run/commands/type.js b/src/test-run/commands/type.js
@@ -41,6 +41,7 @@ export default {
     closeWindow:                         'close-window',
     getCurrentWindow:                    'get-current-window',
     getCurrentWindows:                   'get-current-windows',
+    getCurrentCDPSession:                'get-current-c-d-p-session',
     switchToWindow:                      'switch-to-window',
     switchToWindowByPredicate:           'switch-to-window-by-predicate',
     switchToParentWindow:                'switch-to-parent-window',

diff --git a/src/test-run/index.ts b/src/test-run/index.ts
@@ -136,6 +136,7 @@ import {
 import NativeAutomationRequestPipeline from '../native-automation/request-pipeline';
 import { NativeAutomationBase } from '../native-automation';
 import ReportDataLog from '../reporter/report-data-log';
+import remoteChrome from 'chrome-remote-interface';
 
 const lazyRequire                   = require('import-lazy')(require);
 const ClientFunctionBuilder         = lazyRequire('../client-functions/client-function-builder');
@@ -445,6 +446,10 @@ export default class TestRun extends AsyncEventEmitter {
             : this.testExecutionTimeout || null;
     }
 
+    public async getCurrentCDPSession (): Promise<remoteChrome.ProtocolApi | null> {
+        return this.browserConnection.getCurrentCDPSession();
+    }
+
     private _addClientScriptContentWarningsIfNecessary (): void {
         const { empty, duplicatedContent } = findProblematicScripts(this.test.clientScripts as ClientScript[]);
 
@@ -1227,6 +1232,9 @@ export default class TestRun extends AsyncEventEmitter {
         if (command.type === COMMAND_TYPE.removeRequestHooks)
             return Promise.all((command as RemoveRequestHooksCommand).hooks.map(hook => this._removeRequestHook(hook)));
 
+        if (command.type === COMMAND_TYPE.getCurrentCDPSession)
+            return this.getCurrentCDPSession();
+
         return this._enqueueCommand(command, callsite as CallsiteRecord);
     }
 

diff --git a/test/functional/fixtures/api/es-next/get-current-cdp-session/pages/child.html b/test/functional/fixtures/api/es-next/get-current-cdp-session/pages/child.html
@@ -0,0 +1,10 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <title>Title</title>
+</head>
+<body>
+    <h1>new window</h1>
+</body>
+</html>
diff --git a/test/functional/fixtures/api/es-next/get-current-cdp-session/pages/index.html b/test/functional/fixtures/api/es-next/get-current-cdp-session/pages/index.html
@@ -0,0 +1,10 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <title>Title</title>
+</head>
+<body>
+    <h1><a href="child.html" target="_blank">Open new window</a></h1>
+</body>
+</html>
diff --git a/test/functional/fixtures/api/es-next/get-current-cdp-session/test.js b/test/functional/fixtures/api/es-next/get-current-cdp-session/test.js
@@ -0,0 +1,7 @@
+const { onlyInNativeAutomation } = require('../../../../utils/skip-in');
+
+describe('[API] Get current CDP session', function () {
+    onlyInNativeAutomation('Should return current CPD', function () {
+        return runTests('./testcafe-fixtures/index.js', 'Get current CDP session', { experimentalMultipleWindows: true });
+    });
+});
diff --git a/test/functional/fixtures/api/es-next/get-current-cdp-session/testcafe-fixtures/index.js b/test/functional/fixtures/api/es-next/get-current-cdp-session/testcafe-fixtures/index.js
@@ -0,0 +1,19 @@
+
+fixture `Get current CDP session`
+    .page `http://localhost:3000/fixtures/api/es-next/get-current-cdp-session/pages/index.html`;
+
+test(`Get current CDP session`, async t => {
+    const mainWindowId = await t.testRun.activeWindowId;
+
+    let clientCDP = await t.getCurrentCDPSession();
+
+    await t.expect(clientCDP.webSocketUrl).contains(mainWindowId);
+
+    await t.click('a').click('h1');
+
+    const childWindowId = await t.testRun.activeWindowId;
+
+    clientCDP = await t.getCurrentCDPSession();
+
+    await t.expect(clientCDP.webSocketUrl).contains(childWindowId);
+});
diff --git a/test/server/data/test-controller-reporter-expected/index.js b/test/server/data/test-controller-reporter-expected/index.js
@@ -649,6 +649,24 @@ module.exports = {
             },
             browser:   { alias: 'test-browser', headless: false },
         },
+        {
+            testRunId: 'test-run-id',
+            name:      'getCurrentCDPSession',
+            command:   {
+                type:     'get-current-c-d-p-session',
+                actionId: 'GetCurrentCDPSessionCommand',
+            },
+            test:      {
+                id:    'test-id',
+                name:  'test-name',
+                phase: 'initial',
+            },
+            fixture:   {
+                id:   'fixture-id',
+                name: 'fixture-name',
+            },
+            browser:   { alias: 'test-browser', headless: false },
+        },
         {
             testRunId: 'test-run-id',
             name:      'switchToParentWindow',

diff --git a/test/server/test-controller-events-test.js b/test/server/test-controller-events-test.js
@@ -128,6 +128,7 @@ const actions = {
     switchToWindow:            [{ id: 'window-id' }],
     closeWindow:               [{ id: 'window-id' }],
     getCurrentWindow:          [],
+    getCurrentCDPSession:      [],
     switchToParentWindow:      [],
     switchToPreviousWindow:    [],
     setNativeDialogHandler:    [() => true],

diff --git a/ts-defs-src/test-api/test-controller.d.ts b/ts-defs-src/test-api/test-controller.d.ts
@@ -90,6 +90,8 @@ interface Browser {
 
 type WindowDescriptor = unknown;
 
+type CDPSession = unknown;
+
 interface WindowFilterData {
     /**
      * The window title.
@@ -413,6 +415,11 @@ interface TestController {
      * Retrieves a `window` object that corresponds to the currently open window.
      */
     getCurrentWindow(): WindowDescriptorPromise;
+
+    /**
+     * Retrieves a `Chrome DevTools Protocol` object that corresponds to the currently open window in native automation.
+     */
+    getCurrentCDPSession(): CDPSessionPromise;
 
     /**
      * Activates the window that corresponds to the `window` object.
@@ -576,3 +583,6 @@ interface TestControllerPromise<T=any> extends TestController, Promise<T> {
 interface WindowDescriptorPromise extends TestControllerPromise<WindowDescriptor> {
 }
 
+interface CDPSessionPromise extends TestControllerPromise<CDPSession> {
+}
+