Web socket should end

Author: swhitty

FlyingFox/Sources/WebSocket/WSHandler.swift

@@ -105,6 +105,7 @@ public struct MessageFrameWSHandler: WSHandler {
                             framesOut.yield(frame)
                         }
                     }
+                    framesOut.finish(throwing: nil)
                 } catch FrameError.closed {
                     framesOut.yield(.close(message: "Goodbye"))
                     framesOut.finish(throwing: nil)

FlyingFox/Tests/WebSocket/WSHandlerTests.swift

@@ -116,6 +116,15 @@ final class WSHandlerTests: XCTestCase {
         )
     }
 
+    func testResponseFramesEnds() async throws {
+        let handler = MessageFrameWSHandler.make()
+        let frames = try await handler.makeFrames(for: [.ping])
+
+        await AsyncAssertEqual(
+            try await frames.collectAll(),
+            [.pong]
+        )
+    }
 }
 
 private extension MessageFrameWSHandler {