516 lines
38 KiB
Swift
516 lines
38 KiB
Swift
//
|
|
// Copyright (c) 2019 Open Whisper Systems. All rights reserved.
|
|
//
|
|
|
|
import XCTest
|
|
import SignalServiceKit
|
|
@testable import Signal
|
|
@testable import SignalMessaging
|
|
|
|
class YDBToGRDBMigrationModelTest: SignalBaseTest {
|
|
|
|
override func setUp() {
|
|
super.setUp()
|
|
// Put setup code here. This method is called before the invocation of each test method in the class.
|
|
}
|
|
|
|
override func tearDown() {
|
|
// Put teardown code here. This method is called after the invocation of each test method in the class.
|
|
super.tearDown()
|
|
}
|
|
|
|
func randomStickerPackInfo() -> StickerPackInfo {
|
|
let packId = Randomness.generateRandomBytes(16)
|
|
let packKey = Randomness.generateRandomBytes(Int32(StickerManager.packKeyLength))
|
|
|
|
return StickerPackInfo(packId: packId, packKey: packKey)
|
|
}
|
|
|
|
func randomKnownStickerPack() -> KnownStickerPack {
|
|
return KnownStickerPack(info: randomStickerPackInfo())
|
|
}
|
|
|
|
func testKnownStickerPack() {
|
|
let model1 = randomKnownStickerPack()
|
|
let model2 = randomKnownStickerPack()
|
|
let model3 = randomKnownStickerPack()
|
|
|
|
self.yapRead { transaction in
|
|
XCTAssertNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model1.info), transaction: transaction.asAnyRead))
|
|
XCTAssertNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model2.info), transaction: transaction.asAnyRead))
|
|
XCTAssertNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model3.info), transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(0, KnownStickerPack.anyCount(transaction: transaction.asAnyRead))
|
|
}
|
|
self.read { transaction in
|
|
XCTAssertNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model1.info), transaction: transaction))
|
|
XCTAssertNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model2.info), transaction: transaction))
|
|
XCTAssertNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model3.info), transaction: transaction))
|
|
XCTAssertEqual(0, KnownStickerPack.anyCount(transaction: transaction))
|
|
}
|
|
|
|
self.yapWrite { transaction in
|
|
model1.anyInsert(transaction: transaction.asAnyWrite)
|
|
model2.anyInsert(transaction: transaction.asAnyWrite)
|
|
// Don't insert 3.
|
|
}
|
|
|
|
self.yapRead { transaction in
|
|
XCTAssertNotNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model1.info), transaction: transaction.asAnyRead))
|
|
XCTAssertNotNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model2.info), transaction: transaction.asAnyRead))
|
|
XCTAssertNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model3.info), transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, KnownStickerPack.anyCount(transaction: transaction.asAnyRead))
|
|
}
|
|
self.read { transaction in
|
|
XCTAssertNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model1.info), transaction: transaction))
|
|
XCTAssertNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model2.info), transaction: transaction))
|
|
XCTAssertNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model3.info), transaction: transaction))
|
|
XCTAssertEqual(0, KnownStickerPack.anyCount(transaction: transaction))
|
|
}
|
|
|
|
let migratorGroups = [
|
|
GRDBMigratorGroup { ydbTransaction in
|
|
return [
|
|
GRDBUnorderedRecordMigrator<KnownStickerPack>(label: "KnownStickerPack", ydbTransaction: ydbTransaction, memorySamplerRatio: 0.2)
|
|
]
|
|
}
|
|
]
|
|
|
|
try! YDBToGRDBMigration().migrate(migratorGroups: migratorGroups)
|
|
|
|
self.yapRead { transaction in
|
|
XCTAssertNotNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model1.info), transaction: transaction.asAnyRead))
|
|
XCTAssertNotNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model2.info), transaction: transaction.asAnyRead))
|
|
XCTAssertNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model3.info), transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, KnownStickerPack.anyCount(transaction: transaction.asAnyRead))
|
|
}
|
|
self.read { transaction in
|
|
XCTAssertNotNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model1.info), transaction: transaction))
|
|
XCTAssertNotNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model2.info), transaction: transaction))
|
|
XCTAssertNil(KnownStickerPack.anyFetch(uniqueId: KnownStickerPack.uniqueId(for: model3.info), transaction: transaction))
|
|
XCTAssertEqual(2, KnownStickerPack.anyCount(transaction: transaction))
|
|
}
|
|
}
|
|
|
|
func testJobs() {
|
|
// SSKMessageDecryptJobRecord
|
|
let messageDecryptData1 = Randomness.generateRandomBytes(1024)
|
|
let messageDecryptData2 = Randomness.generateRandomBytes(1024)
|
|
// OWSSessionResetJobRecord
|
|
let contactThread1 = TSContactThread(contactAddress: SignalServiceAddress(phoneNumber: "+13213334444"))
|
|
let contactThread2 = TSContactThread(contactAddress: SignalServiceAddress(phoneNumber: "+13213334445"))
|
|
// SSKMessageSenderJobRecord
|
|
let outgoingMessage1 = TSOutgoingMessage(in: contactThread1, messageBody: "good heavens", attachmentId: nil)
|
|
let outgoingMessage2 = TSOutgoingMessage(in: contactThread1, messageBody: "land's sakes", attachmentId: nil)
|
|
let outgoingMessage3 = TSOutgoingMessage(in: contactThread2, messageBody: "oh my word", attachmentId: nil)
|
|
// OWSMessageDecryptJob
|
|
let messageDecryptData3 = Randomness.generateRandomBytes(1024)
|
|
let messageDecryptData4 = Randomness.generateRandomBytes(1024)
|
|
|
|
// SSKMessageDecryptJobRecord
|
|
let messageDecryptJobFinder1 = AnyJobRecordFinder<SSKMessageDecryptJobRecord>()
|
|
let messageDecryptJobQueue = SSKMessageDecryptJobQueue()
|
|
// OWSSessionResetJobRecord
|
|
let sessionResetJobFinder = AnyJobRecordFinder<OWSSessionResetJobRecord>()
|
|
let sessionResetJobQueue = SessionResetJobQueue()
|
|
// SSKMessageSenderJobRecord
|
|
let messageSenderJobFinder = AnyJobRecordFinder<SSKMessageSenderJobRecord>()
|
|
let messageSenderJobQueue = MessageSenderJobQueue()
|
|
// OWSMessageDecryptJob
|
|
let messageDecryptJobFinder2 = OWSMessageDecryptJobFinder()
|
|
|
|
self.yapRead { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
XCTAssertNil(messageDecryptJobFinder1.getNextReady(label: SSKMessageDecryptJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(0, messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
// OWSSessionResetJobRecord
|
|
XCTAssertNil(sessionResetJobFinder.getNextReady(label: sessionResetJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(0, sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
// SSKMessageSenderJobRecord
|
|
XCTAssertNil(messageSenderJobFinder.getNextReady(label: MessageSenderJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(0, messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
// OWSMessageDecryptJob
|
|
XCTAssertNil(messageDecryptJobFinder2.nextJob(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(0, messageDecryptJobFinder2.queuedJobCount(with: transaction.asAnyRead))
|
|
}
|
|
self.read { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
XCTAssertNil(messageDecryptJobFinder1.getNextReady(label: SSKMessageDecryptJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// OWSSessionResetJobRecord
|
|
XCTAssertNil(sessionResetJobFinder.getNextReady(label: sessionResetJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// SSKMessageSenderJobRecord
|
|
XCTAssertNil(messageSenderJobFinder.getNextReady(label: MessageSenderJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// OWSMessageDecryptJob
|
|
//
|
|
// NOTE: We don't need to verify that GRDB contains no
|
|
// OWSMessageDecryptJobs; the table doesn't even exist.
|
|
}
|
|
|
|
self.yapWrite { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
messageDecryptJobQueue.add(envelopeData: messageDecryptData1, transaction: transaction.asAnyWrite)
|
|
messageDecryptJobQueue.add(envelopeData: messageDecryptData2, transaction: transaction.asAnyWrite)
|
|
// OWSSessionResetJobRecord
|
|
sessionResetJobQueue.add(contactThread: contactThread1, transaction: transaction.asAnyWrite)
|
|
sessionResetJobQueue.add(contactThread: contactThread2, transaction: transaction.asAnyWrite)
|
|
// SSKMessageSenderJobRecord
|
|
contactThread1.anyInsert(transaction: transaction.asAnyWrite)
|
|
contactThread2.anyInsert(transaction: transaction.asAnyWrite)
|
|
outgoingMessage1.anyInsert(transaction: transaction.asAnyWrite)
|
|
outgoingMessage2.anyInsert(transaction: transaction.asAnyWrite)
|
|
outgoingMessage3.anyInsert(transaction: transaction.asAnyWrite)
|
|
messageSenderJobQueue.add(message: outgoingMessage1.asPreparer, transaction: transaction.asAnyWrite)
|
|
messageSenderJobQueue.add(message: outgoingMessage2.asPreparer, transaction: transaction.asAnyWrite)
|
|
messageSenderJobQueue.add(message: outgoingMessage3.asPreparer, transaction: transaction.asAnyWrite)
|
|
// OWSMessageDecryptJob
|
|
messageDecryptJobFinder2.addJob(forEnvelopeData: messageDecryptData3, transaction: transaction.asAnyWrite)
|
|
messageDecryptJobFinder2.addJob(forEnvelopeData: messageDecryptData4, transaction: transaction.asAnyWrite)
|
|
}
|
|
|
|
self.yapRead { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
XCTAssertNotNil(messageDecryptJobFinder1.getNextReady(label: SSKMessageDecryptJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
XCTAssertEqual([messageDecryptData1, messageDecryptData2 ], messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).compactMap { $0.envelopeData })
|
|
// OWSSessionResetJobRecord
|
|
XCTAssertNotNil(sessionResetJobFinder.getNextReady(label: sessionResetJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
XCTAssertEqual([contactThread1.uniqueId, contactThread2.uniqueId ], sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).compactMap { $0.contactThreadId })
|
|
// SSKMessageSenderJobRecord
|
|
XCTAssertNotNil(messageSenderJobFinder.getNextReady(label: MessageSenderJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(3, messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
XCTAssertEqual([outgoingMessage1.uniqueId, outgoingMessage2.uniqueId, outgoingMessage3.uniqueId ], messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).compactMap { $0.messageId })
|
|
// OWSMessageDecryptJob
|
|
XCTAssertNotNil(messageDecryptJobFinder2.nextJob(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, messageDecryptJobFinder2.queuedJobCount(with: transaction.asAnyRead))
|
|
XCTAssertEqual(messageDecryptData3, messageDecryptJobFinder2.nextJob(transaction: transaction.asAnyRead)!.envelopeData)
|
|
}
|
|
self.read { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
XCTAssertNil(messageDecryptJobFinder1.getNextReady(label: SSKMessageDecryptJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// OWSSessionResetJobRecord
|
|
XCTAssertNil(sessionResetJobFinder.getNextReady(label: sessionResetJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// SSKMessageSenderJobRecord
|
|
XCTAssertNil(messageSenderJobFinder.getNextReady(label: MessageSenderJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// OWSMessageDecryptJob
|
|
//
|
|
// NOTE: We don't need to verify that GRDB contains no
|
|
// OWSMessageDecryptJobs; the table doesn't even exist.
|
|
}
|
|
|
|
let migratorGroups = [
|
|
GRDBMigratorGroup { ydbTransaction in
|
|
return [
|
|
GRDBJobRecordMigrator(ydbTransaction: ydbTransaction)
|
|
]
|
|
}
|
|
]
|
|
|
|
try! YDBToGRDBMigration().migrate(migratorGroups: migratorGroups)
|
|
|
|
self.yapRead { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
XCTAssertNotNil(messageDecryptJobFinder1.getNextReady(label: SSKMessageDecryptJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
XCTAssertEqual([messageDecryptData1, messageDecryptData2 ], messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).compactMap { $0.envelopeData })
|
|
// OWSSessionResetJobRecord
|
|
XCTAssertNotNil(sessionResetJobFinder.getNextReady(label: sessionResetJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
XCTAssertEqual([contactThread1.uniqueId, contactThread2.uniqueId ], sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).compactMap { $0.contactThreadId })
|
|
// SSKMessageSenderJobRecord
|
|
XCTAssertNotNil(messageSenderJobFinder.getNextReady(label: MessageSenderJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(3, messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
XCTAssertEqual([outgoingMessage1.uniqueId, outgoingMessage2.uniqueId, outgoingMessage3.uniqueId ], messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).compactMap { $0.messageId })
|
|
// OWSMessageDecryptJob
|
|
XCTAssertNotNil(messageDecryptJobFinder2.nextJob(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, messageDecryptJobFinder2.queuedJobCount(with: transaction.asAnyRead))
|
|
XCTAssertEqual(messageDecryptData3, messageDecryptJobFinder2.nextJob(transaction: transaction.asAnyRead)!.envelopeData)
|
|
}
|
|
self.read { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
XCTAssertNotNil(messageDecryptJobFinder1.getNextReady(label: SSKMessageDecryptJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(2, messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
XCTAssertEqual([messageDecryptData1, messageDecryptData2 ], messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction).compactMap { $0.envelopeData })
|
|
// OWSSessionResetJobRecord
|
|
XCTAssertNotNil(sessionResetJobFinder.getNextReady(label: sessionResetJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(2, sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
XCTAssertEqual([contactThread1.uniqueId, contactThread2.uniqueId ], sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction).compactMap { $0.contactThreadId })
|
|
// SSKMessageSenderJobRecord
|
|
XCTAssertNotNil(messageSenderJobFinder.getNextReady(label: MessageSenderJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(3, messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
XCTAssertEqual([outgoingMessage1.uniqueId, outgoingMessage2.uniqueId, outgoingMessage3.uniqueId ], messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction).compactMap { $0.messageId })
|
|
// OWSMessageDecryptJob
|
|
//
|
|
// NOTE: These jobs are _NOT_ migrated.
|
|
// See testDecryptJobs() which uses GRDBDecryptJobMigrator.
|
|
//
|
|
// NOTE: We don't need to verify that GRDB contains no
|
|
// OWSMessageDecryptJobs; the table doesn't even exist.
|
|
}
|
|
}
|
|
|
|
func testDecryptJobs() {
|
|
// SSKMessageDecryptJobRecord
|
|
let messageDecryptData1 = Randomness.generateRandomBytes(1024)
|
|
let messageDecryptData2 = Randomness.generateRandomBytes(1024)
|
|
// OWSSessionResetJobRecord
|
|
let contactThread1 = TSContactThread(contactAddress: SignalServiceAddress(phoneNumber: "+13213334444"))
|
|
let contactThread2 = TSContactThread(contactAddress: SignalServiceAddress(phoneNumber: "+13213334445"))
|
|
// SSKMessageSenderJobRecord
|
|
let outgoingMessage1 = TSOutgoingMessage(in: contactThread1, messageBody: "good heavens", attachmentId: nil)
|
|
let outgoingMessage2 = TSOutgoingMessage(in: contactThread1, messageBody: "land's sakes", attachmentId: nil)
|
|
let outgoingMessage3 = TSOutgoingMessage(in: contactThread2, messageBody: "oh my word", attachmentId: nil)
|
|
// OWSMessageDecryptJob
|
|
let messageDecryptData3 = Randomness.generateRandomBytes(1024)
|
|
let messageDecryptData4 = Randomness.generateRandomBytes(1024)
|
|
|
|
// SSKMessageDecryptJobRecord
|
|
let messageDecryptJobFinder1 = AnyJobRecordFinder<SSKMessageDecryptJobRecord>()
|
|
let messageDecryptJobQueue = SSKMessageDecryptJobQueue()
|
|
// OWSSessionResetJobRecord
|
|
let sessionResetJobFinder = AnyJobRecordFinder<OWSSessionResetJobRecord>()
|
|
let sessionResetJobQueue = SessionResetJobQueue()
|
|
// SSKMessageSenderJobRecord
|
|
let messageSenderJobFinder = AnyJobRecordFinder<SSKMessageSenderJobRecord>()
|
|
let messageSenderJobQueue = MessageSenderJobQueue()
|
|
// OWSMessageDecryptJob
|
|
let messageDecryptJobFinder2 = OWSMessageDecryptJobFinder()
|
|
|
|
self.yapRead { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
XCTAssertNil(messageDecryptJobFinder1.getNextReady(label: SSKMessageDecryptJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(0, messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
// OWSSessionResetJobRecord
|
|
XCTAssertNil(sessionResetJobFinder.getNextReady(label: sessionResetJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(0, sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
// SSKMessageSenderJobRecord
|
|
XCTAssertNil(messageSenderJobFinder.getNextReady(label: MessageSenderJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(0, messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
// OWSMessageDecryptJob
|
|
XCTAssertNil(messageDecryptJobFinder2.nextJob(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(0, messageDecryptJobFinder2.queuedJobCount(with: transaction.asAnyRead))
|
|
}
|
|
self.read { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
XCTAssertNil(messageDecryptJobFinder1.getNextReady(label: SSKMessageDecryptJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// OWSSessionResetJobRecord
|
|
XCTAssertNil(sessionResetJobFinder.getNextReady(label: sessionResetJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// SSKMessageSenderJobRecord
|
|
XCTAssertNil(messageSenderJobFinder.getNextReady(label: MessageSenderJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// OWSMessageDecryptJob
|
|
//
|
|
// NOTE: We don't need to verify that GRDB contains no
|
|
// OWSMessageDecryptJobs; the table doesn't even exist.
|
|
}
|
|
|
|
self.yapWrite { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
messageDecryptJobQueue.add(envelopeData: messageDecryptData1, transaction: transaction.asAnyWrite)
|
|
messageDecryptJobQueue.add(envelopeData: messageDecryptData2, transaction: transaction.asAnyWrite)
|
|
// OWSSessionResetJobRecord
|
|
sessionResetJobQueue.add(contactThread: contactThread1, transaction: transaction.asAnyWrite)
|
|
sessionResetJobQueue.add(contactThread: contactThread2, transaction: transaction.asAnyWrite)
|
|
// SSKMessageSenderJobRecord
|
|
contactThread1.anyInsert(transaction: transaction.asAnyWrite)
|
|
contactThread2.anyInsert(transaction: transaction.asAnyWrite)
|
|
outgoingMessage1.anyInsert(transaction: transaction.asAnyWrite)
|
|
outgoingMessage2.anyInsert(transaction: transaction.asAnyWrite)
|
|
outgoingMessage3.anyInsert(transaction: transaction.asAnyWrite)
|
|
messageSenderJobQueue.add(message: outgoingMessage1.asPreparer, transaction: transaction.asAnyWrite)
|
|
messageSenderJobQueue.add(message: outgoingMessage2.asPreparer, transaction: transaction.asAnyWrite)
|
|
messageSenderJobQueue.add(message: outgoingMessage3.asPreparer, transaction: transaction.asAnyWrite)
|
|
// OWSMessageDecryptJob
|
|
messageDecryptJobFinder2.addJob(forEnvelopeData: messageDecryptData3, transaction: transaction.asAnyWrite)
|
|
messageDecryptJobFinder2.addJob(forEnvelopeData: messageDecryptData4, transaction: transaction.asAnyWrite)
|
|
}
|
|
|
|
self.yapRead { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
XCTAssertNotNil(messageDecryptJobFinder1.getNextReady(label: SSKMessageDecryptJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
XCTAssertEqual([messageDecryptData1, messageDecryptData2 ], messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).compactMap { $0.envelopeData })
|
|
// OWSSessionResetJobRecord
|
|
XCTAssertNotNil(sessionResetJobFinder.getNextReady(label: sessionResetJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
XCTAssertEqual([contactThread1.uniqueId, contactThread2.uniqueId ], sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).compactMap { $0.contactThreadId })
|
|
// SSKMessageSenderJobRecord
|
|
XCTAssertNotNil(messageSenderJobFinder.getNextReady(label: MessageSenderJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(3, messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
XCTAssertEqual([outgoingMessage1.uniqueId, outgoingMessage2.uniqueId, outgoingMessage3.uniqueId ], messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).compactMap { $0.messageId })
|
|
// OWSMessageDecryptJob
|
|
XCTAssertNotNil(messageDecryptJobFinder2.nextJob(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, messageDecryptJobFinder2.queuedJobCount(with: transaction.asAnyRead))
|
|
XCTAssertEqual(messageDecryptData3, messageDecryptJobFinder2.nextJob(transaction: transaction.asAnyRead)!.envelopeData)
|
|
}
|
|
self.read { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
XCTAssertNil(messageDecryptJobFinder1.getNextReady(label: SSKMessageDecryptJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// OWSSessionResetJobRecord
|
|
XCTAssertNil(sessionResetJobFinder.getNextReady(label: sessionResetJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// SSKMessageSenderJobRecord
|
|
XCTAssertNil(messageSenderJobFinder.getNextReady(label: MessageSenderJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// OWSMessageDecryptJob
|
|
//
|
|
// NOTE: We don't need to verify that GRDB contains no
|
|
// OWSMessageDecryptJobs; the table doesn't even exist.
|
|
}
|
|
|
|
let migratorGroups = [
|
|
GRDBMigratorGroup { ydbTransaction in
|
|
return [
|
|
GRDBDecryptJobMigrator(ydbTransaction: ydbTransaction)
|
|
]
|
|
}
|
|
]
|
|
|
|
try! YDBToGRDBMigration().migrate(migratorGroups: migratorGroups)
|
|
|
|
self.yapRead { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
XCTAssertNotNil(messageDecryptJobFinder1.getNextReady(label: SSKMessageDecryptJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
XCTAssertEqual([messageDecryptData1, messageDecryptData2 ], messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).compactMap { $0.envelopeData })
|
|
// OWSSessionResetJobRecord
|
|
XCTAssertNotNil(sessionResetJobFinder.getNextReady(label: sessionResetJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
XCTAssertEqual([contactThread1.uniqueId, contactThread2.uniqueId ], sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).compactMap { $0.contactThreadId })
|
|
// SSKMessageSenderJobRecord
|
|
XCTAssertNotNil(messageSenderJobFinder.getNextReady(label: MessageSenderJobQueue.jobRecordLabel, transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(3, messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).count)
|
|
XCTAssertEqual([outgoingMessage1.uniqueId, outgoingMessage2.uniqueId, outgoingMessage3.uniqueId ], messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction.asAnyRead).compactMap { $0.messageId })
|
|
// OWSMessageDecryptJob
|
|
XCTAssertNotNil(messageDecryptJobFinder2.nextJob(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(2, messageDecryptJobFinder2.queuedJobCount(with: transaction.asAnyRead))
|
|
XCTAssertEqual(messageDecryptData3, messageDecryptJobFinder2.nextJob(transaction: transaction.asAnyRead)!.envelopeData)
|
|
}
|
|
self.read { transaction in
|
|
// SSKMessageDecryptJobRecord
|
|
//
|
|
// NOTE: These jobs are migrated from OWSMessageDecryptJob.
|
|
XCTAssertNotNil(messageDecryptJobFinder1.getNextReady(label: SSKMessageDecryptJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(2, messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
XCTAssertEqual([messageDecryptData3, messageDecryptData4 ], messageDecryptJobFinder1.allRecords(label: SSKMessageDecryptJobQueue.jobRecordLabel, status: .ready, transaction: transaction).compactMap { $0.envelopeData })
|
|
// OWSSessionResetJobRecord
|
|
//
|
|
// NOTE: These jobs are _NOT_ migrated.
|
|
XCTAssertNil(sessionResetJobFinder.getNextReady(label: sessionResetJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, sessionResetJobFinder.allRecords(label: sessionResetJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// SSKMessageSenderJobRecord
|
|
//
|
|
// NOTE: These jobs are _NOT_ migrated.
|
|
XCTAssertNil(messageSenderJobFinder.getNextReady(label: MessageSenderJobQueue.jobRecordLabel, transaction: transaction))
|
|
XCTAssertEqual(0, messageSenderJobFinder.allRecords(label: MessageSenderJobQueue.jobRecordLabel, status: .ready, transaction: transaction).count)
|
|
// OWSMessageDecryptJob
|
|
//
|
|
// NOTE: These jobs are migrated to SSKMessageDecryptJobRecord.
|
|
//
|
|
// NOTE: We don't need to verify that GRDB contains no
|
|
// OWSMessageDecryptJobs; the table doesn't even exist.
|
|
}
|
|
}
|
|
|
|
func testInteractions() {
|
|
// Threads
|
|
let contactThread1 = TSContactThread(contactAddress: SignalServiceAddress(phoneNumber: "+13213334444"))
|
|
let contactThread2 = TSContactThread(contactAddress: SignalServiceAddress(phoneNumber: "+13213334445"))
|
|
// Attachments
|
|
let attachmentData1 = Randomness.generateRandomBytes(1024)
|
|
let attachment1 = TSAttachmentStream(attachmentWithContentType: OWSMimeTypeImageGif,
|
|
byteCount: UInt32(attachmentData1.count),
|
|
sourceFilename: "some.gif", caption: nil, albumMessageId: nil)
|
|
let attachmentData2 = Randomness.generateRandomBytes(2048)
|
|
let attachment2 = TSAttachmentStream(attachmentWithContentType: OWSMimeTypeImagePdf,
|
|
byteCount: UInt32(attachmentData2.count),
|
|
sourceFilename: "some.df", caption: nil, albumMessageId: nil)
|
|
// Messages
|
|
let outgoingMessage1 = TSOutgoingMessage(in: contactThread1, messageBody: "good heavens", attachmentId: attachment1.uniqueId)
|
|
let outgoingMessage2 = TSOutgoingMessage(in: contactThread2, messageBody: "land's sakes", attachmentId: attachment2.uniqueId)
|
|
let outgoingMessage3 = TSOutgoingMessage(in: contactThread2, messageBody: "oh my word", attachmentId: nil)
|
|
|
|
self.yapRead { transaction in
|
|
XCTAssertEqual(0, TSThread.anyCount(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(0, TSInteraction.anyCount(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual(0, TSAttachment.anyCount(transaction: transaction.asAnyRead))
|
|
}
|
|
self.read { transaction in
|
|
XCTAssertEqual(0, TSThread.anyCount(transaction: transaction))
|
|
XCTAssertEqual(0, TSInteraction.anyCount(transaction: transaction))
|
|
XCTAssertEqual(0, TSAttachment.anyCount(transaction: transaction))
|
|
}
|
|
|
|
self.yapWrite { transaction in
|
|
// Threads
|
|
contactThread1.anyInsert(transaction: transaction.asAnyWrite)
|
|
contactThread2.anyInsert(transaction: transaction.asAnyWrite)
|
|
// Attachments
|
|
attachment1.anyInsert(transaction: transaction.asAnyWrite)
|
|
attachment2.anyInsert(transaction: transaction.asAnyWrite)
|
|
// Messages
|
|
outgoingMessage1.anyInsert(transaction: transaction.asAnyWrite)
|
|
outgoingMessage2.anyInsert(transaction: transaction.asAnyWrite)
|
|
outgoingMessage3.anyInsert(transaction: transaction.asAnyWrite)
|
|
}
|
|
|
|
self.yapRead { transaction in
|
|
XCTAssertEqual(2, TSThread.anyCount(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual([contactThread1.uniqueId, contactThread2.uniqueId].sorted(), TSThread.anyAllUniqueIds(transaction: transaction.asAnyRead).sorted())
|
|
XCTAssertEqual(3, TSInteraction.anyCount(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual([outgoingMessage1.uniqueId, outgoingMessage2.uniqueId, outgoingMessage3.uniqueId].sorted(), TSInteraction.anyAllUniqueIds(transaction: transaction.asAnyRead).sorted())
|
|
XCTAssertEqual([outgoingMessage1, outgoingMessage2, outgoingMessage3].compactMap { $0.body }.sorted(), TSInteraction.anyFetchAll(transaction: transaction.asAnyRead).compactMap { ($0 as! TSMessage).body }.sorted())
|
|
XCTAssertEqual(2, TSAttachment.anyCount(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual([attachment1.uniqueId, attachment2.uniqueId].sorted(), TSAttachment.anyAllUniqueIds(transaction: transaction.asAnyRead).sorted())
|
|
}
|
|
self.read { transaction in
|
|
XCTAssertEqual(0, TSThread.anyCount(transaction: transaction))
|
|
XCTAssertEqual(0, TSInteraction.anyCount(transaction: transaction))
|
|
XCTAssertEqual(0, TSAttachment.anyCount(transaction: transaction))
|
|
}
|
|
|
|
let migratorGroups = [
|
|
GRDBMigratorGroup { ydbTransaction in
|
|
return [
|
|
GRDBUnorderedRecordMigrator<TSAttachment>(label: "attachments", ydbTransaction: ydbTransaction, memorySamplerRatio: 0.003),
|
|
GRDBUnorderedRecordMigrator<TSThread>(label: "threads", ydbTransaction: ydbTransaction, memorySamplerRatio: 0.2)
|
|
]
|
|
},
|
|
GRDBMigratorGroup { ydbTransaction in
|
|
return [
|
|
GRDBInteractionMigrator(ydbTransaction: ydbTransaction)
|
|
]
|
|
}
|
|
]
|
|
|
|
try! YDBToGRDBMigration().migrate(migratorGroups: migratorGroups)
|
|
|
|
self.yapRead { transaction in
|
|
XCTAssertEqual(2, TSThread.anyCount(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual([contactThread1.uniqueId, contactThread2.uniqueId].sorted(), TSThread.anyAllUniqueIds(transaction: transaction.asAnyRead).sorted())
|
|
XCTAssertEqual(3, TSInteraction.anyCount(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual([outgoingMessage1.uniqueId, outgoingMessage2.uniqueId, outgoingMessage3.uniqueId].sorted(), TSInteraction.anyAllUniqueIds(transaction: transaction.asAnyRead).sorted())
|
|
XCTAssertEqual([outgoingMessage1, outgoingMessage2, outgoingMessage3].compactMap { $0.body }.sorted(), TSInteraction.anyFetchAll(transaction: transaction.asAnyRead).compactMap { ($0 as! TSMessage).body }.sorted())
|
|
XCTAssertEqual(2, TSAttachment.anyCount(transaction: transaction.asAnyRead))
|
|
XCTAssertEqual([attachment1.uniqueId, attachment2.uniqueId].sorted(), TSAttachment.anyAllUniqueIds(transaction: transaction.asAnyRead).sorted())
|
|
}
|
|
self.read { transaction in
|
|
XCTAssertEqual(2, TSThread.anyCount(transaction: transaction))
|
|
XCTAssertEqual([contactThread1.uniqueId, contactThread2.uniqueId].sorted(), TSThread.anyAllUniqueIds(transaction: transaction).sorted())
|
|
XCTAssertEqual(3, TSInteraction.anyCount(transaction: transaction))
|
|
XCTAssertEqual([outgoingMessage1.uniqueId, outgoingMessage2.uniqueId, outgoingMessage3.uniqueId].sorted(), TSInteraction.anyAllUniqueIds(transaction: transaction).sorted())
|
|
XCTAssertEqual([outgoingMessage1, outgoingMessage2, outgoingMessage3].compactMap { $0.body }.sorted(), TSInteraction.anyFetchAll(transaction: transaction).compactMap { ($0 as! TSMessage).body }.sorted())
|
|
XCTAssertEqual(2, TSAttachment.anyCount(transaction: transaction))
|
|
XCTAssertEqual([attachment1.uniqueId, attachment2.uniqueId].sorted(), TSAttachment.anyAllUniqueIds(transaction: transaction).sorted())
|
|
}
|
|
}
|
|
}
|