TM-SGNL-iOS/SignalServiceKit/tests/Contacts/PhoneNumberUtilTest.swift
TeleMessage developers dde0620daf initial commit
2025-05-03 12:28:28 -07:00

138 lines
10 KiB
Swift

//
// Copyright 2021 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
//
import XCTest
@testable import SignalServiceKit
class PhoneNumberUtilTestSwift: XCTestCase {
private var phoneNumberUtilRef: PhoneNumberUtil!
override func setUp() {
super.setUp()
self.phoneNumberUtilRef = PhoneNumberUtil()
}
func testCountryCodeForParsing() {
let alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
for ch1 in alphabet {
for ch2 in alphabet {
_ = self.phoneNumberUtilRef.countryCodeForParsing(fromCountryCode: String(ch1) + String(ch2))
}
}
}
func test_callingCodeFromCountryCode() {
func plusPrefixedCallingCode(fromCountryCode countryCode: String) -> String {
let callingCode = phoneNumberUtilRef.getCallingCode(forRegion: countryCode)
return "+\(callingCode)"
}
XCTAssertEqual("+1", plusPrefixedCallingCode(fromCountryCode: "US"))
XCTAssertEqual("+44", plusPrefixedCallingCode(fromCountryCode: "GB"))
XCTAssertEqual("+598", plusPrefixedCallingCode(fromCountryCode: "UY"))
XCTAssertEqual("+0", plusPrefixedCallingCode(fromCountryCode: "QG"))
XCTAssertEqual("+0", plusPrefixedCallingCode(fromCountryCode: "EK"))
XCTAssertEqual("+0", plusPrefixedCallingCode(fromCountryCode: "ZZZ"))
XCTAssertEqual("+0", plusPrefixedCallingCode(fromCountryCode: ""))
XCTAssertEqual("+0", plusPrefixedCallingCode(fromCountryCode: "+"))
XCTAssertEqual("+0", plusPrefixedCallingCode(fromCountryCode: "9"))
XCTAssertEqual("+1", plusPrefixedCallingCode(fromCountryCode: "US "))
}
func test_examplePhoneNumberForCountryCode() {
XCTAssertEqual("(201) 555-0123", phoneNumberUtilRef.exampleNationalNumber(forCountryCode: "US"))
XCTAssertEqual("07400 123456", phoneNumberUtilRef.exampleNationalNumber(forCountryCode: "GB"))
XCTAssertEqual("094 231 234", phoneNumberUtilRef.exampleNationalNumber(forCountryCode: "UY"))
}
func test_getRegionCodeForCountryCode() {
XCTAssertEqual("US", phoneNumberUtilRef.getRegionCodeForCallingCode(1))
XCTAssertEqual("GB", phoneNumberUtilRef.getRegionCodeForCallingCode(44))
XCTAssertEqual("UY", phoneNumberUtilRef.getRegionCodeForCallingCode(598))
XCTAssertEqual("ZZ", phoneNumberUtilRef.getRegionCodeForCallingCode(0))
XCTAssertEqual("ZZ", phoneNumberUtilRef.getRegionCodeForCallingCode(99999))
XCTAssertEqual("ZZ", phoneNumberUtilRef.getRegionCodeForCallingCode(-1))
}
func test_getCallingCodeForRegion() {
XCTAssertEqual(1, phoneNumberUtilRef.getCallingCode(forRegion: "US"))
XCTAssertEqual(44, phoneNumberUtilRef.getCallingCode(forRegion: "GB"))
XCTAssertEqual(598, phoneNumberUtilRef.getCallingCode(forRegion: "UY"))
// Invalid regions.
XCTAssertEqual(0, phoneNumberUtilRef.getCallingCode(forRegion: "UK"))
XCTAssertEqual(1, phoneNumberUtilRef.getCallingCode(forRegion: "US "))
XCTAssertEqual(0, phoneNumberUtilRef.getCallingCode(forRegion: " "))
XCTAssertEqual(0, phoneNumberUtilRef.getCallingCode(forRegion: ""))
XCTAssertEqual(0, phoneNumberUtilRef.getCallingCode(forRegion: "+1"))
XCTAssertEqual(0, phoneNumberUtilRef.getCallingCode(forRegion: "ZZ"))
XCTAssertEqual(0, phoneNumberUtilRef.getCallingCode(forRegion: "+"))
XCTAssertEqual(0, phoneNumberUtilRef.getCallingCode(forRegion: "ZQ"))
}
func testTranslateCursorPosition() {
XCTAssertEqual(0, PhoneNumberUtil.translateCursorPosition(0, from: "", to: "", stickingRightward: true))
XCTAssertEqual(0, PhoneNumberUtil.translateCursorPosition(0, from: "", to: "", stickingRightward: true))
XCTAssertEqual(0, PhoneNumberUtil.translateCursorPosition(0, from: "12", to: "1", stickingRightward: true))
XCTAssertEqual(1, PhoneNumberUtil.translateCursorPosition(1, from: "12", to: "1", stickingRightward: true))
XCTAssertEqual(1, PhoneNumberUtil.translateCursorPosition(2, from: "12", to: "1", stickingRightward: true))
XCTAssertEqual(0, PhoneNumberUtil.translateCursorPosition(0, from: "1", to: "12", stickingRightward: false))
XCTAssertEqual(0, PhoneNumberUtil.translateCursorPosition(0, from: "1", to: "12", stickingRightward: true))
XCTAssertEqual(1, PhoneNumberUtil.translateCursorPosition(1, from: "1", to: "12", stickingRightward: false))
XCTAssertEqual(2, PhoneNumberUtil.translateCursorPosition(1, from: "1", to: "12", stickingRightward: true))
XCTAssertEqual(0, PhoneNumberUtil.translateCursorPosition(0, from: "12", to: "132", stickingRightward: false))
XCTAssertEqual(0, PhoneNumberUtil.translateCursorPosition(0, from: "12", to: "132", stickingRightward: true))
XCTAssertEqual(1, PhoneNumberUtil.translateCursorPosition(1, from: "12", to: "132", stickingRightward: false))
XCTAssertEqual(2, PhoneNumberUtil.translateCursorPosition(1, from: "12", to: "132", stickingRightward: true))
XCTAssertEqual(3, PhoneNumberUtil.translateCursorPosition(2, from: "12", to: "132", stickingRightward: false))
XCTAssertEqual(3, PhoneNumberUtil.translateCursorPosition(2, from: "12", to: "132", stickingRightward: true))
XCTAssertEqual(0, PhoneNumberUtil.translateCursorPosition(0, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: true))
XCTAssertEqual(1, PhoneNumberUtil.translateCursorPosition(1, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: true))
XCTAssertEqual(2, PhoneNumberUtil.translateCursorPosition(2, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: true))
XCTAssertEqual(3, PhoneNumberUtil.translateCursorPosition(3, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: true))
XCTAssertEqual(3, PhoneNumberUtil.translateCursorPosition(4, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: true))
XCTAssertEqual(3, PhoneNumberUtil.translateCursorPosition(5, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: true))
XCTAssertEqual(6, PhoneNumberUtil.translateCursorPosition(6, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: true))
XCTAssertEqual(7, PhoneNumberUtil.translateCursorPosition(7, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: true))
XCTAssertEqual(8, PhoneNumberUtil.translateCursorPosition(8, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: true))
XCTAssertEqual(8, PhoneNumberUtil.translateCursorPosition(9, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: true))
XCTAssertEqual(0, PhoneNumberUtil.translateCursorPosition(0, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: false))
XCTAssertEqual(1, PhoneNumberUtil.translateCursorPosition(1, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: false))
XCTAssertEqual(2, PhoneNumberUtil.translateCursorPosition(2, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: false))
XCTAssertEqual(3, PhoneNumberUtil.translateCursorPosition(3, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: false))
XCTAssertEqual(3, PhoneNumberUtil.translateCursorPosition(4, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: false))
XCTAssertEqual(3, PhoneNumberUtil.translateCursorPosition(5, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: false))
XCTAssertEqual(4, PhoneNumberUtil.translateCursorPosition(6, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: false))
XCTAssertEqual(7, PhoneNumberUtil.translateCursorPosition(7, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: false))
XCTAssertEqual(8, PhoneNumberUtil.translateCursorPosition(8, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: false))
XCTAssertEqual(8, PhoneNumberUtil.translateCursorPosition(9, from: "(55) 123-4567", to: "(551) 234-567", stickingRightward: false))
XCTAssertEqual(0, PhoneNumberUtil.translateCursorPosition(0, from: "(5551) 234-567", to: "(555) 123-4567", stickingRightward: true))
XCTAssertEqual(1, PhoneNumberUtil.translateCursorPosition(1, from: "(5551) 234-567", to: "(555) 123-4567", stickingRightward: true))
XCTAssertEqual(2, PhoneNumberUtil.translateCursorPosition(2, from: "(5551) 234-567", to: "(555) 123-4567", stickingRightward: true))
XCTAssertEqual(3, PhoneNumberUtil.translateCursorPosition(3, from: "(5551) 234-567", to: "(555) 123-4567", stickingRightward: true))
XCTAssertEqual(6, PhoneNumberUtil.translateCursorPosition(4, from: "(5551) 234-567", to: "(555) 123-4567", stickingRightward: true))
XCTAssertEqual(7, PhoneNumberUtil.translateCursorPosition(5, from: "(5551) 234-567", to: "(555) 123-4567", stickingRightward: true))
XCTAssertEqual(0, PhoneNumberUtil.translateCursorPosition(0, from: "(5551) 234-567", to: "(555) 123-4567", stickingRightward: false))
XCTAssertEqual(1, PhoneNumberUtil.translateCursorPosition(1, from: "(5551) 234-567", to: "(555) 123-4567", stickingRightward: false))
XCTAssertEqual(2, PhoneNumberUtil.translateCursorPosition(2, from: "(5551) 234-567", to: "(555) 123-4567", stickingRightward: false))
XCTAssertEqual(3, PhoneNumberUtil.translateCursorPosition(3, from: "(5551) 234-567", to: "(555) 123-4567", stickingRightward: false))
XCTAssertEqual(4, PhoneNumberUtil.translateCursorPosition(4, from: "(5551) 234-567", to: "(555) 123-4567", stickingRightward: false))
XCTAssertEqual(7, PhoneNumberUtil.translateCursorPosition(5, from: "(5551) 234-567", to: "(555) 123-4567", stickingRightward: false))
}
func testCountryNameFromCountryCode() {
XCTAssertEqual(PhoneNumberUtil.countryName(fromCountryCode: "US"), "United States")
XCTAssertEqual(PhoneNumberUtil.countryName(fromCountryCode: "GB"), "United Kingdom")
XCTAssertEqual(PhoneNumberUtil.countryName(fromCountryCode: "EK"), "Unknown")
XCTAssertEqual(PhoneNumberUtil.countryName(fromCountryCode: "ZZZ"), "Unknown")
XCTAssertNotEqual(PhoneNumberUtil.countryName(fromCountryCode: ""), "")
}
}