Signal-iOS/SignalServiceKit/tests/Messages/Interactions/TSMessageTest.swift

//
//  Copyright (c) 2022 Open Whisper Systems. All rights reserved.
//

@testable import SignalServiceKit
import XCTest

class TSMessageTest: SSKBaseTestSwift {
    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 = builder.buildWithSneakyTransaction()
        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 = builder.buildWithSneakyTransaction()
        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 = builder.buildWithSneakyTransaction()

            XCTAssert(message.canBeRemotelyDeleted)
        }

        do {
            let builder = TSIncomingMessageBuilder.incomingMessageBuilder(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 = builder.buildWithSneakyTransaction()
            self.databaseStorage.write { transaction in
                message.updateWithRemotelyDeletedAndRemoveRenderableContent(with: transaction)
            }

            XCTAssertFalse(message.canBeRemotelyDeleted)
        }

        do {
            let builder = TSOutgoingMessageBuilder.outgoingMessageBuilder(thread: self.thread)
            builder.timestamp = now - kMinuteInMs
            builder.giftBadge = OWSGiftBadge()
            let message = builder.buildWithSneakyTransaction()

            XCTAssertFalse(message.canBeRemotelyDeleted)
        }

        do {
            let builder = TSOutgoingMessageBuilder.outgoingMessageBuilder(thread: self.thread)
            builder.timestamp = now + kMinuteInMs
            let message = builder.buildWithSneakyTransaction()

            XCTAssertTrue(message.canBeRemotelyDeleted)
        }

        do {
            let builder = TSOutgoingMessageBuilder.outgoingMessageBuilder(thread: self.thread)
            builder.timestamp = now + (4 * kHourInMs)
            let message = builder.buildWithSneakyTransaction()

            XCTAssertTrue(message.canBeRemotelyDeleted)
        }

        do {
            let builder = TSOutgoingMessageBuilder.outgoingMessageBuilder(thread: self.thread)
            builder.timestamp = now - (4 * kHourInMs)
            let message = builder.buildWithSneakyTransaction()

            XCTAssertFalse(message.canBeRemotelyDeleted)
        }
    }
}