//
// Copyright 2023 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
//

import LibSignalClient
import XCTest

@testable import SignalServiceKit

final class RecipientStateMergerTest: XCTestCase {
    private var mockDB: InMemoryDB!
    private var _signalServiceAddressCache: SignalServiceAddressCache!
    private var recipientDatabaseTable: RecipientDatabaseTable!
    private var recipientStateMerger: RecipientStateMerger!

    override func setUp() {
        super.setUp()

        mockDB = InMemoryDB()
        _signalServiceAddressCache = SignalServiceAddressCache()
        recipientDatabaseTable = RecipientDatabaseTable()
        recipientStateMerger = RecipientStateMerger(
            recipientDatabaseTable: recipientDatabaseTable,
            signalServiceAddressCache: _signalServiceAddressCache
        )
    }

    func testNormalize() {
        let aci1 = Aci.constantForTesting("00000000-0000-4000-8000-0000000000a1")
        let pni1 = Pni.constantForTesting("PNI:00000000-0000-4000-8000-0000000000b1")
        let aci2 = Aci.constantForTesting("00000000-0000-4000-8000-0000000000a2")
        let pni3 = Pni.constantForTesting("PNI:00000000-0000-4000-8000-0000000000b3")
        let aci4 = Aci.constantForTesting("00000000-0000-4000-8000-0000000000a4")
        let pni4 = Pni.constantForTesting("PNI:00000000-0000-4000-8000-0000000000b4")

        mockDB.write { tx in
            recipientDatabaseTable.insertRecipient(SignalRecipient(aci: aci1, pni: pni1, phoneNumber: nil), transaction: tx)
            recipientDatabaseTable.insertRecipient(SignalRecipient(aci: aci4, pni: pni4, phoneNumber: nil), transaction: tx)
        }

        var recipientStates: [SignalServiceAddress: TSOutgoingMessageRecipientState]? = [
            makeAddress(pni1): makeState(deliveryTimestamp: 1),
            makeAddress(aci2): makeState(deliveryTimestamp: 2),
            makeAddress(pni3): makeState(deliveryTimestamp: 3),
            makeAddress(aci4): makeState(deliveryTimestamp: 4),
            makeAddress(pni4): makeState(deliveryTimestamp: 5)
        ]
        mockDB.read { tx in
            recipientStateMerger.normalize(&recipientStates, tx: tx)
        }

        func assertDeliveryTimestamp(_ serviceId: ServiceId, equalTo: UInt64) {
            let recipientState = recipientStates!.removeValue(forKey: makeAddress(serviceId))
            XCTAssertEqual(recipientState?.status, .delivered)
            XCTAssertEqual(recipientState?.statusTimestamp, equalTo)
        }

        assertDeliveryTimestamp(aci1, equalTo: 1)
        assertDeliveryTimestamp(aci2, equalTo: 2)
        assertDeliveryTimestamp(pni3, equalTo: 3)
        assertDeliveryTimestamp(aci4, equalTo: 4)
        XCTAssertEqual(recipientStates, [:])
    }

    private func makeAddress(_ serviceId: ServiceId) -> SignalServiceAddress {
        return SignalServiceAddress(
            serviceId: serviceId,
            phoneNumber: nil,
            cache: _signalServiceAddressCache
        )
    }

    private func makeState(deliveryTimestamp: UInt64) -> TSOutgoingMessageRecipientState {
        return TSOutgoingMessageRecipientState(
            status: .delivered,
            statusTimestamp: deliveryTimestamp,
            wasSentByUD: false,
            errorCode: nil
        )
    }
}