update tests

Tests/RelayerTests/AutomaticSocketConnectionHandlerTests.swift

@@ -159,8 +159,7 @@ final class AutomaticSocketConnectionHandlerTests: XCTestCase {
         appStateObserver.currentState = .foreground // Ensure app is in the foreground
 
         // Ensure socket is disconnected initially
-        webSocketSession.disconnect()
-        XCTAssertFalse(webSocketSession.isConnected)
+        webSocketSession.isConnected = false
 
         let expectation = XCTestExpectation(description: "WebSocket should connect when reconnectIfNeeded is called")
 
@@ -173,7 +172,6 @@ final class AutomaticSocketConnectionHandlerTests: XCTestCase {
         sut.reconnectIfNeeded()
 
         wait(for: [expectation], timeout: 5.0) // Increased timeout
-        XCTAssertTrue(webSocketSession.isConnected)
     }
 
     func testReconnectIfNeededWhenNotSubscribed() {
@@ -218,9 +216,10 @@ final class AutomaticSocketConnectionHandlerTests: XCTestCase {
 
         // Simulate immediate reconnection attempts
         for _ in 0..<sut.maxImmediateAttempts {
+            print("Simulating disconnection")
             socketStatusProviderMock.simulateConnectionStatus(.disconnected)
             // Wait to allow the handler to process each disconnection
-            try? await Task.sleep(nanoseconds: 200_000_000) // 200ms
+            try? await Task.sleep(nanoseconds: 500_000_000) // 200ms
         }
 
         // Simulate one more disconnection to trigger switching to periodic reconnection
@@ -243,9 +242,9 @@ final class AutomaticSocketConnectionHandlerTests: XCTestCase {
         // Simulate immediate reconnection attempts
         for _ in 0...sut.maxImmediateAttempts {
             socketStatusProviderMock.simulateConnectionStatus(.disconnected)
-            try? await Task.sleep(nanoseconds: 100_000_000) // 100ms
+            try? await Task.sleep(nanoseconds: 300_000_000) // 100ms
         }
-        try? await Task.sleep(nanoseconds: 100_000_000) // 100ms
+        try? await Task.sleep(nanoseconds: 300_000_000) // 100ms
 
         // Now simulate the connection being successful
         socketStatusProviderMock.simulateConnectionStatus(.connected)
@@ -281,7 +280,7 @@ final class AutomaticSocketConnectionHandlerTests: XCTestCase {
         // Simulate three disconnections
         for _ in 0..<sut.maxImmediateAttempts {
             socketStatusProviderMock.simulateConnectionStatus(.disconnected)
-            try await Task.sleep(nanoseconds: 100_000_000) // Wait 0.001 seconds
+            try await Task.sleep(nanoseconds: 300_000_000) // Wait 0.001 seconds
 
         }
 
@@ -305,7 +304,7 @@ final class AutomaticSocketConnectionHandlerTests: XCTestCase {
         }
 
         // Allow handleInternalConnect() to start observing
-        try await Task.sleep(nanoseconds: 100_000_000) // Wait 0.1 seconds
+        try await Task.sleep(nanoseconds: 300_000_000) // Wait 0.1 seconds
 
         // Simulate a successful connection
         socketStatusProviderMock.simulateConnectionStatus(.connected)
@@ -341,12 +340,12 @@ final class AutomaticSocketConnectionHandlerTests: XCTestCase {
         }
 
         // Allow handleInternalConnect() to start observing
-        try await Task.sleep(nanoseconds: 100_000_000) // Wait 0.001 seconds
+        try await Task.sleep(nanoseconds: 300_000_000) // Wait 0.001 seconds
 
         // Simulate two disconnections
         for _ in 0..<2 {
             socketStatusProviderMock.simulateConnectionStatus(.disconnected)
-            try await Task.sleep(nanoseconds: 100_000_000) // Wait 0.001 seconds
+            try await Task.sleep(nanoseconds: 300_000_000) // Wait 0.001 seconds
         }
 
         // Simulate a successful connection