From cb52f8015f34610fbf9caff4298b718a382fa2e2 Mon Sep 17 00:00:00 2001
From: Niklas Berglund <niklas.berglund@gmail.com>
Date: Wed, 15 May 2024 16:28:18 +0200
Subject: [PATCH] Reset location selection

---
 ios/MullvadVPNUITests/ConnectivityTests.swift                | 4 ++++
 ios/MullvadVPNUITests/Test base classes/BaseUITestCase.swift | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/ios/MullvadVPNUITests/ConnectivityTests.swift b/ios/MullvadVPNUITests/ConnectivityTests.swift
index 07b2eb4a64d5..0ef72cb0fed4 100644
--- a/ios/MullvadVPNUITests/ConnectivityTests.swift
+++ b/ios/MullvadVPNUITests/ConnectivityTests.swift
@@ -68,6 +68,10 @@ class ConnectivityTests: LoggedOutUITestCase {
                 filterCloseButton.tap()
             }
 
+            // Reset selected location to Sweden
+            SelectLocationPage(self.app)
+                .tapLocationCell(withName: "Sweden")
+
             self.returnAccountWithTime(accountNumber: hasTimeAccountNumber)
         }
 
diff --git a/ios/MullvadVPNUITests/Test base classes/BaseUITestCase.swift b/ios/MullvadVPNUITests/Test base classes/BaseUITestCase.swift
index 40fe2eb37d15..3d0d58f52b19 100644
--- a/ios/MullvadVPNUITests/Test base classes/BaseUITestCase.swift	
+++ b/ios/MullvadVPNUITests/Test base classes/BaseUITestCase.swift	
@@ -14,7 +14,7 @@ class BaseUITestCase: XCTestCase {
     static let defaultTimeout = 5.0
     static let longTimeout = 15.0
     static let veryLongTimeout = 20.0
-    static let extremelyLongTimeout = 180.0
+    static let extremelyLongTimeout = 120.0
     static let shortTimeout = 1.0
 
     // swiftlint:disable force_cast