89 lines
2.7 KiB
Swift
89 lines
2.7 KiB
Swift
//
|
|
// Copyright 2019 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
import Foundation
|
|
import XCTest
|
|
|
|
@testable import SignalServiceKit
|
|
|
|
class ViewOnceMessagesTest: SSKBaseTest {
|
|
override func setUp() {
|
|
super.setUp()
|
|
|
|
SSKEnvironment.shared.databaseStorageRef.write { tx in
|
|
(DependenciesBridge.shared.registrationStateChangeManager as! RegistrationStateChangeManagerImpl).registerForTests(
|
|
localIdentifiers: .forUnitTests,
|
|
tx: tx.asV2Write
|
|
)
|
|
}
|
|
}
|
|
|
|
// MARK: -
|
|
|
|
func test_expiration() {
|
|
|
|
let messageCount = { () -> Int in
|
|
return SSKEnvironment.shared.databaseStorageRef.read { transaction in
|
|
return TSInteraction.anyFetchAll(transaction: transaction).count
|
|
}
|
|
}
|
|
let latestCopy = { (message: TSMessage) -> TSMessage in
|
|
let uniqueId = message.uniqueId
|
|
return SSKEnvironment.shared.databaseStorageRef.read { transaction in
|
|
return TSMessage.anyFetch(uniqueId: uniqueId, transaction: transaction) as! TSMessage
|
|
}
|
|
}
|
|
|
|
// Factory 1 builds messages that are not view-once messages.
|
|
let factory1 = IncomingMessageFactory()
|
|
factory1.isViewOnceMessageBuilder = {
|
|
return false
|
|
}
|
|
|
|
self.write { transaction in
|
|
_ = factory1.create(transaction: transaction)
|
|
}
|
|
|
|
XCTAssertEqual(1, messageCount())
|
|
|
|
// Factory 2 builds view-once messages.
|
|
let factory2 = IncomingMessageFactory()
|
|
factory2.isViewOnceMessageBuilder = {
|
|
return true
|
|
}
|
|
|
|
var viewOnceMessage: TSMessage?
|
|
self.write { transaction in
|
|
viewOnceMessage = factory2.create(transaction: transaction)
|
|
}
|
|
|
|
XCTAssertEqual(2, messageCount())
|
|
|
|
guard let message = viewOnceMessage else {
|
|
XCTFail("Missing message.")
|
|
return
|
|
}
|
|
|
|
XCTAssertTrue(message.isViewOnceMessage)
|
|
XCTAssertFalse(message.isViewOnceComplete)
|
|
|
|
XCTAssertTrue(latestCopy(message).isViewOnceMessage)
|
|
XCTAssertFalse(latestCopy(message).isViewOnceComplete)
|
|
|
|
self.write { transaction in
|
|
ViewOnceMessages.markAsComplete(message: message,
|
|
sendSyncMessages: false,
|
|
transaction: transaction)
|
|
}
|
|
|
|
XCTAssertTrue(message.isViewOnceMessage)
|
|
XCTAssertTrue(message.isViewOnceComplete)
|
|
|
|
XCTAssertTrue(latestCopy(message).isViewOnceMessage)
|
|
XCTAssertTrue(latestCopy(message).isViewOnceComplete)
|
|
|
|
XCTAssertEqual(2, messageCount())
|
|
}
|
|
}
|