init

Tests/NatsTests/Integration/EventsTests.swift

@@ -0,0 +1,103 @@
+// Copyright 2024 The NATS Authors
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import Foundation
+import Logging
+import Nats
+import NatsServer
+import XCTest
+
+class TestNatsEvents: XCTestCase {
+
+    static var allTests = [
+        ("testClientConnectedEvent", testClientConnectedEvent),
+        ("testClientClosedEvent", testClientClosedEvent),
+        ("testClientReconnectEvent", testClientReconnectEvent),
+    ]
+
+    var natsServer = NatsServer()
+
+    override func tearDown() {
+        super.tearDown()
+        natsServer.stop()
+    }
+
+    func testClientConnectedEvent() async throws {
+        natsServer.start()
+        logger.logLevel = .critical
+
+        let client = NatsClientOptions().url(URL(string: natsServer.clientURL)!).build()
+
+        let expectation = XCTestExpectation(
+            description: "client was not notified of connection established event")
+        client.on(.connected) { event in
+            XCTAssertEqual(event.kind(), NatsEventKind.connected)
+            expectation.fulfill()
+        }
+        try await client.connect()
+
+        await fulfillment(of: [expectation], timeout: 1.0)
+        try await client.close()
+    }
+
+    func testClientClosedEvent() async throws {
+        natsServer.start()
+        logger.logLevel = .critical
+
+        let client = NatsClientOptions().url(URL(string: natsServer.clientURL)!).build()
+
+        let expectation = XCTestExpectation(
+            description: "client was not notified of connection closed event")
+        client.on(.closed) { event in
+            XCTAssertEqual(event.kind(), NatsEventKind.closed)
+            expectation.fulfill()
+        }
+        try await client.connect()
+
+        try await client.close()
+        await fulfillment(of: [expectation], timeout: 1.0)
+    }
+
+    func testClientReconnectEvent() async throws {
+        natsServer.start()
+        let port = natsServer.port!
+        logger.logLevel = .critical
+
+        let client = NatsClientOptions()
+            .url(URL(string: natsServer.clientURL)!)
+            .reconnectWait(1)
+            .build()
+
+        let disconnected = XCTestExpectation(
+            description: "client was not notified of disconnection event")
+        client.on(.disconnected) { event in
+            XCTAssertEqual(event.kind(), NatsEventKind.disconnected)
+            disconnected.fulfill()
+        }
+        try await client.connect()
+        natsServer.stop()
+
+        let reconnected = XCTestExpectation(
+            description: "client was not notified of reconnection event")
+        client.on(.connected) { event in
+            XCTAssertEqual(event.kind(), NatsEventKind.connected)
+            reconnected.fulfill()
+        }
+        await fulfillment(of: [disconnected], timeout: 5.0)
+
+        natsServer.start(port: port)
+        await fulfillment(of: [reconnected], timeout: 5.0)
+
+        try await client.close()
+    }
+}