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

import XCTest
import SignalServiceKit

class PhoneNumberUtilTestSwift: SSKBaseTestSwift {

    func test_callingCodeFromCountryCode() {
        XCTAssertEqual("+1", phoneNumberUtil.callingCode(fromCountryCode: "US"))
        XCTAssertEqual("+598", phoneNumberUtil.callingCode(fromCountryCode: "UY"))
        XCTAssertEqual("+0", phoneNumberUtil.callingCode(fromCountryCode: "QG"))
        XCTAssertEqual("+0", phoneNumberUtil.callingCode(fromCountryCode: ""))
        XCTAssertEqual("+0", phoneNumberUtil.callingCode(fromCountryCode: "+"))
        XCTAssertEqual("+0", phoneNumberUtil.callingCode(fromCountryCode: "9"))
        XCTAssertEqual("+1", phoneNumberUtil.callingCode(fromCountryCode: "US "))
    }

    func test_countryCodesFromCallingCode() {
        // Order matters here.
        XCTAssertEqual(["US", "CA", "DO", "PR", "JM", "TT", "BS", "BB", "LC", "GU", "VI", "GD", "VC", "AG", "DM", "BM", "AS", "MP", "KN", "KY", "SX", "VG", "TC", "AI", "MS", "UM"], phoneNumberUtil.countryCodes(fromCallingCode: "+1"))
        XCTAssertEqual(["GB", "JE", "IM", "GG"], phoneNumberUtil.countryCodes(fromCallingCode: "+44"))
        XCTAssertEqual(["UY"], phoneNumberUtil.countryCodes(fromCallingCode: "+598"))
        XCTAssertEqual([], phoneNumberUtil.countryCodes(fromCallingCode: "+7945"))
        XCTAssertEqual([], phoneNumberUtil.countryCodes(fromCallingCode: "+"))
        XCTAssertEqual([], phoneNumberUtil.countryCodes(fromCallingCode: ""))
        XCTAssertEqual([], phoneNumberUtil.countryCodes(fromCallingCode: " "))
        XCTAssertEqual([], phoneNumberUtil.countryCodes(fromCallingCode: "a"))
        XCTAssertEqual([], phoneNumberUtil.countryCodes(fromCallingCode: "++598"))
        XCTAssertEqual([], phoneNumberUtil.countryCodes(fromCallingCode: "+1 "))
        XCTAssertEqual([], phoneNumberUtil.countryCodes(fromCallingCode: " +1"))
    }

    func test_parse() {
        XCTAssertEqual(NSNumber(value: 1), try! phoneNumberUtil.parse("+13334445555", defaultRegion: "US").countryCode)
        XCTAssertEqual(NSNumber(value: 1), try! phoneNumberUtil.parse("+13334445555", defaultRegion: "GB").countryCode)
        XCTAssertEqual(NSNumber(value: 1), try! phoneNumberUtil.parse("+13334445555", defaultRegion: "UY").countryCode)
        XCTAssertEqual(NSNumber(value: 44), try! phoneNumberUtil.parse("+441752395464", defaultRegion: "US").countryCode)
        XCTAssertEqual(NSNumber(value: 44), try! phoneNumberUtil.parse("+441752395464", defaultRegion: "GB").countryCode)
        XCTAssertEqual(NSNumber(value: 44), try! phoneNumberUtil.parse("+441752395464", defaultRegion: "UY").countryCode)
        // Invalid regions.
        XCTAssertEqual(NSNumber(value: 1), try! phoneNumberUtil.parse("+13334445555", defaultRegion: "UK").countryCode)
        XCTAssertEqual(NSNumber(value: 44), try! phoneNumberUtil.parse("+441752395464", defaultRegion: "UK").countryCode)
        XCTAssertEqual(NSNumber(value: 1), try! phoneNumberUtil.parse("+13334445555", defaultRegion: "ZQ").countryCode)
        XCTAssertEqual(NSNumber(value: 1), try! phoneNumberUtil.parse("+13334445555", defaultRegion: "").countryCode)
        XCTAssertEqual(NSNumber(value: 1), try! phoneNumberUtil.parse("+13334445555", defaultRegion: "99").countryCode)
        XCTAssertEqual(NSNumber(value: 1), try! phoneNumberUtil.parse("+13334445555", defaultRegion: "+").countryCode)
        XCTAssertEqual(NSNumber(value: 1), try! phoneNumberUtil.parse("+13334445555", defaultRegion: " ").countryCode)
        // Invalid phone numbers.
        XCTAssertEqual(NSNumber(value: 1), try! phoneNumberUtil.parse("+1333444555", defaultRegion: "UY").countryCode)
        XCTAssertEqual(NSNumber(value: 44), try! phoneNumberUtil.parse("+44175239546", defaultRegion: "UY").countryCode)
        XCTAssertEqual(NSNumber(value: 44), try! phoneNumberUtil.parse("+441752395468 ", defaultRegion: "UY").countryCode)
        XCTAssertEqual(NSNumber(value: 1), try! phoneNumberUtil.parse("++13334445555 ", defaultRegion: "UY").countryCode)

        do {
            _ = try phoneNumberUtil.parse("+9764", defaultRegion: "US")
            XCTFail("Missing expected error.")
        } catch {
            // Error is expected.
        }
        do {
            _ = try phoneNumberUtil.parse("", defaultRegion: "US")
            XCTFail("Missing expected error.")
        } catch {
            // Error is expected.
        }
        do {
            _ = try phoneNumberUtil.parse(" ", defaultRegion: "US")
            XCTFail("Missing expected error.")
        } catch {
            // Error is expected.
        }
        do {
            _ = try phoneNumberUtil.parse("a", defaultRegion: "US")
            XCTFail("Missing expected error.")
        } catch {
            // Error is expected.
        }
        do {
            _ = try phoneNumberUtil.parse("+", defaultRegion: "US")
            XCTFail("Missing expected error.")
        } catch {
            // Error is expected.
        }
        do {
            _ = try phoneNumberUtil.parse("9876543210987654321098765432109876543210", defaultRegion: "US")
            XCTFail("Missing expected error.")
        } catch {
            // Error is expected.
        }
        do {
            _ = try phoneNumberUtil.parse("+9876543210987654321098765432109876543210", defaultRegion: "US")
            XCTFail("Missing expected error.")
        } catch {
            // Error is expected.
        }
    }

