fix(MailUITests): Ignore failed undo

Infomaniak · Mar 26, 2024 · 29962c3 · 29962c3
1 parent 9ef6203
commit 29962c3
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/MailUITests/MailUITests.swift b/MailUITests/MailUITests.swift
@@ -109,7 +109,7 @@ class MailUITests: XCTestCase {
 
         app.collectionViews.buttons[MailResourcesStrings.Localizable.actionDelete].tap()
 
-        undo()
+        undo(ignoreUndoFailure: false)
     }
 
     func testMoveAction() {
@@ -168,9 +168,14 @@ class MailUITests: XCTestCase {
         }
     }
 
-    func undo() {
+    func undo(ignoreUndoFailure: Bool = true) {
         let cancelButton = app.buttons[MailResourcesStrings.Localizable.buttonCancel]
         _ = cancelButton.waitForExistence(timeout: 10)
+
+        if !cancelButton.exists && ignoreUndoFailure {
+            return
+        }
+
         cancelButton.tap()
     }