Adds Messenger.onClose for client cleanup

Author: NeedleInAJayStack

Sources/GraphQLTransportWS/Messenger.swift

@@ -8,6 +8,7 @@ public protocol Messenger: AnyObject {
     // AnyObject compliance requires that the implementing object is a class and we can reference it weakly
     func send<S>(_ message: S) -> Void where S: Collection, S.Element == Character
     func onRecieve(callback: @escaping (String) -> Void) -> Void
+    func onClose(callback: @escaping () -> Void) -> Void
     func close() -> Void
     func error(_ message: String, code: Int) -> Void
 }

Tests/GraphQLTransportWSTests/GraphQLTransportWSTests.swift

@@ -43,6 +43,35 @@ class GraphqlTransportWsTests: XCTestCase {
         )
     }
 
+    /// Tests that trying to run methods before `connection_init` is not allowed
+    func testInitialize() throws {
+        var messages = [String]()
+        let completeExpectation = XCTestExpectation()
+        
+        let client = Client(messenger: clientMessenger)
+        client.onMessage { message, _ in
+            messages.append(message)
+            completeExpectation.fulfill()
+        }
+        
+        client.sendStart(
+            payload: GraphQLRequest(
+                query: """
+                    query {
+                        hello
+                    }
+                    """
+            ),
+            id: UUID().uuidString
+        )
+        
+        wait(for: [completeExpectation], timeout: 2)
+        XCTAssertEqual(
+            messages,
+            ["4407: Connection not initialized"]
+        )
+    }
+    
     /// Tests that throwing in the authorization callback forces an unauthorized error
     func testAuth() throws {
         server.auth { payload in

Tests/GraphQLTransportWSTests/TestMessenger.swift

@@ -10,6 +10,7 @@ import Foundation
 class TestMessenger: Messenger {
     weak var other: TestMessenger?
     var onRecieve: (String) -> Void = { _ in }
+    var onClose: () -> Void = { }
     let queue: DispatchQueue = .init(label: "Test messenger")
 
     init() {}
@@ -29,11 +30,16 @@ class TestMessenger: Messenger {
         self.onRecieve = callback
     }
 
+    func onClose(callback: @escaping () -> Void) {
+        self.onClose = callback
+    }
+    
     func error(_ message: String, code: Int) {
         self.send("\(code): \(message)")
     }
 
     func close() {
         // This is a testing no-op
+        self.onClose()
     }
 }