Fix the UITests trying to log in too fast

1 files changed, 7 insertions, 0 deletions

diff --git a/ios/MullvadVPNUITests/Base/BaseUITestCase.swift b/ios/MullvadVPNUITests/Base/BaseUITestCase.swift
index 71fbbf0b05..df03759409 100644
--- a/ios/MullvadVPNUITests/Base/BaseUITestCase.swift
+++ b/ios/MullvadVPNUITests/Base/BaseUITestCase.swift
@@ -320,6 +320,13 @@ class BaseUITestCase: XCTestCase {
                 .getSuccessIconShown()
 
             if successIconShown == false {
+                // If the login happened too fast, the UI harness will miss the success icon being shown
+                // Check if the app is already on main page, and continue if it is.
+                if app.otherElements[.headerBarView].exists {
+                    successIconShown = true
+                    break
+                }
+
                 // Give it some time to show up. App might be waiting for a network connection to timeout.
                 LoginPage(app).waitForAccountNumberSubmitButton()
             }