test(aria/tree): generate additional methods and tests for Tree harness

Author: adolgachev

src/aria/tree/testing/item-harness.ts

@@ -76,6 +76,26 @@ export class TreeItemHarness extends ContentContainerComponentHarness<string> {
     return (await this.host()).click();
   }
 
+  /** Focuses the tree item. */
+  async focus(): Promise<void> {
+    return (await this.host()).focus();
+  }
+
+  /** Blurs the tree item. */
+  async blur(): Promise<void> {
+    return (await this.host()).blur();
+  }
+
+  /** Whether the tree item is active. */
+  async isActive(): Promise<boolean> {
+    return (await this._getHostAttribute('data-active')) === 'true';
+  }
+
+  /** Whether the tree item has focus. */
+  async isFocused(): Promise<boolean> {
+    return (await this.host()).isFocused();
+  }
+
   private async _getHostAttribute(attributeName: string): Promise<string | null> {
     return (await this.host()).getAttribute(attributeName);
   }

src/aria/tree/testing/tree-harness.spec.ts

@@ -109,6 +109,26 @@ describe('TreeHarness', () => {
       ],
     });
   });
+
+  it('should be able to get whether a tree item has focus', async () => {
+    const tree = await loader.getHarness(TreeHarness);
+    const item = (await tree.getItems({text: 'angular.json'}))[0];
+
+    expect(await item.isFocused()).toBeFalse();
+
+    await item.focus();
+    expect(await item.isFocused()).toBeTrue();
+
+    await item.blur();
+    expect(await item.isFocused()).toBeFalse();
+  });
+
+  it('should be able to get whether a tree item is active', async () => {
+    const tree = await loader.getHarness(TreeHarness);
+    const item = (await tree.getItems({text: 'public'}))[0];
+
+    expect(await item.isActive()).toBeTrue();
+  });
 });
 
 interface TreeNode {