test: Increase robustness of waitForKeyboardToBeHidden utility

This utility appeared to fail occasionally due to an empty `addButton` value.
WordPress · Oct 12, 2023 · 6d30ce6 · 6d30ce6
1 parent 1b83f12
commit 6d30ce6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/packages/react-native-editor/__device-tests__/pages/editor-page.js b/packages/react-native-editor/__device-tests__/pages/editor-page.js
@@ -301,7 +301,10 @@ class EditorPage {
  async waitForKeyboardToBeHidden() {
  const { addButtonLocation } = this.initialValues;
  const addButton = await this.getAddBlockButton();
- const location = await addButton.getLocation();
+ let location;
+ if ( addButton ) {
+ location = await addButton.getLocation();
+ }
  let YLocation = addButtonLocation?.y;
  const currentOrientation = await this.driver.getOrientation();
  const isLandscape = currentOrientation === 'LANDSCAPE';
@@ -315,7 +318,7 @@ class EditorPage {
  );
  }
 
- if ( location.y < YLocation ) {
+ if ( ! addButton || location?.y < YLocation ) {
  await this.waitForKeyboardToBeHidden();
  }
  }