Fix DataPath unit tests

passepartoutvpn · Sep 2, 2018 · 57b7457 · 57b7457
1 parent 201da9b
commit 57b7457
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/TunnelKitTests/DataPathEncryptionTests.swift b/TunnelKitTests/DataPathEncryptionTests.swift
@@ -87,8 +87,14 @@ class DataPathEncryptionTests: XCTestCase {
     }
 
     func privateTestDataPathHigh(peerId: UInt32?) {
-        let path = DataPath(encrypter: enc, decrypter: dec, maxPackets: 1000, usesReplayProtection: false)
-        path.setCompressionFraming(.disabled)
+        let path = DataPath(
+            encrypter: enc,
+            decrypter: dec,
+            peerId: peerId ?? PacketPeerIdDisabled,
+            compressionFraming: .disabled,
+            maxPackets: 1000,
+            usesReplayProtection: false
+        )
 
         if let peerId = peerId {
             enc.setPeerId(peerId)

diff --git a/TunnelKitTests/DataPathPerformanceTests.swift b/TunnelKitTests/DataPathPerformanceTests.swift
@@ -54,7 +54,14 @@ class DataPathPerformanceTests: XCTestCase {
         encrypter = crypto.encrypter()
         decrypter = crypto.decrypter()
 
-        dataPath = DataPath(encrypter: encrypter, decrypter: decrypter, maxPackets: 200, usesReplayProtection: false)
+        dataPath = DataPath(
+            encrypter: encrypter,
+            decrypter: decrypter,
+            peerId: PacketPeerIdDisabled,
+            compressionFraming: .disabled,
+            maxPackets: 200,
+            usesReplayProtection: false
+        )
     }
 
     override func tearDown() {