From 3690e9030487700903d52a40f77e04de1ca23567 Mon Sep 17 00:00:00 2001
From: Benoit Marty <benoit@matrix.org>
Date: Wed, 22 Oct 2025 10:31:12 +0200
Subject: [PATCH] Fix tests.

---
 libraries/mediaviewer/impl/build.gradle.kts                 | 1 +
 .../mediaviewer/impl/DefaultMediaViewerEntryPointTest.kt    | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/libraries/mediaviewer/impl/build.gradle.kts b/libraries/mediaviewer/impl/build.gradle.kts
index 4c373da5eb..ea435433f4 100644
--- a/libraries/mediaviewer/impl/build.gradle.kts
+++ b/libraries/mediaviewer/impl/build.gradle.kts
@@ -55,6 +55,7 @@ dependencies {
     implementation(projects.libraries.matrix.api)
 
     testCommonDependencies(libs, true)
+    testImplementation(projects.features.enterprise.test)
     testImplementation(projects.libraries.audio.test)
     testImplementation(projects.libraries.dateformatter.test)
     testImplementation(projects.libraries.featureflag.test)
diff --git a/libraries/mediaviewer/impl/src/test/kotlin/io/element/android/libraries/mediaviewer/impl/DefaultMediaViewerEntryPointTest.kt b/libraries/mediaviewer/impl/src/test/kotlin/io/element/android/libraries/mediaviewer/impl/DefaultMediaViewerEntryPointTest.kt
index 3af2e8cf69..ddc04fc6ce 100644
--- a/libraries/mediaviewer/impl/src/test/kotlin/io/element/android/libraries/mediaviewer/impl/DefaultMediaViewerEntryPointTest.kt
+++ b/libraries/mediaviewer/impl/src/test/kotlin/io/element/android/libraries/mediaviewer/impl/DefaultMediaViewerEntryPointTest.kt
@@ -11,10 +11,12 @@ import android.net.Uri
 import androidx.arch.core.executor.testing.InstantTaskExecutorRule
 import com.bumble.appyx.core.modality.BuildContext
 import com.google.common.truth.Truth.assertThat
+import io.element.android.features.enterprise.test.FakeEnterpriseService
 import io.element.android.libraries.core.mimetype.MimeTypes
 import io.element.android.libraries.matrix.api.core.EventId
 import io.element.android.libraries.matrix.api.core.UserId
 import io.element.android.libraries.matrix.api.media.MediaSource
+import io.element.android.libraries.matrix.test.A_SESSION_ID
 import io.element.android.libraries.matrix.test.media.FakeMatrixMediaLoader
 import io.element.android.libraries.mediaplayer.test.FakeAudioFocus
 import io.element.android.libraries.mediaviewer.api.MediaInfo
@@ -63,6 +65,8 @@ class DefaultMediaViewerEntryPointTest {
                 pagerKeysHandler = PagerKeysHandler(),
                 textFileViewer = { _, _ -> lambdaError() },
                 audioFocus = FakeAudioFocus(),
+                sessionId = A_SESSION_ID,
+                enterpriseService = FakeEnterpriseService(),
             )
         }
         val callback = object : MediaViewerEntryPoint.Callback {
@@ -104,6 +108,8 @@ class DefaultMediaViewerEntryPointTest {
                 pagerKeysHandler = PagerKeysHandler(),
                 textFileViewer = { _, _ -> lambdaError() },
                 audioFocus = FakeAudioFocus(),
+                sessionId = A_SESSION_ID,
+                enterpriseService = FakeEnterpriseService(),
             )
         }
         val callback = object : MediaViewerEntryPoint.Callback {