OpenVPNAdapter/OpenVPN Adapter Tests/OpenVPNAdapterTests.swift

//
//  OpenVPN_Adapter_Tests.swift
//  OpenVPN Adapter Tests
//
//  Created by Sergey Abramchuk on 09.03.17.
//
//

import XCTest
import NetworkExtension
@testable import OpenVPNAdapter

class OpenVPNAdapterTests: XCTestCase {
    
    enum ExpectationsType {
        case connection
    }
    
    var expectations = [ExpectationsType : XCTestExpectation]()
    
    override func setUp() {
        super.setUp()
        expectations.removeAll()
    }
    
    override func tearDown() {
        super.tearDown()
    }
    
    func testApplyConfiguration() {
        let adapter = OpenVPNAdapter()
        
        let configuration = OpenVPNConfiguration()
        configuration.fileContent = ProfileLoader.getVPNProfile(type: .localVPNServer)
        configuration.settings = ["auth-user-pass": ""]
        
        let result: OpenVPNProperties
        do {
            result = try adapter.apply(configuration: configuration)
        } catch {
            XCTFail("Failed to configure OpenVPN adapted due to error: \(error)")
            return
        }
        
        XCTAssert(result.remoteHost == "192.168.1.200")
        XCTAssert(result.remotePort == 1194)
        XCTAssert(result.remoteProto == .UDP)
        XCTAssert(result.autologin == false)
    }
    
    func testProvideCredentials() {
        let adapter = OpenVPNAdapter()
        
        let credentials = OpenVPNCredentials()
        credentials.username = "username"
        credentials.password = "password"
        
        do {
            try adapter.provide(credentials: credentials)
        } catch {
            XCTFail("Failed to provide credentials. \(error)")
            return
        }
    }
    
    // Test connection without specifying username and password
    func testConection() {
        let adapter = OpenVPNAdapter()
        
        let configuration = OpenVPNConfiguration()
        configuration.fileContent = ProfileLoader.getVPNProfile(type: .localVPNServer)
        configuration.settings = ["auth-user-pass": ""]
        
        let result: OpenVPNProperties
        do {
            result = try adapter.apply(configuration: configuration)
        } catch {
            XCTFail("Failed to configure OpenVPN adapted due to error: \(error)")
            return
        }
        
        guard !result.autologin else {
            XCTFail()
            return
        }
        
        let credentials = OpenVPNCredentials()
        credentials.username = "testuser"
        credentials.password = "nonsecure"
        
        do {
            try adapter.provide(credentials: credentials)
        } catch {
            XCTFail("Failed to provide credentials. \(error)")
            return
        }
        
        expectations[.connection] = expectation(description: "me.ss-abramchuk.openvpn-adapter.connection")
        
        adapter.delegate = self
        adapter.connect()
        
        waitForExpectations(timeout: 30.0) { (error) in
            adapter.disconnect()
        }
    }
    
}

extension OpenVPNAdapterTests: OpenVPNAdapterDelegate {
    
    func configureTunnel(settings: NEPacketTunnelNetworkSettings, callback: @escaping (OpenVPNAdapterPacketFlow?) -> Void) {
        callback(self)
    }
    
    func handle(event: OpenVPNEvent, message: String?) {
        switch event {
        case .connected:
            guard let connectionExpectation = expectations[.connection] else { return }
            connectionExpectation.fulfill()
            
        case .disconnected:
            break
            
        default:
            break
        }
    }
    
    func handle(error: Error) {
        if let connectionExpectation = expectations[.connection] {
            XCTFail("Failed to establish conection. \(error.localizedDescription)")
            connectionExpectation.fulfill()
        }
    }
    
    func handle(logMessage: String) {
        print("\(logMessage)")
    }
    
}

extension OpenVPNAdapterTests: OpenVPNAdapterPacketFlow {
    
    func readPackets(completionHandler: @escaping ([Data], [NSNumber]) -> Void) { }
    
    func writePackets(_ packets: [Data], withProtocols protocols: [NSNumber]) -> Bool { return true }
    
}