Fix tests

1 files changed, 3 insertions, 8 deletions

diff --git a/android/test/common/src/main/kotlin/net/mullvad/mullvadvpn/test/common/interactor/AppInteractor.kt b/android/test/common/src/main/kotlin/net/mullvad/mullvadvpn/test/common/interactor/AppInteractor.kt
index 340dad5ea0..1434de600e 100644
--- a/android/test/common/src/main/kotlin/net/mullvad/mullvadvpn/test/common/interactor/AppInteractor.kt
+++ b/android/test/common/src/main/kotlin/net/mullvad/mullvadvpn/test/common/interactor/AppInteractor.kt
@@ -71,7 +71,7 @@ class AppInteractor(
     }
 
     fun ensureLoggedIn() {
-        device.findObjectWithTimeout(By.text("UNSECURED CONNECTION"), VERY_LONG_TIMEOUT)
+        device.findObjectWithTimeout(By.text("DISCONNECTED"), VERY_LONG_TIMEOUT)
     }
 
     fun ensureOutOfTime() {
@@ -82,8 +82,8 @@ class AppInteractor(
         device.findObjectWithTimeout(By.text("Account"))
     }
 
-    fun extractIpAddress(): String {
-        device.findObjectWithTimeout(By.res("location_info_test_tag")).click()
+    fun extractOutIpv4Address(): String {
+        device.findObjectWithTimeout(By.res("connect_card_header_test_tag")).click()
         return device
             .findObjectWithTimeout(
                 // Text exist and contains IP address
@@ -91,7 +91,6 @@ class AppInteractor(
                 VERY_LONG_TIMEOUT,
             )
             .text
-            .extractIpAddress()
     }
 
     fun clickSettingsCog() {
@@ -118,8 +117,4 @@ class AppInteractor(
         device.findObjectWithTimeout(By.desc("Remove")).click()
         clickActionButtonByText("Yes, log out device")
     }
-
-    private fun String.extractIpAddress(): String {
-        return split(" ")[1].split(" ")[0]
-    }
 }