Skip to content

Commit 4df6b87

Browse files
Fix attachment table archive eligibility tests.
1 parent 7d16e85 commit 4df6b87

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

app/src/androidTest/java/org/thoughtcrime/securesms/database/AttachmentTableTest.kt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)