From ba52fcb2079a515ce5535b5a5740c57c8412203c Mon Sep 17 00:00:00 2001
From: Jon Petersson <jon.petersson@mullvad.net>
Date: Wed, 22 Jan 2025 15:31:44 +0100
Subject: [PATCH] Fix testWireGuardOverTCPAutomatically and
 testConnectionRetryLogic

---
 ios/MullvadVPNUITests/RelayTests.swift | 2 ++
 ios/MullvadVPNUITests/tests.json       | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/ios/MullvadVPNUITests/RelayTests.swift b/ios/MullvadVPNUITests/RelayTests.swift
index f92feee29ec7..e38881429a3f 100644
--- a/ios/MullvadVPNUITests/RelayTests.swift
+++ b/ios/MullvadVPNUITests/RelayTests.swift
@@ -126,6 +126,7 @@ class RelayTests: LoggedInWithTimeUITestCase {
 
         // Should be two UDP connection attempts but sometimes only one is shown in the UI
         TunnelControlPage(app)
+            .tapRelayStatusExpandCollapseButton()
             .verifyConnectionAttemptsOrder()
             .tapCancelButton()
     }
@@ -252,6 +253,7 @@ class RelayTests: LoggedInWithTimeUITestCase {
 
         // Should be two UDP connection attempts but sometimes only one is shown in the UI
         TunnelControlPage(app)
+            .tapRelayStatusExpandCollapseButton()
             .verifyConnectingOverTCPAfterUDPAttempts()
             .waitForConnectedLabel()
             .tapDisconnectButton()
diff --git a/ios/MullvadVPNUITests/tests.json b/ios/MullvadVPNUITests/tests.json
index 19f5ec287a82..3841f81c2282 100644
--- a/ios/MullvadVPNUITests/tests.json
+++ b/ios/MullvadVPNUITests/tests.json
@@ -9,7 +9,8 @@
         ],
         "pr-merge-to-main": [
             "AccountTests/testLogin",
-            "AccountTests/testCreateAccount"
+            "AccountTests/testCreateAccount",
+            "RelayTests"
         ],
         "api-tests": [
             "AccountTests"