    func test_format() {
        let phoneNumber1 = try! phoneNumberUtil.parse("+13334445555", defaultRegion: "US")
        let phoneNumber2 = try! phoneNumberUtil.parse("+441752395464", defaultRegion: "US")

        XCTAssertEqual("+13334445555", try! phoneNumberUtil.format(phoneNumber1, numberFormat: .E164))
        XCTAssertEqual("(333) 444-5555", try! phoneNumberUtil.format(phoneNumber1, numberFormat: .NATIONAL))
        XCTAssertEqual("+1 333-444-5555", try! phoneNumberUtil.format(phoneNumber1, numberFormat: .INTERNATIONAL))
        XCTAssertEqual("tel:+1-333-444-5555", try! phoneNumberUtil.format(phoneNumber1, numberFormat: .RFC3966))

        XCTAssertEqual("+441752395464", try! phoneNumberUtil.format(phoneNumber2, numberFormat: .E164))
        XCTAssertEqual("01752 395464", try! phoneNumberUtil.format(phoneNumber2, numberFormat: .NATIONAL))
        XCTAssertEqual("+44 1752 395464", try! phoneNumberUtil.format(phoneNumber2, numberFormat: .INTERNATIONAL))
        XCTAssertEqual("tel:+44-1752-395464", try! phoneNumberUtil.format(phoneNumber2, numberFormat: .RFC3966))
    }

    func test_examplePhoneNumberForCountryCode() {
        XCTAssertEqual("+12015550123", phoneNumberUtil.examplePhoneNumber(forCountryCode: "US"))
        XCTAssertEqual("+447400123456", phoneNumberUtil.examplePhoneNumber(forCountryCode: "GB"))
        XCTAssertEqual("+59894231234", phoneNumberUtil.examplePhoneNumber(forCountryCode: "UY"))
        XCTAssertNil(phoneNumberUtil.examplePhoneNumber(forCountryCode: "+1"))
        XCTAssertNil(phoneNumberUtil.examplePhoneNumber(forCountryCode: "44"))
        XCTAssertNil(phoneNumberUtil.examplePhoneNumber(forCountryCode: " "))
        XCTAssertNil(phoneNumberUtil.examplePhoneNumber(forCountryCode: ""))
        XCTAssertNil(phoneNumberUtil.examplePhoneNumber(forCountryCode: "UK "))
        XCTAssertNil(phoneNumberUtil.examplePhoneNumber(forCountryCode: "UKK"))
    }

    func test_isPossibleNumber() {
        let phoneNumber1 = try! phoneNumberUtil.parse("+13334445555", defaultRegion: "US")
        let phoneNumber2 = try! phoneNumberUtil.parse("+441752395464", defaultRegion: "US")
        // Invalid numbers.
        let phoneNumber3 = try! phoneNumberUtil.parse("+44175239546", defaultRegion: "US")
        let phoneNumber4 = try! phoneNumberUtil.parse("44", defaultRegion: "US")

        XCTAssertEqual(true, phoneNumberUtil.isPossibleNumber(phoneNumber1))
        XCTAssertEqual(true, phoneNumberUtil.isPossibleNumber(phoneNumber2))
        XCTAssertEqual(true, phoneNumberUtil.isPossibleNumber(phoneNumber3))
        XCTAssertEqual(false, phoneNumberUtil.isPossibleNumber(phoneNumber4))
    }

