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

import Foundation
@testable import SignalServiceKit
import XCTest

class SignalServiceAddressTest: SSKBaseTestSwift {

    var cache: SignalServiceAddressCache {
        return Self.signalServiceAddressCache
    }

    override func setUp() {
        super.setUp()
        // Put setup code here. This method is called before the invocation of each test method in the class.
    }

    override func tearDown() {
        // Put teardown code here. This method is called after the invocation of each test method in the class.
        super.tearDown()
    }

    func test_isEqualPermissive() {
        let phoneNumber1 = "+13213214321"
        let phoneNumber2 = "+13213214322"
        let uuid1 = UUID()
        let uuid2 = UUID()

        // Double match
        XCTAssertEqual(SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1),
                       SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1))

        // Single match works, ignores single missing.
        //
        // SignalServiceAddress's getters use a cache to fill in the blanks.
        cache.updateMapping(uuid: uuid1, phoneNumber: phoneNumber1)

        XCTAssertEqual(SignalServiceAddress(uuid: nil, phoneNumber: phoneNumber1),
                       SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1))
        XCTAssertEqual(SignalServiceAddress(uuid: uuid1),
                       SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1))
        XCTAssertEqual(SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1),
                       SignalServiceAddress(uuid: nil, phoneNumber: phoneNumber1))
        XCTAssertEqual(SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1),
                       SignalServiceAddress(uuid: uuid1))

        // Single match works, ignores double missing.
        XCTAssertEqual(SignalServiceAddress(uuid: nil, phoneNumber: phoneNumber1),
                       SignalServiceAddress(uuid: nil, phoneNumber: phoneNumber1))
        XCTAssertEqual(SignalServiceAddress(uuid: uuid1),
                       SignalServiceAddress(uuid: uuid1))

        // Ignores phone number when UUIDs match.
        XCTAssertEqual(SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1),
                       SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber2))

        // Match fails if no common value.
        XCTAssertEqual(SignalServiceAddress(uuid: uuid1),
                       SignalServiceAddress(uuid: nil, phoneNumber: phoneNumber1))

        // Match fails if either value doesn't match.
        XCTAssertNotEqual(SignalServiceAddress(uuid: uuid1),
                          SignalServiceAddress(uuid: uuid2))
        XCTAssertNotEqual(SignalServiceAddress(uuid: nil, phoneNumber: phoneNumber1),
                          SignalServiceAddress(uuid: nil, phoneNumber: phoneNumber2))
        XCTAssertNotEqual(SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1),
                          SignalServiceAddress(uuid: uuid2))
        XCTAssertNotEqual(SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1),
                          SignalServiceAddress(uuid: nil, phoneNumber: phoneNumber2))
        XCTAssertNotEqual(SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1),
                          SignalServiceAddress(uuid: uuid2, phoneNumber: phoneNumber1))
        XCTAssertNotEqual(SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1),
                          SignalServiceAddress(uuid: uuid2, phoneNumber: phoneNumber2))
    }

    func test_mappingChanges() {
        let uuid1 = UUID()
        let uuid2 = UUID()
        let phoneNumber1 = "+13213214321"
        let phoneNumber2 = "+13213214322"
        let phoneNumber3 = "+13213214323"

        autoreleasepool {
            let address1a = SignalServiceAddress(uuid: uuid1, phoneNumber: nil)
            let address1b = SignalServiceAddress(uuid: uuid1, phoneNumber: nil)
            let address1c = SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1)
            let address2a = SignalServiceAddress(uuid: uuid2, phoneNumber: nil)
            let address2b = SignalServiceAddress(uuid: uuid2, phoneNumber: phoneNumber2)

            // We use the "unresolved" accessors unresolvedUuid(), unresolvedPhoneNumber()
            // to avoid filling in the backing values.

            XCTAssertEqual(address1a.unresolvedUuid, uuid1)
            XCTAssertNil(address1a.unresolvedPhoneNumber)
            XCTAssertEqual(address1b.unresolvedUuid, uuid1)
            XCTAssertNil(address1b.unresolvedPhoneNumber)
            XCTAssertEqual(address1c.unresolvedUuid, uuid1)
            XCTAssertEqual(address1c.unresolvedPhoneNumber, phoneNumber1)
            XCTAssertEqual(address2a.unresolvedUuid, uuid2)
            XCTAssertNil(address2a.unresolvedPhoneNumber)
            XCTAssertEqual(address2b.unresolvedUuid, uuid2)
            XCTAssertEqual(address2b.unresolvedPhoneNumber, phoneNumber2)

            Self.signalServiceAddressCache.updateMapping(uuid: uuid1, phoneNumber: phoneNumber1)

            XCTAssertEqual(address1a.unresolvedUuid, uuid1)
            XCTAssertEqual(address1a.unresolvedPhoneNumber, phoneNumber1)
            XCTAssertEqual(address1b.unresolvedUuid, uuid1)
            XCTAssertEqual(address1b.unresolvedPhoneNumber, phoneNumber1)
            XCTAssertEqual(address1c.unresolvedUuid, uuid1)
            XCTAssertEqual(address1c.unresolvedPhoneNumber, phoneNumber1)
            XCTAssertEqual(address2a.unresolvedUuid, uuid2)
            XCTAssertNil(address2a.unresolvedPhoneNumber)
            XCTAssertEqual(address2b.unresolvedUuid, uuid2)
            XCTAssertEqual(address2b.unresolvedPhoneNumber, phoneNumber2)

            Self.signalServiceAddressCache.updateMapping(uuid: uuid1, phoneNumber: phoneNumber3)

            XCTAssertEqual(address1a.unresolvedUuid, uuid1)
            XCTAssertEqual(address1a.unresolvedPhoneNumber, phoneNumber3)
            XCTAssertEqual(address1b.unresolvedUuid, uuid1)
            XCTAssertEqual(address1b.unresolvedPhoneNumber, phoneNumber3)
            XCTAssertEqual(address1c.unresolvedUuid, uuid1)
            XCTAssertEqual(address1c.unresolvedPhoneNumber, phoneNumber3)
            XCTAssertEqual(address2a.unresolvedUuid, uuid2)
            XCTAssertNil(address2a.unresolvedPhoneNumber)
            XCTAssertEqual(address2b.unresolvedUuid, uuid2)
            XCTAssertEqual(address2b.unresolvedPhoneNumber, phoneNumber2)

            // MARK: - Resolved values

            XCTAssertEqual(address1a.uuid, uuid1)
            XCTAssertEqual(address1a.phoneNumber, phoneNumber3)
            XCTAssertEqual(address1b.uuid, uuid1)
            XCTAssertEqual(address1b.phoneNumber, phoneNumber3)
            XCTAssertEqual(address1c.uuid, uuid1)
            XCTAssertEqual(address1c.phoneNumber, phoneNumber3)
            XCTAssertEqual(address2a.uuid, uuid2)
            XCTAssertNil(address2a.phoneNumber)
            XCTAssertEqual(address2b.uuid, uuid2)
            XCTAssertEqual(address2b.phoneNumber, phoneNumber2)
        }
    }

    private static var mockPhoneNumberCounter = AtomicUInt(13333333333)

    private func mockPhoneNumber() -> String {
        // e.g. "+13213214321"
        return "+\(Self.mockPhoneNumberCounter.increment())"
    }

    func test_mappingChanges1a() {
        let uuid1 = UUID()
        let uuid2 = UUID()
        let phoneNumber1 = mockPhoneNumber()
        let phoneNumber2 = mockPhoneNumber()

        _ = SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1, trustLevel: .high)
        // uuid1, phoneNumber1 -> phoneNumber2
        _ = SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber2, trustLevel: .high)

        XCTAssertEqual(uuid1, SignalServiceAddress(uuid: uuid1).uuid)
        XCTAssertEqual(phoneNumber2, SignalServiceAddress(uuid: uuid1).phoneNumber)
        XCTAssertNil(SignalServiceAddress(phoneNumber: phoneNumber1).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(phoneNumber: phoneNumber1).phoneNumber)
        XCTAssertEqual(uuid1, SignalServiceAddress(phoneNumber: phoneNumber2).uuid)
        XCTAssertEqual(phoneNumber2, SignalServiceAddress(phoneNumber: phoneNumber2).phoneNumber)

        // uuid2, phoneNumber1
        _ = SignalServiceAddress(uuid: uuid2, phoneNumber: phoneNumber1, trustLevel: .high)

        XCTAssertEqual(uuid1, SignalServiceAddress(uuid: uuid1).uuid)
        XCTAssertEqual(phoneNumber2, SignalServiceAddress(uuid: uuid1).phoneNumber)
        XCTAssertEqual(uuid2, SignalServiceAddress(uuid: uuid2).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(uuid: uuid2).phoneNumber)
        XCTAssertEqual(uuid2, SignalServiceAddress(phoneNumber: phoneNumber1).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(phoneNumber: phoneNumber1).phoneNumber)
        XCTAssertEqual(uuid1, SignalServiceAddress(phoneNumber: phoneNumber2).uuid)
        XCTAssertEqual(phoneNumber2, SignalServiceAddress(phoneNumber: phoneNumber2).phoneNumber)
    }

    func test_mappingChanges1b() {
        let uuid1 = UUID()
        let uuid2 = UUID()
        let phoneNumber1 = mockPhoneNumber()
        let phoneNumber2 = mockPhoneNumber()

        _ = SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1, trustLevel: .high)
        // uuid1, phoneNumber1 -> phoneNumber2
        Self.signalServiceAddressCache.updateMapping(uuid: uuid1, phoneNumber: phoneNumber2)

        XCTAssertEqual(uuid1, SignalServiceAddress(uuid: uuid1).uuid)
        XCTAssertEqual(phoneNumber2, SignalServiceAddress(uuid: uuid1).phoneNumber)
        XCTAssertNil(SignalServiceAddress(phoneNumber: phoneNumber1).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(phoneNumber: phoneNumber1).phoneNumber)
        XCTAssertEqual(uuid1, SignalServiceAddress(phoneNumber: phoneNumber2).uuid)
        XCTAssertEqual(phoneNumber2, SignalServiceAddress(phoneNumber: phoneNumber2).phoneNumber)

        // uuid2, phoneNumber1
        Self.signalServiceAddressCache.updateMapping(uuid: uuid2, phoneNumber: phoneNumber1)

        XCTAssertEqual(uuid1, SignalServiceAddress(uuid: uuid1).uuid)
        XCTAssertEqual(phoneNumber2, SignalServiceAddress(uuid: uuid1).phoneNumber)
        XCTAssertEqual(uuid2, SignalServiceAddress(uuid: uuid2).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(uuid: uuid2).phoneNumber)
        XCTAssertEqual(uuid2, SignalServiceAddress(phoneNumber: phoneNumber1).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(phoneNumber: phoneNumber1).phoneNumber)
        XCTAssertEqual(uuid1, SignalServiceAddress(phoneNumber: phoneNumber2).uuid)
        XCTAssertEqual(phoneNumber2, SignalServiceAddress(phoneNumber: phoneNumber2).phoneNumber)
    }

    func test_mappingChanges2a() {
        let uuid1 = UUID()
        let uuid2 = UUID()
        let phoneNumber1 = mockPhoneNumber()
        let phoneNumber2 = mockPhoneNumber()

        _ = SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1, trustLevel: .high)
        // uuid1 -> uuid2, phoneNumber1
        _ = SignalServiceAddress(uuid: uuid2, phoneNumber: phoneNumber1, trustLevel: .high)

        XCTAssertEqual(uuid1, SignalServiceAddress(uuid: uuid1).uuid)
        XCTAssertNil(SignalServiceAddress(uuid: uuid1).phoneNumber)
        XCTAssertEqual(uuid2, SignalServiceAddress(uuid: uuid2).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(uuid: uuid2).phoneNumber)
        XCTAssertEqual(uuid2, SignalServiceAddress(phoneNumber: phoneNumber1).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(phoneNumber: phoneNumber1).phoneNumber)

        // uuid1, phoneNumber2
        _ = SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber2, trustLevel: .high)

        XCTAssertEqual(uuid1, SignalServiceAddress(uuid: uuid1).uuid)
        XCTAssertEqual(phoneNumber2, SignalServiceAddress(uuid: uuid1).phoneNumber)
        XCTAssertEqual(uuid2, SignalServiceAddress(uuid: uuid2).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(uuid: uuid2).phoneNumber)
        XCTAssertEqual(uuid2, SignalServiceAddress(phoneNumber: phoneNumber1).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(phoneNumber: phoneNumber1).phoneNumber)
        XCTAssertEqual(uuid1, SignalServiceAddress(phoneNumber: phoneNumber2).uuid)
        XCTAssertEqual(phoneNumber2, SignalServiceAddress(phoneNumber: phoneNumber2).phoneNumber)
    }

    func test_mappingChanges2b() {
        let uuid1 = UUID()
        let uuid2 = UUID()
        let phoneNumber1 = mockPhoneNumber()
        let phoneNumber2 = mockPhoneNumber()

        _ = SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1, trustLevel: .high)
        // uuid1 -> uuid2, phoneNumber1
        Self.signalServiceAddressCache.updateMapping(uuid: uuid2, phoneNumber: phoneNumber1)

        XCTAssertEqual(uuid1, SignalServiceAddress(uuid: uuid1).uuid)
        XCTAssertNil(SignalServiceAddress(uuid: uuid1).phoneNumber)
        XCTAssertEqual(uuid2, SignalServiceAddress(uuid: uuid2).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(uuid: uuid2).phoneNumber)
        XCTAssertEqual(uuid2, SignalServiceAddress(phoneNumber: phoneNumber1).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(phoneNumber: phoneNumber1).phoneNumber)

        // uuid1, phoneNumber2
        Self.signalServiceAddressCache.updateMapping(uuid: uuid1, phoneNumber: phoneNumber2)

        XCTAssertEqual(uuid1, SignalServiceAddress(uuid: uuid1).uuid)
        XCTAssertEqual(phoneNumber2, SignalServiceAddress(uuid: uuid1).phoneNumber)
        XCTAssertEqual(uuid2, SignalServiceAddress(uuid: uuid2).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(uuid: uuid2).phoneNumber)
        XCTAssertEqual(uuid2, SignalServiceAddress(phoneNumber: phoneNumber1).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(phoneNumber: phoneNumber1).phoneNumber)
        XCTAssertEqual(uuid1, SignalServiceAddress(phoneNumber: phoneNumber2).uuid)
        XCTAssertEqual(phoneNumber2, SignalServiceAddress(phoneNumber: phoneNumber2).phoneNumber)
    }

    func test_mappingChanges3a() {
        let uuid1 = UUID()
        let phoneNumber1 = mockPhoneNumber()

        _ = SignalServiceAddress(uuid: uuid1, phoneNumber: nil, trustLevel: .high)
        _ = SignalServiceAddress(uuid: nil, phoneNumber: phoneNumber1, trustLevel: .high)
        // Associate uuid1, phoneNumber1
        _ = SignalServiceAddress(uuid: uuid1, phoneNumber: phoneNumber1, trustLevel: .high)

        XCTAssertEqual(uuid1, SignalServiceAddress(uuid: uuid1).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(uuid: uuid1).phoneNumber)
        XCTAssertEqual(uuid1, SignalServiceAddress(phoneNumber: phoneNumber1).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(phoneNumber: phoneNumber1).phoneNumber)
    }

    func test_mappingChanges3b() {
        let uuid1 = UUID()
        let phoneNumber1 = mockPhoneNumber()

        _ = SignalServiceAddress(uuid: uuid1, phoneNumber: nil, trustLevel: .high)
        _ = SignalServiceAddress(uuid: nil, phoneNumber: phoneNumber1, trustLevel: .high)
        // Associate uuid1, phoneNumber1
        Self.signalServiceAddressCache.updateMapping(uuid: uuid1, phoneNumber: phoneNumber1)

        XCTAssertEqual(uuid1, SignalServiceAddress(uuid: uuid1).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(uuid: uuid1).phoneNumber)
        XCTAssertEqual(uuid1, SignalServiceAddress(phoneNumber: phoneNumber1).uuid)
        XCTAssertEqual(phoneNumber1, SignalServiceAddress(phoneNumber: phoneNumber1).phoneNumber)
    }
}