@@ -17,9 +17,6 @@ final class ConnectionManagerTests: XCTestCase {
1717 var errorDescription : String ? { " sample error " }
1818 }
1919
20- let url = URL ( string: " ws://localhost " ) !
21- let headers = [ " apikey " : " key " ]
22-
2320 var sut : ConnectionManager !
2421 var ws : FakeWebSocket !
2522 var transportCallCount = 0
@@ -62,7 +59,7 @@ final class ConnectionManagerTests: XCTestCase {
6259 }
6360
6461 func testConnectTransitionsThroughConnectingAndConnectedStates( ) async throws {
65- sut = makeSUT ( url : url , headers : headers )
62+ sut = makeSUT ( headers : [ " apikey " : " key " ] )
6663
6764 let connectingExpectation = expectation ( description: " connecting state observed " )
6865 let connectedExpectation = expectation ( description: " connected state observed " )
@@ -81,15 +78,15 @@ final class ConnectionManagerTests: XCTestCase {
8178 }
8279 }
8380
84- let initiallyConnected = await sut. isConnected
81+ let initiallyConnected = await sut. connection != nil
8582 XCTAssertFalse ( initiallyConnected)
8683 try await sut. connect ( )
8784
88- let isConnected = await sut. isConnected
85+ let isConnected = await sut. connection != nil
8986 XCTAssertTrue ( isConnected)
9087 XCTAssertEqual ( transportCallCount, 1 )
91- XCTAssertEqual ( lastConnectURL, url )
92- XCTAssertEqual ( lastConnectHeaders, headers )
88+ XCTAssertEqual ( lastConnectURL? . absoluteString , " ws://localhost " )
89+ XCTAssertEqual ( lastConnectHeaders, [ " apikey " : " key " ] )
9390
9491 await fulfillment ( of: [ connectingExpectation, connectedExpectation] , timeout: 1 )
9592 stateObserver. cancel ( )
@@ -103,7 +100,7 @@ final class ConnectionManagerTests: XCTestCase {
103100
104101 try await sut. connect ( )
105102
106- let stillConnected = await sut. isConnected
103+ let stillConnected = await sut. connection != nil
107104 XCTAssertTrue ( stillConnected)
108105 XCTAssertEqual ( transportCallCount, 1 , " Second connect should reuse existing connection " )
109106 }
@@ -137,7 +134,7 @@ final class ConnectionManagerTests: XCTestCase {
137134 try await secondConnect. value
138135
139136 XCTAssertTrue ( secondConnectFinished. value)
140- let isConnected = await sut. isConnected
137+ let isConnected = await sut. connection != nil
141138 XCTAssertTrue ( isConnected)
142139 XCTAssertEqual ( transportCallCount, 1 )
143140 }
@@ -148,7 +145,7 @@ final class ConnectionManagerTests: XCTestCase {
148145
149146 await sut. disconnect ( reason: " test reason " )
150147
151- let isConnected = await sut. isConnected
148+ let isConnected = await sut. connection != nil
152149 XCTAssertFalse ( isConnected)
153150 guard case . close( let closeCode, let closeReason) ? = ws. sentEvents. last else {
154151 return XCTFail ( " Expected close event to be sent " )
@@ -181,7 +178,7 @@ final class ConnectionManagerTests: XCTestCase {
181178
182179 await Task . yield ( )
183180 XCTAssertTrue ( wasCancelled. value, " Cancellation handler should run when disconnecting " )
184- let isConnected = await sut. isConnected
181+ let isConnected = await sut. connection != nil
185182 XCTAssertFalse ( isConnected)
186183
187184 connectTask. cancel ( )
@@ -218,7 +215,7 @@ final class ConnectionManagerTests: XCTestCase {
218215
219216 await fulfillment ( of: [ reconnectingExpectation, secondConnectionExpectation] , timeout: 2 )
220217 XCTAssertEqual ( connectionCount. value, 2 , " Reconnection should trigger a second transport call " )
221- let isConnected = await sut. isConnected
218+ let isConnected = await sut. connection != nil
222219 XCTAssertTrue ( isConnected)
223220
224221 stateObserver. cancel ( )
@@ -230,7 +227,7 @@ final class ConnectionManagerTests: XCTestCase {
230227
231228 await sut. handleClose ( code: 4001 , reason: " server closing " )
232229
233- let isConnected = await sut. isConnected
230+ let isConnected = await sut. connection != nil
234231 XCTAssertFalse ( isConnected)
235232 guard case . close( let closeCode, let closeReason) ? = ws. sentEvents. last else {
236233 return XCTFail ( " Expected close event to be sent " )
0 commit comments