From c1e908a8e64e2699bc70a05265bdfd3122cfa876 Mon Sep 17 00:00:00 2001
From: Benoit Marty <benoit@matrix.org>
Date: Wed, 29 Apr 2026 09:19:41 +0200
Subject: [PATCH] Fix tests.

---
 libraries/matrix/impl/build.gradle.kts                      | 1 +
 .../matrix/impl/auth/RustMatrixAuthenticationServiceTest.kt | 4 ++++
 .../wellknown/impl/DefaultSessionWellknownRetrieverTest.kt  | 6 +++++-
 3 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/libraries/matrix/impl/build.gradle.kts b/libraries/matrix/impl/build.gradle.kts
index 7dcc835781..32bbeb082a 100644
--- a/libraries/matrix/impl/build.gradle.kts
+++ b/libraries/matrix/impl/build.gradle.kts
@@ -48,6 +48,7 @@ dependencies {
     implementation(libs.kotlinx.collections.immutable)
 
     testCommonDependencies(libs)
+    testImplementation(projects.features.enterprise.test)
     testImplementation(projects.libraries.featureflag.test)
     testImplementation(projects.libraries.matrix.test)
     testImplementation(projects.libraries.preferences.test)
diff --git a/libraries/matrix/impl/src/test/kotlin/io/element/android/libraries/matrix/impl/auth/RustMatrixAuthenticationServiceTest.kt b/libraries/matrix/impl/src/test/kotlin/io/element/android/libraries/matrix/impl/auth/RustMatrixAuthenticationServiceTest.kt
index bff2711bc5..7f422acfcf 100644
--- a/libraries/matrix/impl/src/test/kotlin/io/element/android/libraries/matrix/impl/auth/RustMatrixAuthenticationServiceTest.kt
+++ b/libraries/matrix/impl/src/test/kotlin/io/element/android/libraries/matrix/impl/auth/RustMatrixAuthenticationServiceTest.kt
@@ -9,6 +9,8 @@
 package io.element.android.libraries.matrix.impl.auth
 
 import com.google.common.truth.Truth.assertThat
+import io.element.android.features.enterprise.api.EnterpriseService
+import io.element.android.features.enterprise.test.FakeEnterpriseService
 import io.element.android.libraries.matrix.impl.ClientBuilderProvider
 import io.element.android.libraries.matrix.impl.FakeClientBuilderProvider
 import io.element.android.libraries.matrix.impl.createRustMatrixClientFactory
@@ -50,6 +52,7 @@ class RustMatrixAuthenticationServiceTest {
     private fun TestScope.createRustMatrixAuthenticationService(
         sessionStore: SessionStore = InMemorySessionStore(),
         clientBuilderProvider: ClientBuilderProvider = FakeClientBuilderProvider(),
+        enterpriseService: EnterpriseService = FakeEnterpriseService(),
     ): RustMatrixAuthenticationService {
         val baseDirectory = File("/base")
         val cacheDirectory = File("/cache")
@@ -68,6 +71,7 @@ class RustMatrixAuthenticationServiceTest {
                 buildMeta = aBuildMeta(),
                 oAuthRedirectUrlProvider = FakeOAuthRedirectUrlProvider(),
             ),
+            enterpriseService = enterpriseService,
         )
     }
 }
diff --git a/libraries/wellknown/impl/src/test/kotlin/io/element/android/libraries/wellknown/impl/DefaultSessionWellknownRetrieverTest.kt b/libraries/wellknown/impl/src/test/kotlin/io/element/android/libraries/wellknown/impl/DefaultSessionWellknownRetrieverTest.kt
index faad139a36..8a230c0317 100644
--- a/libraries/wellknown/impl/src/test/kotlin/io/element/android/libraries/wellknown/impl/DefaultSessionWellknownRetrieverTest.kt
+++ b/libraries/wellknown/impl/src/test/kotlin/io/element/android/libraries/wellknown/impl/DefaultSessionWellknownRetrieverTest.kt
@@ -36,6 +36,7 @@ class DefaultSessionWellknownRetrieverTest {
                     rageshakeUrl = null,
                     brandColor = null,
                     notificationSound = null,
+                    identityProviderAppScheme = null,
                 )
             )
         )
@@ -53,7 +54,8 @@ class DefaultSessionWellknownRetrieverTest {
                     "enforce_element_pro": true,
                     "rageshake_url": "a_rageshake_url",
                     "brand_color": "#FF0000",
-                    "notification_sound": "a_notification_sound.flac"
+                    "notification_sound": "a_notification_sound.flac",
+                    "idp_app_scheme": "an_app_scheme"
                 }""".trimIndent().toByteArray()
                 )
             }
@@ -66,6 +68,7 @@ class DefaultSessionWellknownRetrieverTest {
                     rageshakeUrl = "a_rageshake_url",
                     brandColor = "#FF0000",
                     notificationSound = "a_notification_sound.flac",
+                    identityProviderAppScheme = "an_app_scheme",
                 )
             )
         )
@@ -93,6 +96,7 @@ class DefaultSessionWellknownRetrieverTest {
                     rageshakeUrl = "a_rageshake_url",
                     brandColor = null,
                     notificationSound = null,
+                    identityProviderAppScheme = null,
                 )
             )
         )