PhoneNumberValidatorTest.swift

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

import XCTest

@testable import Signal
@testable import SignalServiceKit

class PhoneNumberValidatorTest: XCTestCase {
    private var phoneNumberUtilRef: PhoneNumberUtil!

    override func setUp() {
        super.setUp()
        phoneNumberUtilRef = PhoneNumberUtil()
    }

    func assertValid(e164: String, file: StaticString = #filePath, line: UInt = #line) {
        let validator = PhoneNumberValidator()
        guard let phoneNumber = E164(phoneNumberUtilRef.parseE164(e164)?.e164) else {
            XCTFail("unparsable phone number", file: file, line: line)
            return
        }
        let isValid = validator.isValidForRegistration(phoneNumber: phoneNumber)
        XCTAssertTrue(isValid, file: file, line: line)
    }

    func assertInvalid(e164: String, file: StaticString = #filePath, line: UInt = #line) {
        let validator = PhoneNumberValidator()
        guard let phoneNumber = E164(phoneNumberUtilRef.parsePhoneNumber(userSpecifiedText: e164)?.e164) else {
            // number wasn't even parsable
            return
        }
        let isValid = validator.isValidForRegistration(phoneNumber: phoneNumber)
        XCTAssertFalse(isValid, file: file, line: line)
    }

    func testUnitedStates() {
        // valid us number
        assertValid(e164: "[phone]")

        // too short
        assertInvalid(e164: "[phone]")

        // too long
        assertInvalid(e164: "[phone]")

        // not a US phone number
        assertValid(e164: "[phone]")
    }

    func testBrazil() {
        // valid mobile
        assertValid(e164: "[phone]")

        // valid landline
        assertValid(e164: "[phone]")

        // mobile length, but with out the leading '9'
        assertInvalid(e164: "[phone]")

        // too short
        assertInvalid(e164: "[phone]")

        // too long landline
        assertInvalid(e164: "[phone]")
        assertInvalid(e164: "[phone]")

        // too long mobile
        assertInvalid(e164: "[phone]")
        assertInvalid(e164: "[phone]")
    }
}