105 lines
3.7 KiB
Swift
105 lines
3.7 KiB
Swift
//
|
|
// Copyright 2022 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
@testable import SignalServiceKit
|
|
import XCTest
|
|
|
|
class TSMessageTest: SSKBaseTest {
|
|
private var thread: TSThread!
|
|
|
|
override func setUp() {
|
|
super.setUp()
|
|
|
|
self.thread = TSContactThread.getOrCreateThread(contactAddress: SignalServiceAddress(phoneNumber: "fake-thread-id"))
|
|
}
|
|
|
|
func testExpiresAtWithoutStartedTimer() {
|
|
let builder = TSOutgoingMessageBuilder.outgoingMessageBuilder(thread: self.thread, messageBody: "foo")
|
|
builder.timestamp = 1
|
|
builder.expiresInSeconds = 100
|
|
|
|
let message = databaseStorage.read { builder.build(transaction: $0) }
|
|
XCTAssertEqual(0, message.expiresAt)
|
|
}
|
|
|
|
func testExpiresAtWithStartedTimer() {
|
|
let now = Date.ows_millisecondTimestamp()
|
|
let expirationSeconds: UInt32 = 10
|
|
|
|
let builder = TSOutgoingMessageBuilder.outgoingMessageBuilder(thread: self.thread, messageBody: "foo")
|
|
builder.timestamp = 1
|
|
builder.expiresInSeconds = expirationSeconds
|
|
builder.expireStartedAt = now
|
|
|
|
let message = databaseStorage.read { builder.build(transaction: $0) }
|
|
XCTAssertEqual(now + UInt64(expirationSeconds * 1000), message.expiresAt)
|
|
}
|
|
|
|
func testCanBeRemotelyDeleted() {
|
|
let now = Date.ows_millisecondTimestamp()
|
|
|
|
do {
|
|
let builder = TSOutgoingMessageBuilder.outgoingMessageBuilder(thread: self.thread)
|
|
builder.timestamp = now - kMinuteInMs
|
|
let message = databaseStorage.read { builder.build(transaction: $0) }
|
|
|
|
XCTAssert(message.canBeRemotelyDeleted)
|
|
}
|
|
|
|
do {
|
|
let builder: TSIncomingMessageBuilder = .withDefaultValues(thread: self.thread)
|
|
builder.timestamp = now - kMinuteInMs
|
|
let message = builder.build()
|
|
|
|
XCTAssertFalse(message.canBeRemotelyDeleted)
|
|
}
|
|
|
|
do {
|
|
let builder = TSOutgoingMessageBuilder.outgoingMessageBuilder(thread: self.thread)
|
|
builder.timestamp = now - kMinuteInMs
|
|
let message = databaseStorage.read { builder.build(transaction: $0) }
|
|
self.databaseStorage.write { transaction in
|
|
message.anyInsert(transaction: transaction)
|
|
message.updateWithRemotelyDeletedAndRemoveRenderableContent(with: transaction)
|
|
}
|
|
|
|
XCTAssertFalse(message.canBeRemotelyDeleted)
|
|
}
|
|
|
|
do {
|
|
let builder = TSOutgoingMessageBuilder.outgoingMessageBuilder(thread: self.thread)
|
|
builder.timestamp = now - kMinuteInMs
|
|
builder.giftBadge = OWSGiftBadge()
|
|
let message = databaseStorage.read { builder.build(transaction: $0) }
|
|
|
|
XCTAssertFalse(message.canBeRemotelyDeleted)
|
|
}
|
|
|
|
do {
|
|
let builder = TSOutgoingMessageBuilder.outgoingMessageBuilder(thread: self.thread)
|
|
builder.timestamp = now + kMinuteInMs
|
|
let message = databaseStorage.read { builder.build(transaction: $0) }
|
|
|
|
XCTAssertTrue(message.canBeRemotelyDeleted)
|
|
}
|
|
|
|
do {
|
|
let builder = TSOutgoingMessageBuilder.outgoingMessageBuilder(thread: self.thread)
|
|
builder.timestamp = now + (25 * kHourInMs)
|
|
let message = databaseStorage.read { builder.build(transaction: $0) }
|
|
|
|
XCTAssertTrue(message.canBeRemotelyDeleted)
|
|
}
|
|
|
|
do {
|
|
let builder = TSOutgoingMessageBuilder.outgoingMessageBuilder(thread: self.thread)
|
|
builder.timestamp = now - (25 * kHourInMs)
|
|
let message = databaseStorage.read { builder.build(transaction: $0) }
|
|
|
|
XCTAssertFalse(message.canBeRemotelyDeleted)
|
|
}
|
|
}
|
|
}
|