Compress png file and generate png thumbnail for png files.

Fix tests
This commit is contained in:
Benoit Marty 2024-10-31 15:45:36 +01:00
parent b43064fa42
commit 26d8b74255
7 changed files with 318 additions and 151 deletions

View file

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:77276f9b174f8823eaf787ab0a659199ef5d30c0361ec8b9b4f0890adb1907a1
size 9986336

View file

@ -35,113 +35,194 @@ import kotlin.time.Duration
@RunWith(RobolectricTestRunner::class)
class AndroidMediaPreProcessorTest {
@Test
fun `test processing image`() = runTest {
private suspend fun TestScope.process(
asset: Asset,
compressIfPossible: Boolean,
sdkIntVersion: Int = Build.VERSION_CODES.P,
deleteOriginal: Boolean = false,
): MediaUploadInfo {
val context = InstrumentationRegistry.getInstrumentation().context
val sut = createAndroidMediaPreProcessor(context)
val file = getFileFromAssets(context, "image.png")
val sut = createAndroidMediaPreProcessor(context, sdkIntVersion)
val file = getFileFromAssets(context, asset.filename)
val result = sut.process(
uri = file.toUri(),
mimeType = MimeTypes.Png,
deleteOriginal = false,
compressIfPossible = true,
mimeType = asset.mimeType,
deleteOriginal = deleteOriginal,
compressIfPossible = compressIfPossible,
)
val data = result.getOrThrow()
assertThat(data.file.path).endsWith("image.png")
val info = data as MediaUploadInfo.Image
assertThat(info.thumbnailFile).isNotNull()
assertThat(info.imageInfo).isEqualTo(
ImageInfo(
height = 1_178,
width = 1_818,
mimetype = MimeTypes.Png,
size = 109_908,
ThumbnailInfo(height = 294, width = 454, mimetype = "image/jpeg", size = 4484),
thumbnailSource = null,
blurhash = "K13]7q%zWC00R4of%\$baad"
)
)
assertThat(file.exists()).isTrue()
assertThat(data.file.path).endsWith(asset.filename)
return data
}
@Test
fun `test processing image api Q`() = runTest {
val context = InstrumentationRegistry.getInstrumentation().context
val sut = createAndroidMediaPreProcessor(context, sdkIntVersion = Build.VERSION_CODES.Q)
val file = getFileFromAssets(context, "image.png")
val result = sut.process(
uri = file.toUri(),
mimeType = MimeTypes.Png,
deleteOriginal = false,
fun `test processing png`() = runTest {
val mediaUploadInfo = process(
asset = assetImagePng,
compressIfPossible = true,
)
val data = result.getOrThrow()
assertThat(data.file.path).endsWith("image.png")
val info = data as MediaUploadInfo.Image
val info = mediaUploadInfo as MediaUploadInfo.Image
assertThat(info.thumbnailFile).isNotNull()
assertThat(info.imageInfo).isEqualTo(
ImageInfo(
height = assetImagePng.height,
width = assetImagePng.width,
mimetype = assetImagePng.mimeType,
size = assetImagePng.size,
ThumbnailInfo(height = 25, width = 25, mimetype = MimeTypes.Png, size = 91),
thumbnailSource = null,
blurhash = "K00000fQfQfQfQfQfQfQfQ"
)
)
}
@Test
fun `test processing png api Q`() = runTest {
val mediaUploadInfo = process(
asset = assetImagePng,
compressIfPossible = true,
sdkIntVersion = Build.VERSION_CODES.Q,
)
val info = mediaUploadInfo as MediaUploadInfo.Image
assertThat(info.thumbnailFile).isNull()
assertThat(info.imageInfo).isEqualTo(
ImageInfo(
height = 1_178,
width = 1_818,
mimetype = MimeTypes.Png,
size = 109_908,
height = assetImagePng.height,
width = assetImagePng.width,
mimetype = assetImagePng.mimeType,
size = assetImagePng.size,
thumbnailInfo = null,
thumbnailSource = null,
blurhash = null,
)
)
assertThat(file.exists()).isTrue()
}
@Test
fun `test processing image no compression`() = runTest {
val context = InstrumentationRegistry.getInstrumentation().context
val sut = createAndroidMediaPreProcessor(context)
val file = getFileFromAssets(context, "image.png")
val result = sut.process(
uri = file.toUri(),
mimeType = MimeTypes.Png,
deleteOriginal = false,
fun `test processing png no compression`() = runTest {
val mediaUploadInfo = process(
asset = assetImagePng,
compressIfPossible = false,
).getOrThrow()
assertThat(result.file.path).endsWith("image.png")
val info = result as MediaUploadInfo.Image
)
val info = mediaUploadInfo as MediaUploadInfo.Image
assertThat(info.thumbnailFile).isNotNull()
assertThat(info.imageInfo).isEqualTo(
ImageInfo(
height = 1_178,
width = 1_818,
mimetype = MimeTypes.Png,
size = 1_856_786,
thumbnailInfo = ThumbnailInfo(height = 25, width = 25, mimetype = MimeTypes.Jpeg, size = 643),
height = assetImagePng.height,
width = assetImagePng.width,
mimetype = assetImagePng.mimeType,
size = assetImagePng.size,
thumbnailInfo = ThumbnailInfo(height = 25, width = 25, mimetype = MimeTypes.Png, size = 91),
thumbnailSource = null,
blurhash = "K00000fQfQfQfQfQfQfQfQ",
)
)
assertThat(file.exists()).isTrue()
}
@Test
fun `test processing image and delete`() = runTest {
val context = InstrumentationRegistry.getInstrumentation().context
val sut = createAndroidMediaPreProcessor(context)
val file = getFileFromAssets(context, "image.png")
val result = sut.process(
uri = file.toUri(),
mimeType = MimeTypes.Png,
deleteOriginal = true,
fun `test processing png and delete`() = runTest {
val mediaUploadInfo = process(
asset = assetImagePng,
compressIfPossible = false,
).getOrThrow()
assertThat(result.file.path).endsWith("image.png")
val info = result as MediaUploadInfo.Image
deleteOriginal = true,
)
val info = mediaUploadInfo as MediaUploadInfo.Image
assertThat(info.thumbnailFile).isNotNull()
assertThat(info.imageInfo).isEqualTo(
ImageInfo(
height = 1_178,
width = 1_818,
mimetype = MimeTypes.Png,
size = 1_856_786,
thumbnailInfo = ThumbnailInfo(height = 25, width = 25, mimetype = MimeTypes.Jpeg, size = 643),
height = assetImagePng.height,
width = assetImagePng.width,
mimetype = assetImagePng.mimeType,
size = assetImagePng.size,
thumbnailInfo = ThumbnailInfo(height = 25, width = 25, mimetype = MimeTypes.Png, size = 91),
thumbnailSource = null,
blurhash = "K00000fQfQfQfQfQfQfQfQ",
)
)
// Does not work
// assertThat(file.exists()).isFalse()
}
@Test
fun `test processing jpeg`() = runTest {
val mediaUploadInfo = process(
asset = assetImageJpeg,
compressIfPossible = true,
)
val info = mediaUploadInfo as MediaUploadInfo.Image
assertThat(info.thumbnailFile).isNotNull()
assertThat(info.imageInfo).isEqualTo(
ImageInfo(
height = 979,
width = 3006,
mimetype = MimeTypes.Jpeg,
size = 84_845,
ThumbnailInfo(height = 244, width = 751, mimetype = MimeTypes.Jpeg, size = 7_178),
thumbnailSource = null,
blurhash = "K07gBzX=j_D4xZjoaSe,s:"
)
)
}
@Test
fun `test processing jpeg api Q`() = runTest {
val mediaUploadInfo = process(
asset = assetImageJpeg,
compressIfPossible = true,
sdkIntVersion = Build.VERSION_CODES.Q,
)
val info = mediaUploadInfo as MediaUploadInfo.Image
assertThat(info.thumbnailFile).isNull()
assertThat(info.imageInfo).isEqualTo(
ImageInfo(
height = 979,
width = 3_006,
mimetype = MimeTypes.Jpeg,
size = 84_845,
thumbnailInfo = null,
thumbnailSource = null,
blurhash = null,
)
)
}
@Test
fun `test processing jpeg no compression`() = runTest {
val mediaUploadInfo = process(
asset = assetImageJpeg,
compressIfPossible = false,
)
val info = mediaUploadInfo as MediaUploadInfo.Image
assertThat(info.thumbnailFile).isNotNull()
assertThat(info.imageInfo).isEqualTo(
ImageInfo(
height = assetImageJpeg.height,
width = assetImageJpeg.width,
mimetype = assetImageJpeg.mimeType,
size = assetImageJpeg.size,
thumbnailInfo = ThumbnailInfo(height = 6, width = 6, mimetype = MimeTypes.Jpeg, size = 631),
thumbnailSource = null,
blurhash = "K00000fQfQfQfQfQfQfQfQ",
)
)
}
@Test
fun `test processing jpeg and delete`() = runTest {
val mediaUploadInfo = process(
asset = assetImageJpeg,
compressIfPossible = false,
deleteOriginal = true,
)
val info = mediaUploadInfo as MediaUploadInfo.Image
assertThat(info.thumbnailFile).isNotNull()
assertThat(info.imageInfo).isEqualTo(
ImageInfo(
height = assetImageJpeg.height,
width = assetImageJpeg.width,
mimetype = assetImageJpeg.mimeType,
size = assetImageJpeg.size,
thumbnailInfo = ThumbnailInfo(height = 6, width = 6, mimetype = MimeTypes.Jpeg, size = 631),
thumbnailSource = null,
blurhash = "K00000fQfQfQfQfQfQfQfQ",
)
@ -152,70 +233,50 @@ class AndroidMediaPreProcessorTest {
@Test
fun `test processing gif`() = runTest {
val context = InstrumentationRegistry.getInstrumentation().context
val sut = createAndroidMediaPreProcessor(context)
val file = getFileFromAssets(context, "animated_gif.gif")
val result = sut.process(
uri = file.toUri(),
mimeType = MimeTypes.Gif,
deleteOriginal = false,
val mediaUploadInfo = process(
asset = assetAnimatedGif,
compressIfPossible = true,
).getOrThrow()
assertThat(result.file.path).endsWith("animated_gif.gif")
val info = result as MediaUploadInfo.Image
)
val info = mediaUploadInfo as MediaUploadInfo.Image
assertThat(info.thumbnailFile).isNotNull()
assertThat(info.imageInfo).isEqualTo(
ImageInfo(
height = 600,
width = 800,
mimetype = MimeTypes.Gif,
size = 687_979,
height = assetAnimatedGif.height,
width = assetAnimatedGif.width,
mimetype = assetAnimatedGif.mimeType,
size = assetAnimatedGif.size,
thumbnailInfo = ThumbnailInfo(height = 50, width = 50, mimetype = MimeTypes.Jpeg, size = 691),
thumbnailSource = null,
blurhash = "K00000fQfQfQfQfQfQfQfQ",
)
)
assertThat(file.exists()).isTrue()
}
@Test
fun `test processing file`() = runTest {
val context = InstrumentationRegistry.getInstrumentation().context
val sut = createAndroidMediaPreProcessor(context)
val file = getFileFromAssets(context, "text.txt")
val result = sut.process(
uri = file.toUri(),
mimeType = MimeTypes.PlainText,
deleteOriginal = false,
val mediaUploadInfo = process(
asset = assetText,
compressIfPossible = true,
).getOrThrow()
assertThat(result.file.path).endsWith("text.txt")
val info = result as MediaUploadInfo.AnyFile
)
val info = mediaUploadInfo as MediaUploadInfo.AnyFile
assertThat(info.fileInfo).isEqualTo(
FileInfo(
mimetype = MimeTypes.PlainText,
size = 13,
mimetype = assetText.mimeType,
size = assetText.size,
thumbnailInfo = null,
thumbnailSource = null,
)
)
assertThat(file.exists()).isTrue()
}
@Ignore("Compressing video is not working with Robolectric")
@Test
fun `test processing video`() = runTest {
val context = InstrumentationRegistry.getInstrumentation().context
val sut = createAndroidMediaPreProcessor(context)
val file = getFileFromAssets(context, "video.mp4")
val result = sut.process(
uri = file.toUri(),
mimeType = MimeTypes.Mp4,
deleteOriginal = false,
val mediaUploadInfo = process(
asset = assetVideo,
compressIfPossible = true,
).getOrThrow()
assertThat(result.file.path).endsWith("video.mp4")
val info = result as MediaUploadInfo.Video
)
val info = mediaUploadInfo as MediaUploadInfo.Video
assertThat(info.thumbnailFile).isNotNull()
assertThat(info.videoInfo).isEqualTo(
VideoInfo(
@ -230,22 +291,16 @@ class AndroidMediaPreProcessorTest {
blurhash = null,
)
)
assertThat(file.exists()).isTrue()
}
@Ignore("Compressing video is not working with Robolectric")
@Test
fun `test processing video no compression`() = runTest {
val context = InstrumentationRegistry.getInstrumentation().context
val sut = createAndroidMediaPreProcessor(context)
val file = getFileFromAssets(context, "video.mp4")
val result = sut.process(
uri = file.toUri(),
mimeType = MimeTypes.Mp4,
deleteOriginal = false,
val mediaUploadInfo = process(
asset = assetVideo,
compressIfPossible = false,
).getOrThrow()
assertThat(result.file.path).endsWith("video.mp4")
val info = result as MediaUploadInfo.Video
)
val info = mediaUploadInfo as MediaUploadInfo.Video
// Computing thumbnailFile is failing with Robolectric
assertThat(info.thumbnailFile).isNull()
assertThat(info.videoInfo).isEqualTo(
@ -263,22 +318,15 @@ class AndroidMediaPreProcessorTest {
blurhash = null,
)
)
assertThat(file.exists()).isTrue()
}
@Test
fun `test processing audio`() = runTest {
val context = InstrumentationRegistry.getInstrumentation().context
val sut = createAndroidMediaPreProcessor(context)
val file = getFileFromAssets(context, "sample3s.mp3")
val result = sut.process(
uri = file.toUri(),
mimeType = MimeTypes.Mp3,
deleteOriginal = false,
val mediaUploadInfo = process(
asset = assetAudio,
compressIfPossible = true,
).getOrThrow()
assertThat(result.file.path).endsWith("sample3s.mp3")
val info = result as MediaUploadInfo.Audio
)
val info = mediaUploadInfo as MediaUploadInfo.Audio
assertThat(info.audioInfo).isEqualTo(
AudioInfo(
// Not available with Robolectric?
@ -287,7 +335,6 @@ class AndroidMediaPreProcessorTest {
mimetype = MimeTypes.Mp3,
)
)
assertThat(file.exists()).isTrue()
}
@Test

View file

@ -0,0 +1,85 @@
/*
* Copyright 2024 New Vector Ltd.
*
* SPDX-License-Identifier: AGPL-3.0-only
* Please see LICENSE in the repository root for full details.
*/
package io.element.android.libraries.mediaupload.impl
import io.element.android.libraries.core.mimetype.MimeTypes
data class Asset(
val filename: String,
val mimeType: String,
val size: Long,
val width: Long?,
val height: Long?,
)
/**
* "image.png" is a 1_818 x 1_178 PNG image with a size of 1_856_786 bytes.
*/
val assetImagePng = Asset(
filename = "image.png",
mimeType = MimeTypes.Png,
size = 1_856_786,
width = 1_818,
height = 1_178,
)
/**
* "image.jpeg" is a 12_024 x 3_916, JPEG image with a size of 9_986_336 bytes.
*/
val assetImageJpeg = Asset(
filename = "image.jpeg",
mimeType = MimeTypes.Jpeg,
size = 9_986_336,
width = 12_024,
height = 3_916,
)
/**
* "video.mp4" is a 1_280 x 720, MP4 video with a size of 1_673_712 bytes.
*/
val assetVideo = Asset(
filename = "video.mp4",
mimeType = MimeTypes.Mp4,
size = 1_673_712,
width = 1_280,
height = 720,
)
/**
* "sample3s.mp3" is a 3 seconds MP3 audio file with a size of 52_079 bytes.
*/
val assetAudio = Asset(
filename = "sample3s.mp3",
mimeType = MimeTypes.Mp3,
size = 52_079,
width = null,
height = null,
)
/**
* "text.txt" is a 13 bytes text file.
*/
val assetText = Asset(
filename = "text.txt",
mimeType = MimeTypes.PlainText,
size = 13,
width = null,
height = null,
)
/**
* "animated_gif.gif" is a 800 x 600, GIF image with a size of 687_979 bytes.
*/
val assetAnimatedGif = Asset(
filename = "animated_gif.gif",
mimeType = MimeTypes.Gif,
size = 687_979,
width = 800,
height = 600,
)