Add testCloseTimeout

hummingbird-project · Dec 18, 2024 · 9b6d0ab · 9b6d0ab
1 parent 32091be
commit 9b6d0ab
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/Tests/HummingbirdWebSocketTests/WebSocketTests.swift b/Tests/HummingbirdWebSocketTests/WebSocketTests.swift
@@ -573,6 +573,25 @@ final class HummingbirdWebSocketTests: XCTestCase {
         }
     }
 
+    func testCloseTimeout() async throws {
+        let app = Application(
+            router: Router(),
+            server: .http1WebSocketUpgrade { _, _, _ in
+                .upgrade([:]) { _, _, _ in
+                    try await cancelWhenGracefulShutdown {
+                        try await Task.sleep(for: .seconds(15))
+                        XCTFail("Should not reach here")
+                    }
+                }
+            }
+        )
+        try await app.test(.live) { client in
+            _ = try await client.ws("/", configuration: .init(closeTimeout: .seconds(1))) { _, outbound, _ in
+                try await outbound.write(.text("Hello"))
+            }
+        }
+    }
+
     func testUnrecognisedOpcode() async throws {
         let app = Application(
             router: Router(),