104 lines
3.3 KiB
Swift
104 lines
3.3 KiB
Swift
// 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()
|
|
}
|
|
}
|