From efee010b8c4189eecf327312d6ec51c305955ceb Mon Sep 17 00:00:00 2001
From: ganfra <francoisg@element.io>
Date: Thu, 8 Jun 2023 16:10:38 +0200
Subject: [PATCH 1/3] Tests: fix tests for media

---
 .../messages/impl/MessagesFlowNode.kt         |  9 ++--
 .../impl/media/helper/fileExtensionAndSize.kt | 13 +-----
 .../media/local/AndroidLocalMediaFactory.kt   |  8 +++-
 .../impl/media/local/LocalMediaView.kt        |  2 +-
 .../messages/impl/media/local/MediaInfo.kt    |  9 ++--
 .../TimelineItemContentMessageFactory.kt      | 13 ++++--
 .../model/event/TimelineItemFileContent.kt    |  3 +-
 .../event/TimelineItemFileContentProvider.kt  |  5 ++-
 .../model/event/TimelineItemImageContent.kt   |  1 +
 .../event/TimelineItemImageContentProvider.kt |  3 +-
 .../model/event/TimelineItemVideoContent.kt   |  1 +
 .../event/TimelineItemVideoContentProvider.kt |  5 ++-
 .../timeline/util/FileExtensionExtractor.kt   | 45 +++++++++++++++++++
 .../messages/MessagesPresenterTest.kt         | 10 +++--
 .../messages/fixtures/timelineItemsFactory.kt |  3 +-
 .../messages/media/FakeLocalMediaFactory.kt   | 13 ++++--
 .../media/viewer/MediaViewerPresenterTest.kt  |  3 +-
 17 files changed, 107 insertions(+), 39 deletions(-)
 create mode 100644 features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/util/FileExtensionExtractor.kt

diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/MessagesFlowNode.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/MessagesFlowNode.kt
index b5d3c40cf8..6fe5a6beed 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/MessagesFlowNode.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/MessagesFlowNode.kt
@@ -120,7 +120,8 @@ class MessagesFlowNode @AssistedInject constructor(
                     mediaInfo = MediaInfo(
                         name = event.content.body,
                         mimeType = event.content.mimeType,
-                        formattedFileSize = event.content.formattedFileSize
+                        formattedFileSize = event.content.formattedFileSize,
+                        fileExtension = event.content.fileExtension
                     ),
                     mediaSource = event.content.mediaSource,
                     thumbnailSource = event.content.mediaSource,
@@ -133,7 +134,8 @@ class MessagesFlowNode @AssistedInject constructor(
                     mediaInfo = MediaInfo(
                         name = event.content.body,
                         mimeType = event.content.mimeType,
-                        formattedFileSize = event.content.formattedFileSize
+                        formattedFileSize = event.content.formattedFileSize,
+                        fileExtension = event.content.fileExtension
                     ),
                     mediaSource = mediaSource,
                     thumbnailSource = event.content.thumbnailSource,
@@ -146,7 +148,8 @@ class MessagesFlowNode @AssistedInject constructor(
                     mediaInfo = MediaInfo(
                         name = event.content.body,
                         mimeType = event.content.mimeType,
-                        formattedFileSize = event.content.formattedFileSize
+                        formattedFileSize = event.content.formattedFileSize,
+                        fileExtension = event.content.fileExtension
                     ),
                     mediaSource = mediaSource,
                     thumbnailSource = event.content.thumbnailSource,
diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/helper/fileExtensionAndSize.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/helper/fileExtensionAndSize.kt
index fcf64eb24f..251d7a0f06 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/helper/fileExtensionAndSize.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/helper/fileExtensionAndSize.kt
@@ -16,18 +16,9 @@
 
 package io.element.android.features.messages.impl.media.helper
 
-import android.webkit.MimeTypeMap
-
-fun formatFileExtensionAndSize(name: String, size: String?): String {
-    val fileExtension = name.substringAfterLast('.', "")
-    // Makes sure the extension is known by the system, otherwise default to binary extension.
-    val safeExtension = if (MimeTypeMap.getSingleton().hasExtension(fileExtension)) {
-        fileExtension.uppercase()
-    } else {
-        "BIN"
-    }
+fun formatFileExtensionAndSize(extension: String, size: String?): String {
     return buildString {
-        append(safeExtension)
+        append(extension.uppercase())
         if (size != null) {
             append(' ')
             append("($size)")
diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/local/AndroidLocalMediaFactory.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/local/AndroidLocalMediaFactory.kt
index 22d03831b5..7500f0d91e 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/local/AndroidLocalMediaFactory.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/local/AndroidLocalMediaFactory.kt
@@ -20,6 +20,7 @@ import android.content.Context
 import android.net.Uri
 import androidx.core.net.toUri
 import com.squareup.anvil.annotations.ContributesBinding
+import io.element.android.features.messages.impl.timeline.util.FileExtensionExtractor
 import io.element.android.features.messages.impl.timeline.util.FileSizeFormatter
 import io.element.android.libraries.androidutils.file.getFileName
 import io.element.android.libraries.androidutils.file.getFileSize
@@ -35,6 +36,7 @@ import javax.inject.Inject
 class AndroidLocalMediaFactory @Inject constructor(
     @ApplicationContext private val context: Context,
     private val fileSizeFormatter: FileSizeFormatter,
+    private val fileExtensionExtractor: FileExtensionExtractor,
 ) : LocalMediaFactory {
 
     override fun createFromMediaFile(mediaFile: MediaFile, mediaInfo: MediaInfo): LocalMedia {
@@ -43,7 +45,7 @@ class AndroidLocalMediaFactory @Inject constructor(
             uri = uri,
             mimeType = mediaInfo.mimeType,
             name = mediaInfo.name,
-            formattedFileSize = mediaInfo.formattedFileSize
+            formattedFileSize = mediaInfo.formattedFileSize,
         )
     }
 
@@ -56,12 +58,14 @@ class AndroidLocalMediaFactory @Inject constructor(
         val resolvedMimeType = mimeType ?: context.getMimeType(uri) ?: MimeTypes.OctetStream
         val fileName = name ?: context.getFileName(uri) ?: ""
         val fileSize = formattedFileSize ?: fileSizeFormatter.format(context.getFileSize(uri))
+        val fileExtension = fileExtensionExtractor.extractFromName(fileName)
         return LocalMedia(
             uri = uri,
             info = MediaInfo(
                 mimeType = resolvedMimeType,
                 name = fileName,
-                formattedFileSize = fileSize
+                formattedFileSize = fileSize,
+                fileExtension = fileExtension
             )
         )
     }
diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/local/LocalMediaView.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/local/LocalMediaView.kt
index 13f0568511..04022b9f34 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/local/LocalMediaView.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/local/LocalMediaView.kt
@@ -246,7 +246,7 @@ fun MediaFileView(
                 )
                 Spacer(modifier = Modifier.height(4.dp))
                 Text(
-                    text = formatFileExtensionAndSize(info.name, info.formattedFileSize),
+                    text = formatFileExtensionAndSize(info.fileExtension, info.formattedFileSize),
                     fontSize = 14.sp,
                     maxLines = 1,
                     overflow = TextOverflow.Ellipsis,
diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/local/MediaInfo.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/local/MediaInfo.kt
index 57cd788bb8..2fc47e0d2c 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/local/MediaInfo.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/media/local/MediaInfo.kt
@@ -25,20 +25,21 @@ data class MediaInfo(
     val name: String,
     val mimeType: String,
     val formattedFileSize: String,
+    val fileExtension: String,
 ) : Parcelable
 
 fun anImageInfo(): MediaInfo = MediaInfo(
-    "an image file.jpg", MimeTypes.Jpeg, "4MB"
+    "an image file.jpg", MimeTypes.Jpeg, "4MB","jpg"
 )
 
 fun aVideoInfo(): MediaInfo = MediaInfo(
-    "a video file.mp4", MimeTypes.Mp4, "14MB"
+    "a video file.mp4", MimeTypes.Mp4, "14MB", "mp4"
 )
 
 fun aPdfInfo(): MediaInfo = MediaInfo(
-    "a pdf file.pdf", MimeTypes.Pdf, "23MB"
+    "a pdf file.pdf", MimeTypes.Pdf, "23MB", "pdf"
 )
 
 fun aFileInfo(): MediaInfo = MediaInfo(
-    "an apk file.apk", MimeTypes.Apk, "50MB"
+    "an apk file.apk", MimeTypes.Apk, "50MB", "apk"
 )
diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/factories/event/TimelineItemContentMessageFactory.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/factories/event/TimelineItemContentMessageFactory.kt
index 8941878334..f1a29ebcf9 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/factories/event/TimelineItemContentMessageFactory.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/factories/event/TimelineItemContentMessageFactory.kt
@@ -24,6 +24,7 @@ import io.element.android.features.messages.impl.timeline.model.event.TimelineIt
 import io.element.android.features.messages.impl.timeline.model.event.TimelineItemTextContent
 import io.element.android.features.messages.impl.timeline.model.event.TimelineItemUnknownContent
 import io.element.android.features.messages.impl.timeline.model.event.TimelineItemVideoContent
+import io.element.android.features.messages.impl.timeline.util.FileExtensionExtractor
 import io.element.android.features.messages.impl.timeline.util.FileSizeFormatter
 import io.element.android.features.messages.impl.timeline.util.toHtmlDocument
 import io.element.android.libraries.core.mimetype.MimeTypes
@@ -37,7 +38,8 @@ import io.element.android.libraries.matrix.api.timeline.item.event.VideoMessageT
 import javax.inject.Inject
 
 class TimelineItemContentMessageFactory @Inject constructor(
-    private val fileSizeFormatter: FileSizeFormatter
+    private val fileSizeFormatter: FileSizeFormatter,
+    private val fileExtensionExtractor: FileExtensionExtractor,
 ) {
 
     fun create(content: MessageContent): TimelineItemEventContent {
@@ -57,7 +59,8 @@ class TimelineItemContentMessageFactory @Inject constructor(
                     width = messageType.info?.width?.toInt(),
                     height = messageType.info?.height?.toInt(),
                     aspectRatio = aspectRatio,
-                    formattedFileSize = fileSizeFormatter.format(messageType.info?.size ?: 0)
+                    formattedFileSize = fileSizeFormatter.format(messageType.info?.size ?: 0),
+                    fileExtension = fileExtensionExtractor.extractFromName(messageType.body)
                 )
             }
             is VideoMessageType -> {
@@ -72,7 +75,8 @@ class TimelineItemContentMessageFactory @Inject constructor(
                     duration = messageType.info?.duration ?: 0L,
                     blurHash = messageType.info?.blurhash,
                     aspectRatio = aspectRatio,
-                    formattedFileSize = fileSizeFormatter.format(messageType.info?.size ?: 0)
+                    formattedFileSize = fileSizeFormatter.format(messageType.info?.size ?: 0),
+                    fileExtension = fileExtensionExtractor.extractFromName(messageType.body)
                 )
             }
             is FileMessageType -> TimelineItemFileContent(
@@ -80,7 +84,8 @@ class TimelineItemContentMessageFactory @Inject constructor(
                 thumbnailSource = messageType.info?.thumbnailSource,
                 fileSource = messageType.source,
                 mimeType = messageType.info?.mimetype ?: MimeTypes.OctetStream,
-                formattedFileSize = fileSizeFormatter.format(messageType.info?.size ?: 0)
+                formattedFileSize = fileSizeFormatter.format(messageType.info?.size ?: 0),
+                fileExtension = fileExtensionExtractor.extractFromName(messageType.body)
             )
             is NoticeMessageType -> TimelineItemNoticeContent(
                 body = messageType.body,
diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemFileContent.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemFileContent.kt
index 197bae2dda..aa35f5a117 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemFileContent.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemFileContent.kt
@@ -24,9 +24,10 @@ data class TimelineItemFileContent(
     val fileSource: MediaSource,
     val thumbnailSource: MediaSource?,
     val formattedFileSize: String,
+    val fileExtension: String,
     val mimeType: String,
 ) : TimelineItemEventContent {
     override val type: String = "TimelineItemFileContent"
 
-    val fileExtensionAndSize = formatFileExtensionAndSize(body, formattedFileSize)
+    val fileExtensionAndSize = formatFileExtensionAndSize(fileExtension, formattedFileSize)
 }
diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemFileContentProvider.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemFileContentProvider.kt
index 08125ea777..2a4335a92b 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemFileContentProvider.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemFileContentProvider.kt
@@ -33,6 +33,7 @@ fun aTimelineItemFileContent(fileName: String) = TimelineItemFileContent(
     body = fileName,
     thumbnailSource = MediaSource(url = ""),
     fileSource = MediaSource(url = ""),
-    mimeType = MimeTypes.OctetStream,
-    formattedFileSize = "100kB"
+    mimeType = MimeTypes.Apk,
+    formattedFileSize = "100kB",
+    fileExtension = "apk"
 )
diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemImageContent.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemImageContent.kt
index 18be8d404d..a5ef890c82 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemImageContent.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemImageContent.kt
@@ -22,6 +22,7 @@ data class TimelineItemImageContent(
     val body: String,
     val mediaSource: MediaSource,
     val formattedFileSize: String,
+    val fileExtension: String,
     val mimeType: String,
     val blurhash: String?,
     val width: Int?,
diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemImageContentProvider.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemImageContentProvider.kt
index 66b35a07d3..519f1e58a4 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemImageContentProvider.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemImageContentProvider.kt
@@ -37,5 +37,6 @@ fun aTimelineItemImageContent() = TimelineItemImageContent(
     width = null,
     height = 300,
     aspectRatio = 0.5f,
-    formattedFileSize = "4MB"
+    formattedFileSize = "4MB",
+    fileExtension = "jpg"
 )
diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemVideoContent.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemVideoContent.kt
index 1432ebbda0..a76be8c13c 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemVideoContent.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemVideoContent.kt
@@ -29,6 +29,7 @@ data class TimelineItemVideoContent(
     val width: Int?,
     val mimeType: String,
     val formattedFileSize: String,
+    val fileExtension: String,
 ) : TimelineItemEventContent {
     override val type: String = "TimelineItemImageContent"
 }
diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemVideoContentProvider.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemVideoContentProvider.kt
index 937f35b349..5b4793d9db 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemVideoContentProvider.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemVideoContentProvider.kt
@@ -30,7 +30,7 @@ open class TimelineItemVideoContentProvider : PreviewParameterProvider<TimelineI
 }
 
 fun aTimelineItemVideoContent() = TimelineItemVideoContent(
-    body = "a video",
+    body = "Video.mp4",
     thumbnailSource = MediaSource(url = ""),
     blurHash = "TQF5:I_NtRE4kXt7Z#MwkCIARPjr",
     aspectRatio = 0.5f,
@@ -39,5 +39,6 @@ fun aTimelineItemVideoContent() = TimelineItemVideoContent(
     height = 300,
     width = 150,
     mimeType = MimeTypes.Mp4,
-    formattedFileSize = "14MB"
+    formattedFileSize = "14MB",
+    fileExtension = "mp4"
 )
diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/util/FileExtensionExtractor.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/util/FileExtensionExtractor.kt
new file mode 100644
index 0000000000..bdb4e0a23d
--- /dev/null
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/util/FileExtensionExtractor.kt
@@ -0,0 +1,45 @@
+/*
+ * Copyright (c) 2023 New Vector Ltd
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.element.android.features.messages.impl.timeline.util
+
+import android.webkit.MimeTypeMap
+import com.squareup.anvil.annotations.ContributesBinding
+import io.element.android.libraries.di.AppScope
+import javax.inject.Inject
+
+interface FileExtensionExtractor {
+    fun extractFromName(name: String): String
+}
+
+@ContributesBinding(AppScope::class)
+class FileExtensionExtractorWithValidation @Inject constructor() : FileExtensionExtractor {
+    override fun extractFromName(name: String): String {
+        val fileExtension = name.substringAfterLast('.', "")
+        // Makes sure the extension is known by the system, otherwise default to binary extension.
+        return if (MimeTypeMap.getSingleton().hasExtension(fileExtension)) {
+            fileExtension
+        } else {
+            "bin"
+        }
+    }
+}
+
+class FileExtensionExtractorWithoutValidation : FileExtensionExtractor {
+    override fun extractFromName(name: String): String {
+        return name.substringAfterLast('.', "")
+    }
+}
diff --git a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/MessagesPresenterTest.kt b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/MessagesPresenterTest.kt
index cb1c923397..a112b0f7f1 100644
--- a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/MessagesPresenterTest.kt
+++ b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/MessagesPresenterTest.kt
@@ -32,7 +32,6 @@ import io.element.android.features.messages.impl.timeline.TimelinePresenter
 import io.element.android.features.messages.impl.timeline.model.event.TimelineItemFileContent
 import io.element.android.features.messages.impl.timeline.model.event.TimelineItemImageContent
 import io.element.android.features.messages.impl.timeline.model.event.TimelineItemVideoContent
-import io.element.android.features.messages.impl.utils.messagesummary.MessageSummaryFormatterImpl
 import io.element.android.features.messages.media.FakeLocalMediaFactory
 import io.element.android.features.messages.utils.messagesummary.FakeMessageSummaryFormatter
 import io.element.android.features.networkmonitor.test.FakeNetworkMonitor
@@ -145,6 +144,8 @@ class MessagesPresenterTest {
                     width = 20,
                     height = 20,
                     aspectRatio = 1.0f,
+                    fileExtension = "jpg",
+                    formattedFileSize = "4MB"
                 )
             )
             initialState.eventSink.invoke(MessagesEvents.HandleAction(TimelineItemAction.Reply, mediaMessage))
@@ -175,6 +176,8 @@ class MessagesPresenterTest {
                     width = 20,
                     height = 20,
                     aspectRatio = 1.0f,
+                    fileExtension = "mp4",
+                    formattedFileSize = "50MB"
                 )
             )
             initialState.eventSink.invoke(MessagesEvents.HandleAction(TimelineItemAction.Reply, mediaMessage))
@@ -196,11 +199,12 @@ class MessagesPresenterTest {
             val initialState = awaitItem()
             val mediaMessage = aMessageEvent(
                 content = TimelineItemFileContent(
-                    body = "video.mp4",
+                    body = "video.pdf",
                     fileSource = MediaSource(AN_AVATAR_URL),
                     thumbnailSource = MediaSource(AN_AVATAR_URL),
                     formattedFileSize = "10 MB",
                     mimeType = MimeTypes.Pdf,
+                    fileExtension = "pdf",
                 )
             )
             initialState.eventSink.invoke(MessagesEvents.HandleAction(TimelineItemAction.Reply, mediaMessage))
@@ -276,7 +280,7 @@ class MessagesPresenterTest {
             mediaPickerProvider = FakePickerProvider(),
             featureFlagService = FakeFeatureFlagService(),
             localMediaFactory = FakeLocalMediaFactory(mockMediaUrl),
-            mediaSender = MediaSender(FakeMediaPreProcessor(),matrixRoom),
+            mediaSender = MediaSender(FakeMediaPreProcessor(), matrixRoom),
             snackbarDispatcher = SnackbarDispatcher(),
         )
         val timelinePresenter = TimelinePresenter(
diff --git a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/fixtures/timelineItemsFactory.kt b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/fixtures/timelineItemsFactory.kt
index c6ea04f407..a5b1780589 100644
--- a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/fixtures/timelineItemsFactory.kt
+++ b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/fixtures/timelineItemsFactory.kt
@@ -31,6 +31,7 @@ import io.element.android.features.messages.impl.timeline.factories.event.Timeli
 import io.element.android.features.messages.impl.timeline.factories.virtual.TimelineItemDaySeparatorFactory
 import io.element.android.features.messages.impl.timeline.factories.virtual.TimelineItemVirtualFactory
 import io.element.android.features.messages.impl.timeline.groups.TimelineItemGrouper
+import io.element.android.features.messages.impl.timeline.util.FileExtensionExtractorWithoutValidation
 import io.element.android.features.messages.timeline.FakeFileSizeFormatter
 import io.element.android.libraries.dateformatter.test.FakeDaySeparatorFormatter
 import io.element.android.libraries.eventformatter.api.TimelineEventFormatter
@@ -43,7 +44,7 @@ internal fun aTimelineItemsFactory(): TimelineItemsFactory {
         dispatchers = testCoroutineDispatchers(),
         eventItemFactory = TimelineItemEventFactory(
             TimelineItemContentFactory(
-                messageFactory = TimelineItemContentMessageFactory(FakeFileSizeFormatter()),
+                messageFactory = TimelineItemContentMessageFactory(FakeFileSizeFormatter(), FileExtensionExtractorWithoutValidation()),
                 redactedMessageFactory = TimelineItemContentRedactedFactory(),
                 stickerFactory = TimelineItemContentStickerFactory(),
                 utdFactory = TimelineItemContentUTDFactory(),
diff --git a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/media/FakeLocalMediaFactory.kt b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/media/FakeLocalMediaFactory.kt
index 976aa049f3..b5efdaf77d 100644
--- a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/media/FakeLocalMediaFactory.kt
+++ b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/media/FakeLocalMediaFactory.kt
@@ -21,10 +21,15 @@ import io.element.android.features.messages.fixtures.aLocalMedia
 import io.element.android.features.messages.impl.media.local.LocalMedia
 import io.element.android.features.messages.impl.media.local.LocalMediaFactory
 import io.element.android.features.messages.impl.media.local.MediaInfo
+import io.element.android.features.messages.impl.timeline.util.FileExtensionExtractor
+import io.element.android.features.messages.impl.timeline.util.FileExtensionExtractorWithoutValidation
 import io.element.android.libraries.core.mimetype.MimeTypes
 import io.element.android.libraries.matrix.api.media.MediaFile
 
-class FakeLocalMediaFactory(private val localMediaUri: Uri) : LocalMediaFactory {
+class FakeLocalMediaFactory(
+    private val localMediaUri: Uri,
+    private val fileExtensionExtractor: FileExtensionExtractor = FileExtensionExtractorWithoutValidation()
+) : LocalMediaFactory {
 
     var fallbackMimeType: String = MimeTypes.OctetStream
     var fallbackName: String = "File name"
@@ -35,10 +40,12 @@ class FakeLocalMediaFactory(private val localMediaUri: Uri) : LocalMediaFactory
     }
 
     override fun createFromUri(uri: Uri, mimeType: String?, name: String?, formattedFileSize: String?): LocalMedia {
+        val safeName = name ?: fallbackName
         val mediaInfo = MediaInfo(
-            name = name ?: fallbackName,
+            name = safeName,
             mimeType = mimeType ?: fallbackMimeType,
-            formattedFileSize = formattedFileSize ?: fallbackFileSize
+            formattedFileSize = formattedFileSize ?: fallbackFileSize,
+            fileExtension = fileExtensionExtractor.extractFromName(safeName)
         )
         return aLocalMedia(uri, mediaInfo)
     }
diff --git a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/media/viewer/MediaViewerPresenterTest.kt b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/media/viewer/MediaViewerPresenterTest.kt
index 86c3f3a7a4..e5d75d3b15 100644
--- a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/media/viewer/MediaViewerPresenterTest.kt
+++ b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/media/viewer/MediaViewerPresenterTest.kt
@@ -42,7 +42,8 @@ import org.junit.Test
 private val TESTED_MEDIA_INFO = MediaInfo(
     name = "",
     mimeType = "",
-    formattedFileSize = ""
+    formattedFileSize = "",
+    fileExtension = ""
 )
 
 class MediaViewerPresenterTest {

From 50c015801f4ca1a988444f7bb6122544f4fbe965 Mon Sep 17 00:00:00 2001
From: ganfra <francoisg@element.io>
Date: Thu, 8 Jun 2023 17:17:13 +0200
Subject: [PATCH 2/3] Clean tests values for file extension

---
 .../model/event/TimelineItemFileContentProvider.kt        | 4 ++--
 .../android/features/messages/MessagesPresenterTest.kt    | 2 +-
 .../messages/media/viewer/MediaViewerPresenterTest.kt     | 8 ++------
 3 files changed, 5 insertions(+), 9 deletions(-)

diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemFileContentProvider.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemFileContentProvider.kt
index 2a4335a92b..e037b9b41d 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemFileContentProvider.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/timeline/model/event/TimelineItemFileContentProvider.kt
@@ -33,7 +33,7 @@ fun aTimelineItemFileContent(fileName: String) = TimelineItemFileContent(
     body = fileName,
     thumbnailSource = MediaSource(url = ""),
     fileSource = MediaSource(url = ""),
-    mimeType = MimeTypes.Apk,
+    mimeType = MimeTypes.Pdf,
     formattedFileSize = "100kB",
-    fileExtension = "apk"
+    fileExtension = "pdf"
 )
diff --git a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/MessagesPresenterTest.kt b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/MessagesPresenterTest.kt
index a112b0f7f1..4c0c62f7d8 100644
--- a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/MessagesPresenterTest.kt
+++ b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/MessagesPresenterTest.kt
@@ -199,7 +199,7 @@ class MessagesPresenterTest {
             val initialState = awaitItem()
             val mediaMessage = aMessageEvent(
                 content = TimelineItemFileContent(
-                    body = "video.pdf",
+                    body = "file.pdf",
                     fileSource = MediaSource(AN_AVATAR_URL),
                     thumbnailSource = MediaSource(AN_AVATAR_URL),
                     formattedFileSize = "10 MB",
diff --git a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/media/viewer/MediaViewerPresenterTest.kt b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/media/viewer/MediaViewerPresenterTest.kt
index e5d75d3b15..ea40cfe791 100644
--- a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/media/viewer/MediaViewerPresenterTest.kt
+++ b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/media/viewer/MediaViewerPresenterTest.kt
@@ -24,6 +24,7 @@ import app.cash.molecule.moleculeFlow
 import app.cash.turbine.test
 import com.google.common.truth.Truth.assertThat
 import io.element.android.features.messages.impl.media.local.MediaInfo
+import io.element.android.features.messages.impl.media.local.aFileInfo
 import io.element.android.features.messages.impl.media.viewer.MediaViewerEvents
 import io.element.android.features.messages.impl.media.viewer.MediaViewerNode
 import io.element.android.features.messages.impl.media.viewer.MediaViewerPresenter
@@ -39,12 +40,7 @@ import kotlinx.coroutines.ExperimentalCoroutinesApi
 import kotlinx.coroutines.test.runTest
 import org.junit.Test
 
-private val TESTED_MEDIA_INFO = MediaInfo(
-    name = "",
-    mimeType = "",
-    formattedFileSize = "",
-    fileExtension = ""
-)
+private val TESTED_MEDIA_INFO = aFileInfo()
 
 class MediaViewerPresenterTest {
 

From d365f902e3315033a284bc65374ed7948b1ebe67 Mon Sep 17 00:00:00 2001
From: ElementBot <benoitm+elementbot@element.io>
Date: Thu, 8 Jun 2023 15:27:01 +0000
Subject: [PATCH 3/3] Update screenshots

---
 ...chmentsPreviewViewDarkPreview_0_null_1,NEXUS_5,1.0,en].png | 4 ++--
 ...up_MediaViewerViewDarkPreview_0_null_6,NEXUS_5,1.0,en].png | 4 ++--
 ...up_MediaViewerViewDarkPreview_0_null_7,NEXUS_5,1.0,en].png | 4 ++--
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/tests/uitests/src/test/snapshots/images/io.element.android.tests.uitests_ScreenshotTest_preview_tests[io.element.android.features.messages.impl.attachments.preview_null_DefaultGroup_AttachmentsPreviewViewDarkPreview_0_null_1,NEXUS_5,1.0,en].png b/tests/uitests/src/test/snapshots/images/io.element.android.tests.uitests_ScreenshotTest_preview_tests[io.element.android.features.messages.impl.attachments.preview_null_DefaultGroup_AttachmentsPreviewViewDarkPreview_0_null_1,NEXUS_5,1.0,en].png
index bc5044cd40..76a0c8dfa3 100644
--- a/tests/uitests/src/test/snapshots/images/io.element.android.tests.uitests_ScreenshotTest_preview_tests[io.element.android.features.messages.impl.attachments.preview_null_DefaultGroup_AttachmentsPreviewViewDarkPreview_0_null_1,NEXUS_5,1.0,en].png
+++ b/tests/uitests/src/test/snapshots/images/io.element.android.tests.uitests_ScreenshotTest_preview_tests[io.element.android.features.messages.impl.attachments.preview_null_DefaultGroup_AttachmentsPreviewViewDarkPreview_0_null_1,NEXUS_5,1.0,en].png
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ade6aa1e0fd7731173f2ae3424a931234d64e0aa8334a93c983ee46ff6dc5ce5
-size 17170
+oid sha256:50e833b56337c65f91bf62a2be960db30452f3333eb4219d235bfe6229f3fbfa
+size 17384
diff --git a/tests/uitests/src/test/snapshots/images/io.element.android.tests.uitests_ScreenshotTest_preview_tests[io.element.android.features.messages.impl.media.viewer_null_DefaultGroup_MediaViewerViewDarkPreview_0_null_6,NEXUS_5,1.0,en].png b/tests/uitests/src/test/snapshots/images/io.element.android.tests.uitests_ScreenshotTest_preview_tests[io.element.android.features.messages.impl.media.viewer_null_DefaultGroup_MediaViewerViewDarkPreview_0_null_6,NEXUS_5,1.0,en].png
index 289d08f536..c288ee5dd7 100644
--- a/tests/uitests/src/test/snapshots/images/io.element.android.tests.uitests_ScreenshotTest_preview_tests[io.element.android.features.messages.impl.media.viewer_null_DefaultGroup_MediaViewerViewDarkPreview_0_null_6,NEXUS_5,1.0,en].png
+++ b/tests/uitests/src/test/snapshots/images/io.element.android.tests.uitests_ScreenshotTest_preview_tests[io.element.android.features.messages.impl.media.viewer_null_DefaultGroup_MediaViewerViewDarkPreview_0_null_6,NEXUS_5,1.0,en].png
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6c8a175985e271948423d677b0d79e9ae930aa227c417d5c78e567abda40e378
-size 16210
+oid sha256:963ad9351110d18b22e1c34fb1b71be9f7a241176503d475ecb7ef7a492fa364
+size 16410
diff --git a/tests/uitests/src/test/snapshots/images/io.element.android.tests.uitests_ScreenshotTest_preview_tests[io.element.android.features.messages.impl.media.viewer_null_DefaultGroup_MediaViewerViewDarkPreview_0_null_7,NEXUS_5,1.0,en].png b/tests/uitests/src/test/snapshots/images/io.element.android.tests.uitests_ScreenshotTest_preview_tests[io.element.android.features.messages.impl.media.viewer_null_DefaultGroup_MediaViewerViewDarkPreview_0_null_7,NEXUS_5,1.0,en].png
index b807b8219f..ca4e198d87 100644
--- a/tests/uitests/src/test/snapshots/images/io.element.android.tests.uitests_ScreenshotTest_preview_tests[io.element.android.features.messages.impl.media.viewer_null_DefaultGroup_MediaViewerViewDarkPreview_0_null_7,NEXUS_5,1.0,en].png
+++ b/tests/uitests/src/test/snapshots/images/io.element.android.tests.uitests_ScreenshotTest_preview_tests[io.element.android.features.messages.impl.media.viewer_null_DefaultGroup_MediaViewerViewDarkPreview_0_null_7,NEXUS_5,1.0,en].png
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:367e2b224070dcc8ceb218b17336cd406b650b66f9c0fff50f8ec6093bb174dd
-size 16182
+oid sha256:d138660e19740b07263bb615198b65b4299d7b5053c19e36e854e814c9ecb551
+size 16381