Test cleanup.

Tests/WebSocketTests.swift

@@ -14,14 +14,6 @@ import XCTest
 
 @available(macOS 10.15, iOS 13, tvOS 13, watchOS 6, *)
 final class WebSocketTests: BaseTestCase {
-//    override var skipVersion: SkipVersion { .twenty }
-
-//    func testMany() {
-//        for _ in 0..<5000 {
-//            testThatWebSocketsCanReceiveMessageEvents()
-//        }
-//    }
-
     func testThatWebSocketsCanReceiveMessageEvents() {
         // Given
         let didConnect = expectation(description: "didConnect")
@@ -70,8 +62,6 @@ final class WebSocketTests: BaseTestCase {
     func testThatWebSocketsCanReceiveAMessage() {
         // Given
         let didReceiveMessage = expectation(description: "didReceiveMessage")
-        let didComplete = expectation(description: "didComplete")
-
         let session = stored(Session())
 
         var receivedMessage: URLSessionWebSocketTask.Message?
@@ -81,9 +71,6 @@ final class WebSocketTests: BaseTestCase {
             receivedMessage = message
             didReceiveMessage.fulfill()
         }
-        .onCompletion {
-            didComplete.fulfill()
-        }
 
         waitForExpectations(timeout: timeout)
 
@@ -142,7 +129,6 @@ final class WebSocketTests: BaseTestCase {
     func testThatWebSocketsCanReceiveADecodableValue() {
         // Given
         let didReceiveValue = expectation(description: "didReceiveMessage")
-        let didComplete = expectation(description: "didComplete")
 
         let session = stored(Session())
 
@@ -153,9 +139,6 @@ final class WebSocketTests: BaseTestCase {
             receivedValue = value
             didReceiveValue.fulfill()
         }
-        .onCompletion {
-            didComplete.fulfill()
-        }
 
         waitForExpectations(timeout: timeout)
 
@@ -603,18 +586,6 @@ final class WebSocketTests: BaseTestCase {
     }
 }
 
-@available(macOS 10.15, iOS 13, tvOS 13, watchOS 6, *)
-extension WebSocketRequest {
-    @discardableResult
-    func onCompletion(_ handler: @escaping () -> Void) -> Self {
-        streamMessageEvents { event in
-            guard case .completed = event.kind else { return }
-
-            handler()
-        }
-    }
-}
-
 @available(macOS 10.15, iOS 13, tvOS 13, watchOS 6, *)
 extension URLSessionWebSocketTask.Message: Equatable {
     public static func ==(lhs: URLSessionWebSocketTask.Message, rhs: URLSessionWebSocketTask.Message) -> Bool {