2 files changed, 17 insertions, 8 deletions
diff --git a/android/app/src/androidTest/kotlin/net/mullvad/mullvadvpn/compose/screen/SelectLocationScreenTest.kt b/android/app/src/androidTest/kotlin/net/mullvad/mullvadvpn/compose/screen/SelectLocationScreenTest.kt
index 411d540f65..bfbee80f9e 100644
--- a/android/app/src/androidTest/kotlin/net/mullvad/mullvadvpn/compose/screen/SelectLocationScreenTest.kt
+++ b/android/app/src/androidTest/kotlin/net/mullvad/mullvadvpn/compose/screen/SelectLocationScreenTest.kt
@@ -156,15 +156,19 @@ class SelectLocationScreenTest {
     companion object {
         private val DUMMY_RELAY_1 =
             net.mullvad.mullvadvpn.model.Relay(
-                "Relay host 1",
-                true,
-                RelayEndpointData.Wireguard(WireguardRelayEndpointData)
+                hostname = "Relay host 1",
+                active = true,
+                endpointData = RelayEndpointData.Wireguard(WireguardRelayEndpointData),
+                owned = true,
+                provider = "PROVIDER"
             )
         private val DUMMY_RELAY_2 =
             net.mullvad.mullvadvpn.model.Relay(
-                "Relay host 2",
-                true,
-                RelayEndpointData.Wireguard(WireguardRelayEndpointData)
+                hostname = "Relay host 2",
+                active = true,
+                endpointData = RelayEndpointData.Wireguard(WireguardRelayEndpointData),
+                owned = true,
+                provider = "PROVIDER"
             )
         private val DUMMY_RELAY_CITY_1 =
             RelayListCity("Relay City 1", "RCi1", arrayListOf(DUMMY_RELAY_1))
diff --git a/android/lib/model/src/main/kotlin/net/mullvad/mullvadvpn/model/Relay.kt b/android/lib/model/src/main/kotlin/net/mullvad/mullvadvpn/model/Relay.kt
index b1abdc3c75..461648209c 100644
--- a/android/lib/model/src/main/kotlin/net/mullvad/mullvadvpn/model/Relay.kt
+++ b/android/lib/model/src/main/kotlin/net/mullvad/mullvadvpn/model/Relay.kt
@@ -4,8 +4,13 @@ import android.os.Parcelable
 import kotlinx.parcelize.Parcelize
 
 @Parcelize
-data class Relay(val hostname: String, val active: Boolean, val endpointData: RelayEndpointData) :
-    Parcelable {
+data class Relay(
+    val hostname: String,
+    val active: Boolean,
+    val owned: Boolean,
+    val provider: String,
+    val endpointData: RelayEndpointData
+) : Parcelable {
     val isWireguardRelay
         get() = endpointData is RelayEndpointData.Wireguard
 }