    func test_isValidNumber() {
        let phoneNumber1 = try! phoneNumberUtil.parse("+12125556789", defaultRegion: "US")
        let phoneNumber2 = try! phoneNumberUtil.parse("+441752395464", defaultRegion: "US")
        let phoneNumber3 = try! phoneNumberUtil.parse("+12125556789", defaultRegion: "GB")
        let phoneNumber4 = try! phoneNumberUtil.parse("+441752395464", defaultRegion: "GB")
        // Invalid numbers.
        let phoneNumber5 = try! phoneNumberUtil.parse("+44175239546", defaultRegion: "US")
        let phoneNumber6 = try! phoneNumberUtil.parse("44", defaultRegion: "US")
        let phoneNumber7 = try! phoneNumberUtil.parse("+13334445555", defaultRegion: "US")
        let phoneNumber8 = try! phoneNumberUtil.parse("+44175239546", defaultRegion: "GB")
        let phoneNumber9 = try! phoneNumberUtil.parse("44", defaultRegion: "GB")
        let phoneNumber10 = try! phoneNumberUtil.parse("+13334445555", defaultRegion: "GB")

        XCTAssertEqual(true, phoneNumberUtil.isValidNumber(phoneNumber1))
        XCTAssertEqual(true, phoneNumberUtil.isValidNumber(phoneNumber2))
        XCTAssertEqual(true, phoneNumberUtil.isValidNumber(phoneNumber3))
        XCTAssertEqual(true, phoneNumberUtil.isValidNumber(phoneNumber4))
        XCTAssertEqual(false, phoneNumberUtil.isValidNumber(phoneNumber5))
        XCTAssertEqual(false, phoneNumberUtil.isValidNumber(phoneNumber6))
        XCTAssertEqual(false, phoneNumberUtil.isValidNumber(phoneNumber7))
        XCTAssertEqual(false, phoneNumberUtil.isValidNumber(phoneNumber8))
        XCTAssertEqual(false, phoneNumberUtil.isValidNumber(phoneNumber9))
        XCTAssertEqual(false, phoneNumberUtil.isValidNumber(phoneNumber10))
    }

    func test_countryCodeByCarrier() {
        // This test might fail depending on how the test device/simulator is configured.
        XCTAssertEqual("ZZ", phoneNumberUtil.countryCodeByCarrier())
    }

    func test_getRegionCodeForCountryCode() {
        XCTAssertEqual("US", phoneNumberUtil.getRegionCodeForCountryCode(NSNumber(value: 1)))
        XCTAssertEqual("GB", phoneNumberUtil.getRegionCodeForCountryCode(NSNumber(value: 44)))
        XCTAssertEqual("UY", phoneNumberUtil.getRegionCodeForCountryCode(NSNumber(value: 598)))
        XCTAssertEqual("ZZ", phoneNumberUtil.getRegionCodeForCountryCode(NSNumber(value: 0)))
        XCTAssertEqual("ZZ", phoneNumberUtil.getRegionCodeForCountryCode(NSNumber(value: 99999)))
        XCTAssertEqual("ZZ", phoneNumberUtil.getRegionCodeForCountryCode(NSNumber(value: -1)))
    }

    func test_getCountryCodeForRegion() {
        XCTAssertEqual(NSNumber(value: 1), phoneNumberUtil.getCountryCode(forRegion: "US"))
        XCTAssertEqual(NSNumber(value: 44), phoneNumberUtil.getCountryCode(forRegion: "GB"))
        XCTAssertEqual(NSNumber(value: 598), phoneNumberUtil.getCountryCode(forRegion: "UY"))
        // Invalid regions.
        XCTAssertEqual(NSNumber(value: 0), phoneNumberUtil.getCountryCode(forRegion: "UK"))
        XCTAssertEqual(NSNumber(value: 1), phoneNumberUtil.getCountryCode(forRegion: "US "))
        XCTAssertEqual(NSNumber(value: 0), phoneNumberUtil.getCountryCode(forRegion: " "))
        XCTAssertEqual(NSNumber(value: 0), phoneNumberUtil.getCountryCode(forRegion: ""))
        XCTAssertEqual(NSNumber(value: 0), phoneNumberUtil.getCountryCode(forRegion: "+1"))
        XCTAssertEqual(NSNumber(value: 0), phoneNumberUtil.getCountryCode(forRegion: "ZZ"))
        XCTAssertEqual(NSNumber(value: 0), phoneNumberUtil.getCountryCode(forRegion: "+"))
        XCTAssertEqual(NSNumber(value: 0), phoneNumberUtil.getCountryCode(forRegion: "ZQ"))
    }
}