improve tree-select driver to allow selecting values across multiple …

…levels
superdesk · Sep 19, 2023 · 2038dec · 2038dec
1 parent 3281f89
commit 2038dec
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/e2e/client/specs/helpers/tree-select-driver.ts b/e2e/client/specs/helpers/tree-select-driver.ts
@@ -22,13 +22,22 @@ export class TreeSelectDriver {
         });
     }
 
-    addValue(value: string): void {
+    addValue(value: string | Array<string>): void {
         el(['open-popover'], null, this._element).click();
 
-        element(by.cssContainingText('[data-test-id="tree-select-popover"] [data-test-id="option"]', value)).click();
+        const values = typeof value === 'string' ? [value] : value;
+
+        for (let i = 0; i < values.length; i++) {
+            element(
+                by.cssContainingText(
+                    '[data-test-id="tree-select-popover"] [data-test-id="option"]',
+                    values[i],
+                ),
+            ).click();
+        }
     }
 
-    setValue(value: string) {
+    setValue(value: string | Array<string>) {
         const maybeClearButton = this._element.element(s(['clear-value']));
 
         maybeClearButton.isPresent().then((present) => {