fix: Asset message resend transfer status #WPB-11035

wireapp · Oct 15, 2024 · aa3699e · aa3699e
1 parent 0bc15da
commit aa3699e
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 1 deletion.
diff --git a/.../src/commonMain/kotlin/com/wire/kalium/logic/feature/message/RetryFailedMessageUseCase.kt b/.../src/commonMain/kotlin/com/wire/kalium/logic/feature/message/RetryFailedMessageUseCase.kt
@@ -178,7 +178,10 @@ class RetryFailedMessageUseCase internal constructor(
 
             else -> handleError("Asset message with transfer status $assetTransferStatus cannot be retried")
         }
-            .onSuccess { retrySendingMessage(it) }
+            .onSuccess {
+                updateAssetMessageTransferStatus(AssetTransferStatus.UPLOADED, message.conversationId, message.id)
+                retrySendingMessage(it)
+            }
             .map { /* returns Unit */ }
     }
 

diff --git a/.../commonTest/kotlin/com/wire/kalium/logic/feature/message/RetryFailedMessageUseCaseTest.kt b/.../commonTest/kotlin/com/wire/kalium/logic/feature/message/RetryFailedMessageUseCaseTest.kt
@@ -159,6 +159,38 @@ class RetryFailedMessageUseCaseTest {
             }.wasInvoked(exactly = once)
         }
 
+    @Test
+    fun givenAValidFailedAndNotUploadedAssetMessage_whenSuccessfullyUploadedAsset_thenAssetTransferShouldBeChangedToUploaded() =
+        runTest(testDispatcher.default) {
+            // given
+            val name = "some_asset.txt"
+            val content = MessageContent.Asset(ASSET_CONTENT.value.copy(name = name))
+            val path = fakeKaliumFileSystem.providePersistentAssetPath(name)
+            val message = assetMessage().copy(content = content, status = Message.Status.Failed)
+            val uploadedAssetId = UploadedAssetId("remote_key", "remote_domain", "remote_token")
+            val uploadedAssetSha = SHA256Key(byteArrayOf())
+            val (arrangement, useCase) = Arrangement()
+                .withGetMessageById(Either.Right(message))
+                .withUpdateMessageStatus(Either.Right(Unit))
+                .withUpdateAssetMessageTransferStatus(UpdateTransferStatusResult.Success)
+                .withFetchPrivateDecodedAsset(Either.Right(path))
+                .withStoredData(mockedLongAssetData(), path)
+                .withGetAssetMessageTransferStatus(AssetTransferStatus.FAILED_UPLOAD)
+                .withUploadAndPersistPrivateAsset(Either.Right(uploadedAssetId to uploadedAssetSha))
+                .withPersistMessage(Either.Right(Unit))
+                .withSendMessage(Either.Right(Unit))
+                .arrange()
+
+            // when
+            useCase.invoke(message.id, message.conversationId)
+            advanceUntilIdle()
+
+            // then
+            coVerify {
+                arrangement.updateAssetMessageTransferStatus.invoke(AssetTransferStatus.UPLOADED, message.conversationId, message.id)
+            }.wasInvoked(exactly = once)
+        }
+
     @Test
     fun givenAValidFailedAndNotUploadedAssetMessage_whenRetryingFailedMessage_thenUploadAssetAndSendAMessageWithProperAssetRemoteData() =
         runTest(testDispatcher.default) {