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

import LibSignalClient
import XCTest

@testable import SignalServiceKit

final class MessageBackupSessionSwitchoverChatUpdateTest: MessageBackupIntegrationTestCase {
    func testSessionSwitchoverChatUpdates() async throws {
        let expectedAci = Aci.parseFrom(aciString: "5F8C568D-0119-47BD-81AA-BB87C9B71995")!

        try await runTest(backupName: "session-switchover-chat-update-message") { sdsTx, tx in
            let allInteractions = try deps.interactionStore.fetchAllInteractions(tx: tx)
            XCTAssertEqual(allInteractions.count, 1)

            guard
                let infoMessage = allInteractions.first! as? TSInfoMessage,
                let sessionSwitchoverPhoneNumber = infoMessage.sessionSwitchoverPhoneNumber
            else {
                XCTFail("Unexpectedly missing session switchover message and properties!")
                return
            }

            guard
                let switchedOverThread = deps.threadStore.fetchThreadForInteraction(infoMessage, tx: tx) as? TSContactThread,
                let switchedOverThreadAci = switchedOverThread.contactAddress.aci
            else {
                XCTFail("Unexpectedly missing merged thread and properties!")
                return
            }

            XCTAssertEqual(infoMessage.messageType, .sessionSwitchover)
            XCTAssertEqual(sessionSwitchoverPhoneNumber, "+17735550199")
            XCTAssertEqual(switchedOverThreadAci, expectedAci)
        }
    }
}

private extension InteractionStore {
    func fetchAllInteractions(tx: any DBReadTransaction) throws -> [TSInteraction] {
        var results = [TSInteraction]()
        try enumerateAllInteractions(tx: tx) { interaction in
            results.append(interaction)
            return true
        }
        return results
    }
}