test: 增加服务器断开单元测试

Author: ArgoZhang

test/UnitTest/Services/DefaultSocketClientProviderTest.cs

@@ -3,7 +3,9 @@
 // See the LICENSE file in the project root for more information.
 // Maintainer: Argo Zhang([email protected]) Website: https://www.blazor.zone
 
+using Microsoft.Extensions.Logging;
 using System.Net;
+using System.Net.Sockets;
 
 namespace UnitTest.Services;
 
@@ -26,6 +28,35 @@ public async Task DefaultSocketClient_Ok()
         Assert.Equal(0, len);
     }
 
+    [Fact]
+    public async Task ReceiveAsync_Ok()
+    {
+        var port = 8100;
+        // 测试接收数据时服务器断开未连接的情况
+        StartTcpServer(port);
+
+        var sc = new ServiceCollection();
+        sc.AddBootstrapBlazorTcpSocketFactory();
+        var provider = sc.BuildServiceProvider();
+        var factory = provider.GetRequiredService<ITcpSocketFactory>();
+        var client = factory.GetOrCreate("provider", op =>
+        {
+            op.LocalEndPoint = Utility.ConvertToIpEndPoint("localhost", 0);
+            op.IsAutoReceive = false;
+            op.EnableLog = false;
+        });
+
+        await client.ConnectAsync("127.0.0.1", port);
+        Assert.True(client.IsConnected);
+
+        var buffer = await client.ReceiveAsync();
+        Assert.Equal(2, buffer.Length);
+
+        await Task.Delay(50);
+        buffer = await client.ReceiveAsync();
+        Assert.False(client.IsConnected);
+    }
+
     [Fact]
     public void SocketClientOptions_Ok()
     {
@@ -45,4 +76,35 @@ public void SocketClientOptions_Ok()
         Assert.Equal(500, options.ReceiveTimeout);
         Assert.Equal(new IPEndPoint(IPAddress.Loopback, 0), options.LocalEndPoint);
     }
+
+    private static TcpListener StartTcpServer(int port)
+    {
+        var server = new TcpListener(IPAddress.Loopback, port);
+        server.Start();
+        Task.Run(() => AcceptClientsAsync(server));
+        return server;
+    }
+
+    private static async Task AcceptClientsAsync(TcpListener server)
+    {
+        while (true)
+        {
+            var client = await server.AcceptTcpClientAsync();
+            _ = Task.Run(async () =>
+            {
+                using var stream = client.GetStream();
+                while (true)
+                {
+                    var buffer = new byte[1024];
+
+                    // 模拟拆包发送第二段数据
+                    await stream.WriteAsync(new byte[] { 0x3, 0x4 }, CancellationToken.None);
+
+                    // 等待 20ms
+                    await Task.Delay(20);
+                    client.Close();
+                }
+            });
+        }
+    }
 }