Reuse the focus editor logic in the page object

ni · Oct 4, 2024 · aed3beb · aed3beb
1 parent c691f00
commit aed3beb
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/packages/nimble-components/src/rich-text/editor/testing/rich-text-editor.pageobject.ts b/packages/nimble-components/src/rich-text/editor/testing/rich-text-editor.pageobject.ts
@@ -248,10 +248,7 @@ export class RichTextEditorPageObject {
             .run();
         await waitForUpdatesAsync();
 
-        if (this.isMentionListboxOpened()) {
-            this.richTextEditorElement.tiptapEditor.commands.focus();
-            await waitForUpdatesAsync();
-        }
+        await this.focusEditorIfMentionListboxOpened();
     }
 
     public async setCursorPosition(position: number): Promise<void> {
@@ -439,10 +436,7 @@ export class RichTextEditorPageObject {
     }
 
     public async clickMentionListboxOption(index: number): Promise<void> {
-        if (this.isMentionListboxOpened()) {
-            this.richTextEditorElement.tiptapEditor.commands.focus();
-            await waitForUpdatesAsync();
-        }
+        await this.focusEditorIfMentionListboxOpened();
 
         const listOption = this.getAllListItemsInMentionBox()[index];
         listOption?.click();
@@ -535,4 +529,11 @@ export class RichTextEditorPageObject {
         );
         return parserMentionConfig;
     }
+
+    private async focusEditorIfMentionListboxOpened(): Promise<void> {
+        if (this.isMentionListboxOpened()) {
+            this.richTextEditorElement.tiptapEditor.commands.focus();
+            await waitForUpdatesAsync();
+        }
+    }
 }