Add tryAgainOnFail to flakey action

getodk · Mar 29, 2024 · 487bf7c · 487bf7c
1 parent 6765b02
commit 487bf7c
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/collect_app/src/androidTest/java/org/odk/collect/android/support/pages/FormEndPage.java b/collect_app/src/androidTest/java/org/odk/collect/android/support/pages/FormEndPage.java
@@ -25,13 +25,16 @@ public FormEndPage assertOnPage() {
     }
 
     public <D extends Page<D>> D clickSaveAsDraft(D destination) {
-        clickOnString(org.odk.collect.strings.R.string.save_as_draft);
-        return destination.assertOnPage();
+        tryAgainOnFail(() -> {
+            clickOnString(org.odk.collect.strings.R.string.save_as_draft);
+            destination.assertOnPage();
+        });
+
+        return destination;
     }
 
     public MainMenuPage clickSaveAsDraft() {
-        clickOnString(org.odk.collect.strings.R.string.save_as_draft);
-        return new MainMenuPage().assertOnPage();
+        return clickSaveAsDraft(new MainMenuPage());
     }
 
     public <D extends Page<D>> D clickFinalize(D destination) {