feat: Support for editor action buttons with dropdown

Signed-off-by: Dominik Jelinek <[email protected]> test Signed-off-by: Dominik Jelinek <[email protected]> test 2 Signed-off-by: Dominik Jelinek <[email protected]> test Signed-off-by: Dominik Jelinek <[email protected]>
redhat-developer · Jul 17, 2024 · 9bffd73 · 9bffd73
1 parent 3b452e7
commit 9bffd73
Show file tree

Hide file tree

Showing 6 changed files with 80 additions and 7 deletions.
diff --git a/packages/locators/lib/1.37.0.ts b/packages/locators/lib/1.37.0.ts
@@ -119,6 +119,7 @@ const editor = {
 		actionContainer: By.className('editor-actions'),
 		actionItem: By.className('action-label'),
 		attribute: 'title',
+		dropdown: 'aria-haspopup',
 	},
 	Editor: {
 		constructor: By.className('editor-instance'),

diff --git a/packages/page-objects/src/components/editor/EditorAction.ts b/packages/page-objects/src/components/editor/EditorAction.ts
@@ -16,10 +16,12 @@
  */
 
 import { EditorGroup } from './EditorView';
-import { AbstractElement } from '../AbstractElement';
-import { WebElement } from '../..';
+import { By, ContextMenu, Key, WebElement } from '../..';
+import { ElementWithContexMenu } from '../ElementWithContextMenu';
+import { ChromiumWebDriver } from 'selenium-webdriver/chromium';
+
+export class EditorAction extends ElementWithContexMenu {
 
-export class EditorAction extends AbstractElement {
 	constructor(element: WebElement, parent: EditorGroup) {
 		super(element, parent);
 	}
@@ -31,3 +33,35 @@ export class EditorAction extends AbstractElement {
 		return await this.getAttribute(EditorAction.locators.EditorView.attribute);
 	}
 }
+
+export class EditorActionDropdown extends EditorAction {
+	async open(): Promise<ContextMenu> {
+		await this.click();
+		const shadowRootHost = await this.enclosingItem.findElements(By.className('shadow-root-host'));
+		const actions = this.getDriver().actions();
+		await actions.clear();
+		await actions.sendKeys(Key.ESCAPE).perform();
+		const webdriverCapabilities = await (this.getDriver() as ChromiumWebDriver).getCapabilities();
+		const chromiumVersion = webdriverCapabilities.getBrowserVersion();
+		if (shadowRootHost.length > 0) {
+			if ((await this.getAttribute('aria-expanded')) !== 'true') {
+				await this.click();
+			}
+			let shadowRoot;
+			const webdriverCapabilities = await (this.getDriver() as ChromiumWebDriver).getCapabilities();
+			const chromiumVersion = webdriverCapabilities.getBrowserVersion();
+			if (chromiumVersion && parseInt(chromiumVersion.split('.')[0]) >= 96) {
+				shadowRoot = await shadowRootHost[0].getShadowRoot();
+				return new ContextMenu(await shadowRoot.findElement(By.className('monaco-menu-container'))).wait();
+			} else {
+				shadowRoot = (await this.getDriver().executeScript('return arguments[0].shadowRoot', shadowRootHost[0])) as WebElement;
+				return new ContextMenu(shadowRoot).wait();
+			}
+		} else if (chromiumVersion && parseInt(chromiumVersion.split('.')[0]) >= 100) {
+			await this.click();
+			const workbench = await this.getDriver().findElement(ElementWithContexMenu.locators.Workbench.constructor);
+			return new ContextMenu(workbench).wait();
+		}
+		return await super.openContextMenu();
+	}
+}
diff --git a/packages/page-objects/src/components/editor/EditorView.ts b/packages/page-objects/src/components/editor/EditorView.ts
@@ -21,7 +21,7 @@ import { AbstractElement } from '../AbstractElement';
 import { ElementWithContexMenu } from '../ElementWithContextMenu';
 import { DiffEditor } from './DiffEditor';
 import { Editor } from './Editor';
-import { EditorAction } from './EditorAction';
+import { EditorAction, EditorActionDropdown } from './EditorAction';
 import { SettingsEditor } from './SettingsEditor';
 import { WebView } from './WebView';
 
@@ -339,7 +339,16 @@ export class EditorGroup extends AbstractElement {
 	 */
 	async getActions(): Promise<EditorAction[]> {
 		const actions = await this.findElement(EditorGroup.locators.EditorView.actionContainer).findElements(EditorGroup.locators.EditorView.actionItem);
-		return actions.map((action) => new EditorAction(action, this));
+		return Promise.all(
+			actions.map(async (action) => {
+				const dropdown = await action.getAttribute(EditorGroup.locators.EditorView.dropdown);
+				if (dropdown) {
+					return new EditorActionDropdown(action, this);
+				} else {
+					return new EditorAction(action, this);
+				}
+			}),
+		);
 	}
 
 	/**

diff --git a/packages/page-objects/src/locators/locators.ts b/packages/page-objects/src/locators/locators.ts
@@ -124,6 +124,7 @@ export interface Locators {
 		actionContainer: By;
 		actionItem: By;
 		attribute: string;
+		dropdown: string;
 	};
 	Editor: {
 		constructor: By;

diff --git a/tests/test-project/package.json b/tests/test-project/package.json
@@ -168,10 +168,15 @@
       }
     },
     "menus": {
-      "editor/title": [
+      "editor/title/run": [
         {
           "command": "extension.helloWorld",
-          "group": "navigation",
+          "group": "1_run@1",
+          "when": "resourceFilename =~ /Untitled.*$/"
+        },
+        {
+          "command": "extension.helloWorld2",
+          "group": "1_run@2",
           "when": "resourceFilename =~ /Untitled.*$/"
         }
       ],

diff --git a/tests/test-project/src/test/editor/editorView.test.ts b/tests/test-project/src/test/editor/editorView.test.ts
@@ -29,6 +29,8 @@ import {
 	DiffEditor,
 	InputBox,
 	VSBrowser,
+	EditorActionDropdown,
+	NotificationType,
 } from 'vscode-extension-tester';
 
 describe('EditorView', function () {
@@ -134,6 +136,27 @@ describe('EditorView', function () {
 		expect(editorAction).not.undefined;
 	});
 
+	(process.platform === 'darwin' ? it.skip : it)('Editor getAction dropdown', async function () {
+		this.timeout(15_000);
+		await new EditorView().openEditor('Untitled-2');
+		const editorAction = (await view.getAction('Run or Debug...')) as EditorActionDropdown;
+
+		if (editorAction) {
+			const menu = await editorAction.open();
+			await menu.select('Hello a World');
+
+			const center = await new Workbench().openNotificationsCenter();
+			const notifications = await center.getNotifications(NotificationType.Any);
+
+			expect(await notifications.at(0)?.getText()).is.equal('Hello World, Test Project!');
+
+			await center.clearAllNotifications();
+			await center.close();
+		} else {
+			expect.fail('Cannot find any dropdown editor action!');
+		}
+	});
+
 	describe('Editor Tab', function () {
 		let tab2: EditorTab;
 		let tab3: EditorTab;