From c838baef4da05ccffa46480ff8cf3f1b222d6141 Mon Sep 17 00:00:00 2001 From: Sergey Abramchuk Date: Tue, 23 Jan 2018 17:17:48 +0300 Subject: [PATCH] Update tests --- OpenVPN Adapter Tests/CustomFlow.swift | 17 +++++------------ OpenVPN Adapter Tests/OpenVPNAdapterTests.swift | 2 +- 2 files changed, 6 insertions(+), 13 deletions(-) diff --git a/OpenVPN Adapter Tests/CustomFlow.swift b/OpenVPN Adapter Tests/CustomFlow.swift index dbd423f..23e0f58 100644 --- a/OpenVPN Adapter Tests/CustomFlow.swift +++ b/OpenVPN Adapter Tests/CustomFlow.swift @@ -6,22 +6,15 @@ // import NetworkExtension +@testable import OpenVPNAdapter -class CustomFlow: NEPacketTunnelFlow { - - override func readPackets(completionHandler: @escaping ([Data], [NSNumber]) -> Void) { +class CustomFlow: NSObject, OpenVPNAdapterPacketFlow { + + func readPackets(completionHandler: @escaping ([Data], [NSNumber]) -> Void) { } - override func writePackets(_ packets: [Data], withProtocols protocols: [NSNumber]) -> Bool { - return true - } - - override func readPacketObjects(completionHandler: @escaping ([NEPacket]) -> Void) { - - } - - override func writePacketObjects(_ packets: [NEPacket]) -> Bool { + func writePackets(_ packets: [Data], withProtocols protocols: [NSNumber]) -> Bool { return true } diff --git a/OpenVPN Adapter Tests/OpenVPNAdapterTests.swift b/OpenVPN Adapter Tests/OpenVPNAdapterTests.swift index 0642e45..40a83f6 100644 --- a/OpenVPN Adapter Tests/OpenVPNAdapterTests.swift +++ b/OpenVPN Adapter Tests/OpenVPNAdapterTests.swift @@ -135,7 +135,7 @@ class OpenVPNAdapterTests: XCTestCase { extension OpenVPNAdapterTests: OpenVPNAdapterDelegate { - func openVPNAdapter(_ openVPNAdapter: OpenVPNAdapter, configureTunnelWithNetworkSettings networkSettings: NEPacketTunnelNetworkSettings, completionHandler: @escaping (NEPacketTunnelFlow?) -> Void) { + func openVPNAdapter(_ openVPNAdapter: OpenVPNAdapter, configureTunnelWithNetworkSettings networkSettings: NEPacketTunnelNetworkSettings, completionHandler: @escaping (OpenVPNAdapterPacketFlow?) -> Void) { completionHandler(customFlow) }