udpate test cases

open-web3-stack · Nov 4, 2024 · f0e1b04 · f0e1b04
1 parent 8ce22a8
commit f0e1b04
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/Networking/Tests/NetworkingTests/MockPeerEventTests.swift b/Networking/Tests/NetworkingTests/MockPeerEventTests.swift
@@ -147,10 +147,10 @@ final class MockPeerEventTests {
             configuration: clientConfiguration
         )
 
-        // Attempt to connect
-        try clientConnection.connect(to: listenAddress)
-        try? await Task.sleep(for: .milliseconds(200))
-        #expect(throws: Error.self) {
+        try? await Task.sleep(for: .milliseconds(100))
+        await #expect(throws: Error.self) {
+            try clientConnection.connect(to: listenAddress)
+            try? await Task.sleep(for: .milliseconds(200))
             let stream1 = try clientConnection.createStream()
             try stream1.send(data: Data("test data 1".utf8))
         }
@@ -253,11 +253,10 @@ final class MockPeerEventTests {
             registration: registration,
             configuration: clientConfiguration
         )
-
-        // Attempt to connect
-        try clientConnection.connect(to: listenAddress)
-        try? await Task.sleep(for: .milliseconds(200))
-        #expect(throws: Error.self) {
+        try? await Task.sleep(for: .milliseconds(100))
+        await #expect(throws: Error.self) {
+            try clientConnection.connect(to: listenAddress)
+            try? await Task.sleep(for: .milliseconds(200))
             let stream1 = try clientConnection.createStream()
             try stream1.send(data: Data("test data 1".utf8))
         }