Updated to use the new StateHandler

swift-quic · Jul 9, 2023 · 62ba0f4 · 62ba0f4
1 parent e6d4aa4
commit 62ba0f4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Tests/QuicTests/NIOTests/TLSTests.swift b/Tests/QuicTests/NIOTests/TLSTests.swift
@@ -23,7 +23,7 @@ final class TLSClientInitialTests: XCTestCase {
     let version: Version = .version2
     var dcid: ConnectionID!
     var scid: ConnectionID!
-    fileprivate var quicClientHandler: QUICClientHandler!
+    fileprivate var quicClientHandler: QUICStateHandler!
     fileprivate var errorHandler: ErrorEventLogger!
     fileprivate var quiesceEventRecorder: QuiesceEventRecorder!
 
@@ -44,7 +44,8 @@ final class TLSClientInitialTests: XCTestCase {
         let sslContext = try! NIOSSLContext(configuration: configuration)
 
         self.channel = EmbeddedChannel()
-        self.quicClientHandler = try! QUICClientHandler(SocketAddress(ipAddress: "127.0.0.1", port: 0), versions: [self.version], destinationID: self.dcid, sourceID: self.scid, tlsContext: sslContext)
+        self.quicClientHandler = try! QUICStateHandler(SocketAddress(ipAddress: "127.0.0.1", port: 0), perspective: .client, versions: [self.version], destinationID: self.dcid, sourceID: self.scid, tlsContext: sslContext, idleTimeout: .zero)
+
         self.quiesceEventRecorder = QuiesceEventRecorder()
         self.errorHandler = ErrorEventLogger()
         XCTAssertNoThrow(try self.channel.pipeline.addHandler(self.quicClientHandler).wait())