try to find timeout test cases

Author: doghappy

tests/SocketIOClient.UnitTests/Protocol/WebSocket/WebSocketAdapterTests.cs

@@ -140,14 +140,7 @@ public async Task ConnectAsync_NotConnectedEvenReceivedMessage_ObserverCannotGet
     [Fact]
     public async Task ConnectAsync_ReceivedTextMessage_NotifyToObserver()
     {
-        var tcs = new TaskCompletionSource<ProtocolMessage>();
         var observer = Substitute.For<IMyObserver<ProtocolMessage>>();
-        observer.OnNextAsync(Arg.Any<ProtocolMessage>())
-            .Returns(callInfo =>
-            {
-                tcs.SetResult(callInfo.Arg<ProtocolMessage>());
-                return Task.CompletedTask;
-            });
         _wsAdapter.Subscribe(observer);
         _clientAdapter.ReceiveAsync(Arg.Is<CancellationToken>(c => c != CancellationToken.None))
             .Returns(async _ =>
@@ -162,26 +155,23 @@ public async Task ConnectAsync_ReceivedTextMessage_NotifyToObserver()
 
         await _wsAdapter.ConnectAsync(new Uri("ws://127.0.0.1:1234"), CancellationToken.None);
 
-        var message = await tcs.Task.ConfigureAwait(false);
-        message.Should().BeEquivalentTo(new ProtocolMessage
-        {
-            Type = ProtocolMessageType.Text,
-            Text = "Hello World!"
-        });
+        await Task.Delay(400).ConfigureAwait(false);
+
+        await observer.Received()
+            .OnNextAsync(Arg.Is<ProtocolMessage>(m =>
+                m.Type == ProtocolMessageType.Text
+                && m.Text == "Hello World!"));
     }
 
     [Fact]
     public async Task ReceiveAsync_ExceptionOccurred_OnDisconnectInvoked()
     {
-        var tcs = new TaskCompletionSource();
-        _onDisconnect.When(x => x.Invoke())
-            .Do(_ => tcs.SetResult());
         _clientAdapter.ReceiveAsync(Arg.Any<CancellationToken>())
             .ThrowsAsync(new Exception("Aborted"));
 
         await _wsAdapter.ConnectAsync(new Uri("ws://127.0.0.1:1234"), CancellationToken.None);
 
-        await tcs.Task.ConfigureAwait(false);
+        await Task.Delay(400).ConfigureAwait(false);
         _onDisconnect.Received().Invoke();
     }
 

tests/SocketIOClient.UnitTests/SocketIOTests.cs

@@ -1682,7 +1682,7 @@ public async Task SessionOnDisconnected_ReconnectionIsTrue_ReconnectAutomaticall
         _io.OnDisconnected += onDisconnectedHandler;
         _eventRunner
             .When(x => x.RunInBackground(onDisconnectedHandler, Arg.Any<object>(), Arg.Any<string>()))
-            .Do(x => tcs.SetResult());
+            .Do(_ => tcs.SetResult());
 
         await ConnectAsync();
         _session.OnDisconnected();
@@ -1707,7 +1707,7 @@ public async Task DisconnectAsync_ReconnectionIsTrue_NotReconnect()
         _io.OnDisconnected += onDisconnectedHandler;
         _eventRunner
             .When(x => x.RunInBackground(onDisconnectedHandler, Arg.Any<object>(), DisconnectReason.IOClientDisconnect))
-            .Do(x => tcs.SetResult());
+            .Do(_ => tcs.SetResult());
 
         await ConnectAsync();
         await _io.DisconnectAsync();
@@ -1731,7 +1731,7 @@ public async Task ReceivedDisconnectedMessage_ReconnectionIsTrue_NotReconnect()
         _io.OnDisconnected += onDisconnectedHandler;
         _eventRunner
             .When(x => x.RunInBackground(onDisconnectedHandler, Arg.Any<object>(), DisconnectReason.IOServerDisconnect))
-            .Do(x => tcs.SetResult());
+            .Do(_ => tcs.SetResult());
 
         await ConnectAsync();
         await OnNextAsync(_io, new DisconnectedMessage()).ConfigureAwait(false);