test: 更新单元测试

Author: ArgoZhang

src/BootstrapBlazor/Services/TcpSocket/TcpSocketClientBase.cs

@@ -184,8 +184,9 @@ public virtual async ValueTask<Memory<byte>> ReceiveAsync(CancellationToken toke
 
     private async ValueTask AutoReceiveAsync()
     {
+        // 自动接收方法
         _receiveCancellationTokenSource ??= new();
-        while (_receiveCancellationTokenSource is { IsCancellationRequested: false })
+        while (true)
         {
             if (SocketClientProvider is not { IsConnected: true })
             {
@@ -197,6 +198,7 @@ private async ValueTask AutoReceiveAsync()
             var len = await ReceiveCoreAsync(SocketClientProvider, buffer, _receiveCancellationTokenSource.Token);
             if (len == 0)
             {
+                // 远端关闭或者 DisposeAsync 方法被调用时退出
                 break;
             }
         }

test/UnitTest/Services/TcpSocketFactoryTest.cs

@@ -7,7 +7,6 @@
 using System.Net;
 using System.Net.Sockets;
 using System.Reflection;
-using System.Text;
 
 namespace UnitTest.Services;
 
@@ -174,7 +173,6 @@ public async Task ReceiveAsync_Timeout()
 
         var client = CreateClient();
         client.Options.ReceiveTimeout = 100;
-        client.SetDataHandler(new MockReceiveTimeoutHandler());
 
         await client.ConnectAsync("localhost", port);
 
@@ -190,8 +188,6 @@ public async Task ReceiveAsync_Cancel()
         var server = StartTcpServer(port, MockSplitPackageAsync);
 
         var client = CreateClient();
-        client.SetDataHandler(new MockReceiveTimeoutHandler());
-
         await client.ConnectAsync("localhost", port);
 
         var data = new ReadOnlyMemory<byte>([1, 2, 3, 4, 5]);
@@ -272,8 +268,6 @@ public async Task ReceiveAsync_Error()
         client.Options.ReceiveBufferSize = 1024 * 20;
         Assert.Equal(1024 * 20, client.Options.ReceiveBufferSize);
 
-        client.SetDataHandler(new MockReceiveErrorHandler());
-
         ReadOnlyMemory<byte> buffer = ReadOnlyMemory<byte>.Empty;
         var tcs = new TaskCompletionSource();
 
@@ -298,26 +292,6 @@ public async Task ReceiveAsync_Error()
         StopTcpServer(server);
     }
 
-    [Fact]
-    public async Task CloseByRemote_Ok()
-    {
-        var client = CreateClient();
-
-        var port = 8883;
-        var server = StartTcpServer(port, MockAutoClosePackageAsync);
-
-        client.SetDataHandler(new MockReceiveErrorHandler());
-
-        // 连接 TCP Server
-        await client.ConnectAsync("localhost", port);
-
-        // 发送数据
-        await client.SendAsync(new ReadOnlyMemory<byte>([1, 2, 3, 4, 5]));
-
-        // 关闭连接
-        StopTcpServer(server);
-    }
-
     [Fact]
     public async Task FixLengthDataPackageHandler_Ok()
     {
@@ -333,17 +307,6 @@ public async Task FixLengthDataPackageHandler_Ok()
         var tcs = new TaskCompletionSource();
         ReadOnlyMemory<byte> receivedBuffer = ReadOnlyMemory<byte>.Empty;
 
-        // 增加数据处理适配器
-        client.SetDataHandler(new FixLengthDataPackageHandler(7)
-        {
-            ReceivedCallBack = buffer =>
-            {
-                receivedBuffer = buffer;
-                tcs.SetResult();
-                return ValueTask.CompletedTask;
-            }
-        });
-
         // 测试 SendAsync 方法
         var data = new ReadOnlyMemory<byte>([1, 2, 3, 4, 5]);
         var result = await client.SendAsync(data);
@@ -373,17 +336,6 @@ public async Task FixLengthDataPackageHandler_Sticky()
         var tcs = new TaskCompletionSource();
         ReadOnlyMemory<byte> receivedBuffer = ReadOnlyMemory<byte>.Empty;
 
-        // 增加数据库处理适配器
-        client.SetDataHandler(new FixLengthDataPackageHandler(7)
-        {
-            ReceivedCallBack = buffer =>
-            {
-                receivedBuffer = buffer;
-                tcs.SetResult();
-                return ValueTask.CompletedTask;
-            }
-        });
-
         // 发送数据
         var data = new ReadOnlyMemory<byte>([1, 2, 3, 4, 5]);
         await client.SendAsync(data);
@@ -425,17 +377,6 @@ public async Task DelimiterDataPackageHandler_Ok()
         var tcs = new TaskCompletionSource();
         ReadOnlyMemory<byte> receivedBuffer = ReadOnlyMemory<byte>.Empty;
 
-        // 增加数据库处理适配器
-        //client.SetDataHandler(new DelimiterDataPackageHandler([0x13, 0x10])
-        //{
-        //    ReceivedCallBack = buffer =>
-        //    {
-        //        receivedBuffer = buffer;
-        //        tcs.SetResult();
-        //        return ValueTask.CompletedTask;
-        //    }
-        //});
-
         // 发送数据
         var data = new ReadOnlyMemory<byte>([1, 2, 3, 4, 5]);
         await client.SendAsync(data);
@@ -560,12 +501,6 @@ private static async Task MockStickyPackageAsync(TcpClient client)
         }
     }
 
-    private static Task MockAutoClosePackageAsync(TcpClient client)
-    {
-        client.Close();
-        return Task.CompletedTask;
-    }
-
     private static void StopTcpServer(TcpListener server)
     {
         server?.Stop();
@@ -661,26 +596,6 @@ public override async ValueTask ReceiveAsync(ReadOnlyMemory<byte> data, Cancella
         }
     }
 
-    class MockSendTimeoutHandler : DataPackageHandlerBase
-    {
-        public override async ValueTask<ReadOnlyMemory<byte>> SendAsync(ReadOnlyMemory<byte> data, CancellationToken token = default)
-        {
-            // 模拟发送超时
-            await Task.Delay(200, token);
-            return data;
-        }
-    }
-
-    class MockReceiveTimeoutHandler : DataPackageHandlerBase
-    {
-        public override async ValueTask ReceiveAsync(ReadOnlyMemory<byte> data, CancellationToken token = default)
-        {
-            // 模拟接收超时
-            await Task.Delay(200, token);
-            await base.ReceiveAsync(data, token);
-        }
-    }
-
     class MockSendErrorSocketProvider : ISocketClientProvider
     {
         public bool IsConnected { get; private set; }