@@ -259,6 +259,7 @@ class AttachmentTableTest {
259259 val messageId = SignalDatabase .messages.insertMessageInbox(message).map { it.messageId }.get()
260260 SignalDatabase .attachments.setArchiveTransferState(AttachmentId (1L ), AttachmentTable .ArchiveTransferState .NONE )
261261 SignalDatabase .attachments.setTransferState(messageId, AttachmentId (1L ), AttachmentTable .TRANSFER_PROGRESS_DONE )
262+ SignalDatabase .attachments.finalizeAttachmentAfterUpload(AttachmentId (1L ), AttachmentTableTestUtil .createUploadResult(AttachmentId (1L )))
262263
263264 // WHEN
264265 val attachments = SignalDatabase .attachments.getAttachmentsThatNeedArchiveUpload()
@@ -277,6 +278,7 @@ class AttachmentTableTest {
277278 val messageId = SignalDatabase .messages.insertMessageInbox(message).map { it.messageId }.get()
278279 SignalDatabase .attachments.setArchiveTransferState(AttachmentId (1L ), AttachmentTable .ArchiveTransferState .NONE )
279280 SignalDatabase .attachments.setTransferState(messageId, AttachmentId (1L ), AttachmentTable .TRANSFER_PROGRESS_DONE )
281+ SignalDatabase .attachments.finalizeAttachmentAfterUpload(AttachmentId (1L ), AttachmentTableTestUtil .createUploadResult(AttachmentId (1L )))
280282
281283 // WHEN
282284 val attachments = SignalDatabase .attachments.getAttachmentsThatNeedArchiveUpload()
@@ -296,6 +298,7 @@ class AttachmentTableTest {
296298 SignalDatabase .messages.markExpireStarted(messageId)
297299 SignalDatabase .attachments.setArchiveTransferState(AttachmentId (1L ), AttachmentTable .ArchiveTransferState .NONE )
298300 SignalDatabase .attachments.setTransferState(messageId, AttachmentId (1L ), AttachmentTable .TRANSFER_PROGRESS_DONE )
301+ SignalDatabase .attachments.finalizeAttachmentAfterUpload(AttachmentId (1L ), AttachmentTableTestUtil .createUploadResult(AttachmentId (1L )))
299302
300303 // WHEN
301304 val attachments = SignalDatabase .attachments.getAttachmentsThatNeedArchiveUpload()
@@ -314,6 +317,7 @@ class AttachmentTableTest {
314317 val messageId = SignalDatabase .messages.insertMessageInbox(message).map { it.messageId }.get()
315318 SignalDatabase .attachments.setArchiveTransferState(AttachmentId (1L ), AttachmentTable .ArchiveTransferState .NONE )
316319 SignalDatabase .attachments.setTransferState(messageId, AttachmentId (1L ), AttachmentTable .TRANSFER_PROGRESS_DONE )
320+ SignalDatabase .attachments.finalizeAttachmentAfterUpload(AttachmentId (1L ), AttachmentTableTestUtil .createUploadResult(AttachmentId (1L )))
317321
318322 // WHEN
319323 val attachments = SignalDatabase .attachments.getAttachmentsThatNeedArchiveUpload()
0 commit comments