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

import LibSignalClient
import XCTest

@testable import SignalServiceKit

class UserProfileTest: SignalBaseTest {
    override func setUp() {
        super.setUp()
        // Create local account.
        databaseStorage.write { tx in
            (DependenciesBridge.shared.registrationStateChangeManager as! RegistrationStateChangeManagerImpl).registerForTests(
                localIdentifiers: .forUnitTests,
                tx: tx.asV2Write
            )
        }
    }

    func testUserProfileForAci() {
        let aci = Aci.randomForTesting()
        let address = SignalServiceAddress(aci)
        write { transaction in
            OWSUserProfile(address: address).anyInsert(transaction: transaction)
        }
        read { transaction in
            let actual = OWSUserProfile.getFor(address, transaction: transaction)
            XCTAssertEqual(actual?.recipientUUID, aci.serviceIdUppercaseString)
        }
        read { transaction in
            let actual = OWSUserProfile.getFor(SignalServiceAddress.randomForTesting(), transaction: transaction)
            XCTAssertNil(actual)
        }
    }

    func testUserProfileForPni() throws {
        let pni = Pni.randomForTesting()
        let address = SignalServiceAddress(pni)
        write { transaction in
            OWSUserProfile(address: address).anyInsert(transaction: transaction)
        }
        read { transaction in
            let actual = OWSUserProfile.getFor(address, transaction: transaction)
            XCTAssertEqual(actual?.recipientUUID, pni.serviceIdUppercaseString)
        }
        read { transaction in
            let actual = OWSUserProfile.getFor(SignalServiceAddress.randomForTesting(), transaction: transaction)
            XCTAssertNil(actual)
        }
    }

    func testUserProfilesForServiceIds() {
        let addresses = [SignalServiceAddress.randomForTesting(),
                         SignalServiceAddress.randomForTesting()]
        let profiles = addresses.map { OWSUserProfile(address: $0) }
        write { transaction in
            for profile in profiles {
                profile.anyInsert(transaction: transaction)
            }
        }
        read { transaction in
            let bogusAddresses = [SignalServiceAddress.randomForTesting()]
            let actual = SignalServiceKit.OWSUserProfile.getFor(keys: addresses + bogusAddresses,
                                                                transaction: transaction)
            let expected = profiles + [nil]
            XCTAssertEqual(actual.map { $0?.recipientUUID },
                           expected.map { $0?.recipientUUID })
        }
    }

    func testUserProfileForPhoneNumber() {
        let address = SignalServiceAddress(phoneNumber: "+17035550000")
        write { transaction in
            OWSUserProfile(address: address).anyInsert(transaction: transaction)
        }
        read { transaction in
            let actual = OWSUserProfile.getFor(address, transaction: transaction)
            XCTAssertEqual(actual?.recipientPhoneNumber, "+17035550000")
        }
        read { transaction in
            let actual = OWSUserProfile.getFor(SignalServiceAddress(phoneNumber: "+17035550001"), transaction: transaction)
            XCTAssertNil(actual)
        }
    }

    func testUserProfilesForPhoneNumbers() {
        let addresses = [SignalServiceAddress(phoneNumber: "+17035550000"),
                         SignalServiceAddress(phoneNumber: "+17035550001")]
        let profiles = addresses.map { OWSUserProfile(address: $0) }
        write { transaction in
            for profile in profiles {
                profile.anyInsert(transaction: transaction)
            }
        }
        read { transaction in
            let bogusAddresses = [SignalServiceAddress(phoneNumber: "+17035550002")]
            let actual = SignalServiceKit.OWSUserProfile.getFor(keys: addresses + bogusAddresses,
                                                                transaction: transaction)
            let expected = profiles + [nil]
            XCTAssertEqual(actual.map { $0?.recipientPhoneNumber },
                           expected.map { $0?.recipientPhoneNumber })
        }
    }
}