passepartout-apple/Passepartout-CoreTests/InfrastructureTests.swift

//
//  InfrastructureTests.swift
//  Passepartout-CoreTests
//
//  Created by Davide De Rosa on 6/11/18.
//  Copyright (c) 2019 Davide De Rosa. All rights reserved.
//
//  https://github.com/passepartoutvpn
//
//  This file is part of Passepartout.
//
//  Passepartout is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
//
//  Passepartout is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//
//  You should have received a copy of the GNU General Public License
//  along with Passepartout.  If not, see <http://www.gnu.org/licenses/>.
//

import XCTest
@testable import Passepartout_Core
import TunnelKit

class InfrastructureTests: XCTestCase {
    private let infra = InfrastructureFactory.shared.get(.pia)

    override func setUp() {
    }

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

    func testParsing() {
        print(infra.pools)
        XCTAssertEqual(infra.pools.count, 48)
    }

    func testIdentifier() {
        let id = "us-east"
        guard let pool = infra.pool(for: id) else {
            XCTAssert(false)
            return
        }
        print(pool)
        XCTAssertEqual(pool.id, id)
    }
    
    func testStableSort() {
        let original: [EndpointProtocol] = [
            EndpointProtocol(.udp, 1194),
            EndpointProtocol(.udp, 8080),
            EndpointProtocol(.udp, 9201),
            EndpointProtocol(.udp, 53),
            EndpointProtocol(.udp, 1197),
            EndpointProtocol(.udp, 198),
            EndpointProtocol(.tcp, 443),
            EndpointProtocol(.tcp, 110),
            EndpointProtocol(.tcp, 80),
            EndpointProtocol(.tcp, 500),
            EndpointProtocol(.tcp, 501),
            EndpointProtocol(.tcp, 502)
        ]
        var preferredType: SocketType
        
        preferredType = .udp
        let sorted1 = original.stableSorted {
            return ($0.socketType == preferredType) && ($1.socketType != preferredType)
        }
        XCTAssertEqual(sorted1, original)

        preferredType = .tcp
        let sorted2 = original.stableSorted {
            return ($0.socketType == preferredType) && ($1.socketType != preferredType)
        }
        XCTAssertNotEqual(sorted2, original)
    }
}