test: 增加连接超时单元测试

Author: ArgoZhang

test/UnitTest/Services/TcpSocketFactoryTest.cs

@@ -48,11 +48,14 @@ public async Task GetOrCreate_Ok()
     [Fact]
     public async Task ConnectAsync_Timeout()
     {
-        var client = CreateClient();
-        client.Options.ConnectTimeout = 1;
+        var client = CreateClient(builder =>
+        {
+            // 增加发送报错 MockSocket
+            builder.AddTransient<ISocketClientProvider, MockConnectTimeoutSocketProvider>();
+        });
+        client.Options.ConnectTimeout = 10;
 
         var connect = await client.ConnectAsync("localhost", 9999);
-        await Task.Delay(5);
         Assert.False(connect);
     }
 
@@ -703,6 +706,35 @@ public ValueTask<bool> SendAsync(ReadOnlyMemory<byte> data, CancellationToken to
         }
     }
 
+    class MockConnectTimeoutSocketProvider : ISocketClientProvider
+    {
+        public bool IsConnected { get; private set; }
+
+        public IPEndPoint LocalEndPoint { get; set; } = new IPEndPoint(IPAddress.Any, 0);
+
+        public ValueTask CloseAsync()
+        {
+            return ValueTask.CompletedTask;
+        }
+
+        public async ValueTask<bool> ConnectAsync(IPEndPoint endPoint, CancellationToken token = default)
+        {
+            await Task.Delay(1000, token);
+            IsConnected = false;
+            return false;
+        }
+
+        public ValueTask<int> ReceiveAsync(Memory<byte> buffer, CancellationToken token = default)
+        {
+            return ValueTask.FromResult(0);
+        }
+
+        public ValueTask<bool> SendAsync(ReadOnlyMemory<byte> data, CancellationToken token = default)
+        {
+            return ValueTask.FromResult(true);
+        }
+    }
+
     class MockSendTimeoutSocketProvider : ISocketClientProvider
     {
         public bool IsConnected { get; private set; }