feat(material/menu/testing): finish implementing harness

Author: mmalerba

src/cdk/testing/component-harness.ts

@@ -157,7 +157,7 @@ export interface LocatorFactory {
  * should be inherited when defining user's own harness.
  */
 export abstract class ComponentHarness {
-  constructor(private readonly locatorFactory: LocatorFactory) {}
+  constructor(protected readonly locatorFactory: LocatorFactory) {}
 
   /** Gets a `Promise` for the `TestElement` representing the host element of the component. */
   async host(): Promise<TestElement> {

src/material-experimental/mdc-menu/testing/menu-harness.ts

@@ -9,7 +9,7 @@
 import {ComponentHarness, HarnessPredicate} from '@angular/cdk/testing';
 import {coerceBooleanProperty} from '@angular/cdk/coercion';
 import {MatMenuItemHarness} from './menu-item-harness';
-import {MenuHarnessFilters} from '@angular/material/menu/testing';
+import {MenuHarnessFilters, MenuItemHarnessFilters} from '@angular/material/menu/testing';
 
 /**
  * Harness for interacting with a MDC-based mat-menu in tests.
@@ -29,7 +29,7 @@ export class MatMenuHarness extends ComponentHarness {
    */
   static with(options: MenuHarnessFilters = {}): HarnessPredicate<MatMenuHarness> {
     return new HarnessPredicate(MatMenuHarness, options)
-        .addOption('text', options.triggerText,
+        .addOption('triggerText', options.triggerText,
             (harness, text) => HarnessPredicate.stringMatches(harness.getTriggerText(), text));
   }
 
@@ -65,23 +65,7 @@ export class MatMenuHarness extends ComponentHarness {
     throw Error('not implemented');
   }
 
-  async getItems(): Promise<MatMenuItemHarness[]> {
-    throw Error('not implemented');
-  }
-
-  async getItemLabels(): Promise<string[]> {
-    throw Error('not implemented');
-  }
-
-  async getItemByLabel(): Promise<MatMenuItemHarness> {
-    throw Error('not implemented');
-  }
-
-  async getItemByIndex(): Promise<MatMenuItemHarness> {
-    throw Error('not implemented');
-  }
-
-  async getFocusedItem(): Promise<MatMenuItemHarness> {
+  async getItems(filters: MenuItemHarnessFilters = {}): Promise<MatMenuItemHarness[]> {
     throw Error('not implemented');
   }
 }

src/material/menu/testing/BUILD.bazel

@@ -26,6 +26,7 @@ ng_test_library(
     deps = [
         ":testing",
         "//src/cdk/overlay",
+        "//src/cdk/private/testing",
         "//src/cdk/testing",
         "//src/cdk/testing/testbed",
         "//src/material/menu",

src/material/menu/testing/menu-harness-filters.ts

@@ -14,4 +14,5 @@ export interface MenuHarnessFilters extends BaseHarnessFilters {
 
 export interface MenuItemHarnessFilters extends BaseHarnessFilters {
   text?: string | RegExp;
+  hasSubmenu?: boolean;
 }

src/material/menu/testing/menu-harness.ts

@@ -6,10 +6,9 @@
  * found in the LICENSE file at https://angular.io/license
  */
 
-import {ComponentHarness, HarnessPredicate} from '@angular/cdk/testing';
+import {ComponentHarness, HarnessPredicate, TestElement, TestKey} from '@angular/cdk/testing';
 import {coerceBooleanProperty} from '@angular/cdk/coercion';
-import {MenuHarnessFilters} from './menu-harness-filters';
-import {MatMenuItemHarness} from './menu-item-harness';
+import {MenuHarnessFilters, MenuItemHarnessFilters} from './menu-harness-filters';
 
 /**
  * Harness for interacting with a standard mat-menu in tests.
@@ -18,6 +17,8 @@ import {MatMenuItemHarness} from './menu-item-harness';
 export class MatMenuHarness extends ComponentHarness {
   static hostSelector = '.mat-menu-trigger';
 
+  private _documentRootLocator = this.documentRootLocatorFactory();
+
   // TODO: potentially extend MatButtonHarness
 
   /**
@@ -29,7 +30,7 @@ export class MatMenuHarness extends ComponentHarness {
    */
   static with(options: MenuHarnessFilters = {}): HarnessPredicate<MatMenuHarness> {
     return new HarnessPredicate(MatMenuHarness, options)
-        .addOption('text', options.triggerText,
+        .addOption('triggerText', options.triggerText,
             (harness, text) => HarnessPredicate.stringMatches(harness.getTriggerText(), text));
   }
 
@@ -39,8 +40,9 @@ export class MatMenuHarness extends ComponentHarness {
     return coerceBooleanProperty(await disabled);
   }
 
+  /** Whether the menu is open. */
   async isOpen(): Promise<boolean> {
-    throw Error('not implemented');
+    return !!(await this._getMenuPanel());
   }
 
   async getTriggerText(): Promise<string> {
@@ -58,30 +60,117 @@ export class MatMenuHarness extends ComponentHarness {
   }
 
   async open(): Promise<void> {
-    throw Error('not implemented');
+    if (!await this.isOpen()) {
+      return (await this.host()).click();
+    }
   }
 
   async close(): Promise<void> {
-    throw Error('not implemented');
+    const panel = await this._getMenuPanel();
+    if (panel) {
+      return panel.sendKeys(TestKey.ESCAPE);
+    }
+  }
+
+  async getItems(filters: Omit<MenuItemHarnessFilters, 'ancestor'> = {}):
+      Promise<MatMenuItemHarness[]> {
+    const panelId = await this._getPanelId();
+    if (panelId) {
+      return this._documentRootLocator.locatorForAll(
+          MatMenuItemHarness.with({...filters, ancestor: `#${panelId}`}))();
+    }
+    return [];
+  }
+
+  async selectItem(...itemFilters: MenuItemHarnessFilters[]):
+      Promise<void> {
+    itemFilters = itemFilters.length ? itemFilters : [{}];
+    await this.open();
+    const items = await this.getItems(itemFilters[0]);
+    if (!items.length) {
+      throw Error(`Could not find item matching ${JSON.stringify(itemFilters[0])}`);
+    }
+
+    if (itemFilters.length === 1) {
+      return await items[0].click();
+    }
+
+    const menu = await items[0].getSubmenu();
+    if (!menu) {
+      throw Error(`Item matching ${JSON.stringify(itemFilters[0])} does not have a submenu`);
+    }
+    return menu.selectItem(...itemFilters.slice(1));
+  }
+
+  private async _getMenuPanel(): Promise<TestElement | null> {
+    const panelId = await this._getPanelId();
+    return panelId ? this._documentRootLocator.locatorForOptional(`#${panelId}`)() : null;
+  }
+
+  private async _getPanelId(): Promise<string | null> {
+    const panelId = await (await this.host()).getAttribute('aria-controls');
+    return panelId || null;
+  }
+}
+
+
+/**
+ * Harness for interacting with a standard mat-menu in tests.
+ * @dynamic
+ */
+export class MatMenuItemHarness extends ComponentHarness {
+  static hostSelector = '.mat-menu-item';
+
+  /**
+   * Gets a `HarnessPredicate` that can be used to search for a menu with specific attributes.
+   * @param options Options for narrowing the search:
+   *   - `selector` finds a menu item whose host element matches the given selector.
+   *   - `label` finds a menu item with specific label text.
+   * @return a `HarnessPredicate` configured with the given options.
+   */
+  static with(options: MenuItemHarnessFilters = {}): HarnessPredicate<MatMenuItemHarness> {
+    return new HarnessPredicate(MatMenuItemHarness, options)
+        .addOption('text', options.text,
+            (harness, text) => HarnessPredicate.stringMatches(harness.getText(), text))
+        .addOption('hasSubmenu', options.hasSubmenu,
+            async (harness, hasSubmenu) => (await harness.hasSubmenu()) === hasSubmenu);
   }
 
-  async getItems(): Promise<MatMenuItemHarness[]> {
-    throw Error('not implemented');
+  /** Gets a boolean promise indicating if the menu is disabled. */
+  async isDisabled(): Promise<boolean> {
+    const disabled = (await this.host()).getAttribute('disabled');
+    return coerceBooleanProperty(await disabled);
   }
 
-  async getItemLabels(): Promise<string[]> {
-    throw Error('not implemented');
+  async getText(): Promise<string> {
+    return (await this.host()).text();
+  }
+
+  /** Focuses the menu and returns a void promise that indicates when the action is complete. */
+  async focus(): Promise<void> {
+    return (await this.host()).focus();
+  }
+
+  /** Blurs the menu and returns a void promise that indicates when the action is complete. */
+  async blur(): Promise<void> {
+    return (await this.host()).blur();
   }
 
-  async getItemByLabel(): Promise<MatMenuItemHarness> {
-    throw Error('not implemented');
+  /** Clicks the menu item. */
+  async click(): Promise<void> {
+    return (await this.host()).click();
   }
 
-  async getItemByIndex(): Promise<MatMenuItemHarness> {
-    throw Error('not implemented');
+  /** Whether this item has a submenu. */
+  async hasSubmenu(): Promise<boolean> {
+    return (await this.host()).matchesSelector(MatMenuHarness.hostSelector);
   }
 
-  async getFocusedItem(): Promise<MatMenuItemHarness> {
-    throw Error('not implemented');
+  /** Gets the submenu associated with this menu item, or null if none. */
+  async getSubmenu(): Promise<MatMenuHarness | null> {
+    if (await this.hasSubmenu()) {
+      return new MatMenuHarness(this.locatorFactory);
+    }
+    return null;
   }
 }

src/material/menu/testing/menu-item-harness.ts

@@ -8,4 +8,3 @@
 
 export * from './menu-harness';
 export * from './menu-harness-filters';
-export * from './menu-item-harness';