wireguard-apple/WireGuardTests/ValidatorsTests.swift

//
//  ValidatorsTests.swift
//  WireGuardTests
//
//  Created by Jeroen Leenarts on 15-08-18.
//  Copyright © 2018 Jason A. Donenfeld <Jason@zx2c4.com>. All rights reserved.
//

import XCTest
@testable import WireGuard

class ValidatorsTests: XCTestCase {
    func testEndpoint() throws {
        _ = try Endpoint(endpointString: "[2607:f938:3001:4000::aac]:12345")
        _ = try Endpoint(endpointString: "192.168.0.1:12345")
    }

    func testEndpoint_invalidIP() throws {
        func executeTest(endpointString: String, ipString: String, file: StaticString = #file, line: UInt = #line) {
            XCTAssertThrowsError(try Endpoint(endpointString: endpointString)) { (error) in
                guard case EndpointValidationError.invalidIP(let value) = error else {
                    return XCTFail("Unexpected error: \(error)", file: file, line: line)
                }
                XCTAssertEqual(value, ipString, file: file, line: line)
            }
        }

        executeTest(endpointString: "12345:12345", ipString: "12345")
        executeTest(endpointString: ":12345", ipString: "")
    }

    func testEndpoint_invalidPort() throws {
        func executeTest(endpointString: String, portString: String, file: StaticString = #file, line: UInt = #line) {
            XCTAssertThrowsError(try Endpoint(endpointString: endpointString)) { (error) in
                guard case EndpointValidationError.invalidPort(let value) = error else {
                    return XCTFail("Unexpected error: \(error)", file: file, line: line)
                }
                XCTAssertEqual(value, portString, file: file, line: line)
            }
        }

        executeTest(endpointString: ":", portString: "")
        executeTest(endpointString: "[2607:f938:3001:4000::aac]:-12345", portString: "-12345")
        executeTest(endpointString: "[2607:f938:3001:4000::aac]", portString: "aac]")
        executeTest(endpointString: "[2607:f938:3001:4000::aac]:", portString: "")
        executeTest(endpointString: "192.168.0.1:-12345", portString: "-12345")
        executeTest(endpointString: "192.168.0.1:", portString: "")

    }

    func testEndpoint_noIpAndPort() throws {

        func executeTest(endpointString: String, file: StaticString = #file, line: UInt = #line) {
            XCTAssertThrowsError(try Endpoint(endpointString: endpointString)) { (error) in
                guard case EndpointValidationError.noIpAndPort(let value) = error else {
                    return XCTFail("Unexpected error: \(error)", file: file, line: line)
                }
                XCTAssertEqual(value, endpointString, file: file, line: line)
            }
        }

        executeTest(endpointString: "192.168.0.1")
        executeTest(endpointString: "12345")
    }

    func testCIDRAddress() throws {
        _ = try CIDRAddress(stringRepresentation: "2607:f938:3001:4000::aac/24")
        _ = try CIDRAddress(stringRepresentation: "192.168.0.1/24")
    }

    func testIPv4CIDRAddress() throws {
        _ = try CIDRAddress(stringRepresentation: "192.168.0.1/24")
    }

    func testCIDRAddress_invalidIP() throws {
        func executeTest(stringRepresentation: String, ipString: String, file: StaticString = #file, line: UInt = #line) {
            XCTAssertThrowsError(try CIDRAddress(stringRepresentation: stringRepresentation)) { (error) in
                guard case CIDRAddressValidationError.invalidIP(let value) = error else {
                    return XCTFail("Unexpected error: \(error)", file: file, line: line)
                }
                XCTAssertEqual(value, ipString, file: file, line: line)
            }
        }

        executeTest(stringRepresentation: "12345/12345", ipString: "12345")
        executeTest(stringRepresentation: "/12345", ipString: "")
    }

    func testCIDRAddress_invalidSubnet() throws {
        func executeTest(stringRepresentation: String, subnetString: String, file: StaticString = #file, line: UInt = #line) {
            XCTAssertThrowsError(try CIDRAddress(stringRepresentation: stringRepresentation)) { (error) in
                guard case CIDRAddressValidationError.invalidSubnet(let value) = error else {
                    return XCTFail("Unexpected error: \(error)", file: file, line: line)
                }
                XCTAssertEqual(value, subnetString, file: file, line: line)
            }
        }

        executeTest(stringRepresentation: "/", subnetString: "")
        executeTest(stringRepresentation: "2607:f938:3001:4000::aac/a", subnetString: "a")
        executeTest(stringRepresentation: "2607:f938:3001:4000:/aac", subnetString: "aac")
        executeTest(stringRepresentation: "2607:f938:3001:4000::aac/", subnetString: "")
        executeTest(stringRepresentation: "192.168.0.1/a", subnetString: "a")
        executeTest(stringRepresentation: "192.168.0.1/", subnetString: "")

    }

    func testCIDRAddress_noIpAndSubnet() throws {

        func executeTest(stringRepresentation: String, file: StaticString = #file, line: UInt = #line) {
            XCTAssertThrowsError(try CIDRAddress(stringRepresentation: stringRepresentation)) { (error) in
                guard case CIDRAddressValidationError.noIpAndSubnet(let value) = error else {
                    return XCTFail("Unexpected error: \(error)", file: file, line: line)
                }
                XCTAssertEqual(value, stringRepresentation, file: file, line: line)
            }
        }

        executeTest(stringRepresentation: "192.168.0.1")
        executeTest(stringRepresentation: "12345")
    }

}