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();
 		}
 	}