test: 更新单元测试

Author: ArgoZhang

test/UnitTest/Services/TcpSocketFactoryTest.cs

@@ -7,6 +7,7 @@
 using System.Net;
 using System.Net.Sockets;
 using System.Text;
+using static System.Runtime.InteropServices.JavaScript.JSType;
 
 namespace UnitTest.Services;
 
@@ -24,17 +25,17 @@ public async Task GetOrCreate_Ok()
         sc.AddBootstrapBlazorTcpSocketFactory();
         var provider = sc.BuildServiceProvider();
         var factory = provider.GetRequiredService<ITcpSocketFactory>();
-        var client1 = factory.GetOrCreate("demo", key => Utility.ConvertToIpEndPoint("localhost", 0));
+        var client1 = factory.GetOrCreate("demo", op => op.LocalEndPoint = Utility.ConvertToIpEndPoint("localhost", 0));
         await client1.CloseAsync();
 
-        var client2 = factory.GetOrCreate("demo", key => Utility.ConvertToIpEndPoint("localhost", 0));
+        var client2 = factory.GetOrCreate("demo", op => op.LocalEndPoint = Utility.ConvertToIpEndPoint("localhost", 0));
         Assert.Equal(client1, client2);
 
         var ip = Dns.GetHostAddresses(Dns.GetHostName(), AddressFamily.InterNetwork).FirstOrDefault() ?? IPAddress.Loopback;
-        var client3 = factory.GetOrCreate("demo1", key => Utility.ConvertToIpEndPoint(ip.ToString(), 0));
+        var client3 = factory.GetOrCreate("demo1", op => op.LocalEndPoint = Utility.ConvertToIpEndPoint(ip.ToString(), 0));
 
         // 测试不合格 IP 地址
-        var client4 = factory.GetOrCreate("demo2", key => Utility.ConvertToIpEndPoint("256.0.0.1", 0));
+        var client4 = factory.GetOrCreate("demo2", op => op.LocalEndPoint = Utility.ConvertToIpEndPoint("256.0.0.1", 0));
 
         var client5 = factory.Remove("demo2");
         Assert.Equal(client4, client5);
@@ -102,6 +103,17 @@ public async Task SendAsync_Error()
         var data = new ReadOnlyMemory<byte>([1, 2, 3, 4, 5]);
         var ex = await Assert.ThrowsAsync<InvalidOperationException>(async () => await client.SendAsync(data));
         Assert.NotNull(ex);
+
+        // 测试发送失败
+        var port = 8892;
+        var server = StartTcpServer(port, MockSplitPackageAsync);
+
+        client.SetDataHandler(new MockSendErrorHandler());
+        await client.ConnectAsync("localhost", port);
+        Assert.True(client.IsConnected);
+
+        // 内部生成异常日志
+        await client.SendAsync(data);
     }
 
     [Fact]
@@ -126,7 +138,7 @@ public async Task SendAsync_Cancel()
 
         // 设置延时发送适配器
         // 延时发送期间关闭 Socket 连接导致内部报错
-        client.SetDataHandler(new MockSendErrorHandler()
+        client.SetDataHandler(new MockSendCancelHandler()
         {
             Socket = client
         });
@@ -179,7 +191,10 @@ public async Task ReceiveAsync_Cancel()
         await client.SendAsync(data);
 
         // 通过反射取消令牌
-        var fieldInfo = client.GetType().GetField("_receiveCancellationTokenSource", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance);
+        var baseType = client.GetType().BaseType;
+        Assert.NotNull(baseType);
+
+        var fieldInfo = baseType.GetField("_receiveCancellationTokenSource", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance);
         Assert.NotNull(fieldInfo);
         var tokenSource = fieldInfo.GetValue(client) as CancellationTokenSource;
         Assert.NotNull(tokenSource);
@@ -190,36 +205,21 @@ public async Task ReceiveAsync_Cancel()
     [Fact]
     public async Task ReceiveAsync_InvalidOperationException()
     {
-        var port = 8890;
-        var server = StartTcpServer(port, MockSplitPackageAsync);
-
+        // 未连接时调用 ReceiveAsync 方法会抛出 InvalidOperationException 异常
         var client = CreateClient();
-
-        //未连接
         var ex = await Assert.ThrowsAsync<InvalidOperationException>(async () => await client.ReceiveAsync());
         Assert.NotNull(ex);
 
-        // 反射给 _client 赋值但是未连接
-        var fieldInfo = client.GetType().GetField("_client", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance);
-        Assert.NotNull(fieldInfo);
-        fieldInfo.SetValue(client, new TcpClient());
-        ex = null;
-        ex = await Assert.ThrowsAsync<InvalidOperationException>(async () => await client.ReceiveAsync());
-        Assert.NotNull(ex);
-
-        await client.ConnectAsync("localhost", port);
-        ex = null;
-        ex = await Assert.ThrowsAsync<InvalidOperationException>(async () => await client.ReceiveAsync());
+        // 已连接但是启用了自动接收功能时调用 ReceiveAsync 方法会抛出 InvalidOperationException 异常
+        var port = 8893;
+        var server = StartTcpServer(port, MockSplitPackageAsync);
 
-        await client.CloseAsync();
-        client.IsAutoReceive = false;
+        client.IsAutoReceive = true;
         var connected = await client.ConnectAsync("localhost", port);
         Assert.True(connected);
 
-        var data = new ReadOnlyMemory<byte>([1, 2, 3, 4, 5]);
-        await client.SendAsync(data);
-        var payload = await client.ReceiveAsync();
-        Assert.Equal(payload.ToArray(), [1, 2, 3, 4, 5]);
+        ex = await Assert.ThrowsAsync<InvalidOperationException>(async () => await client.ReceiveAsync());
+        Assert.NotNull(ex);
     }
 
     [Fact]
@@ -230,9 +230,13 @@ public async Task ReceiveAsync_Ok()
 
         var client = CreateClient();
         client.IsAutoReceive = false;
-        await client.ConnectAsync("localhost", port);
+        var connected = await client.ConnectAsync("localhost", port);
+        Assert.True(connected);
+
         var data = new ReadOnlyMemory<byte>([1, 2, 3, 4, 5]);
-        await client.SendAsync(data);
+        var send = await client.SendAsync(data);
+        Assert.True(send);
+
         var payload = await client.ReceiveAsync();
         Assert.Equal(payload.ToArray(), [1, 2, 3, 4, 5]);
     }
@@ -243,7 +247,10 @@ public async Task ReceiveAsync_Error()
         var client = CreateClient();
 
         // 测试未建立连接前调用 ReceiveAsync 方法报异常逻辑
-        var methodInfo = client.GetType().GetMethod("AutoReceiveAsync", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance);
+        var baseType = client.GetType().BaseType;
+        Assert.NotNull(baseType);
+
+        var methodInfo = baseType.GetMethod("AutoReceiveAsync", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance);
         Assert.NotNull(methodInfo);
 
         var task = (ValueTask)methodInfo.Invoke(client, null)!;
@@ -567,7 +574,7 @@ private static ITcpSocketClient CreateClient()
         sc.AddBootstrapBlazorTcpSocketFactory();
         var provider = sc.BuildServiceProvider();
         var factory = provider.GetRequiredService<ITcpSocketFactory>();
-        var client = factory.GetOrCreate("test", key => Utility.ConvertToIpEndPoint("localhost", 0));
+        var client = factory.GetOrCreate("test", op => op.LocalEndPoint = Utility.ConvertToIpEndPoint("localhost", 0));
         return client;
     }
 
@@ -606,6 +613,16 @@ class MockSendErrorHandler : DataPackageHandlerBase
     {
         public ITcpSocketClient? Socket { get; set; }
 
+        public override ValueTask<ReadOnlyMemory<byte>> SendAsync(ReadOnlyMemory<byte> data, CancellationToken token = default)
+        {
+            throw new Exception("Mock send failed");
+        }
+    }
+
+    class MockSendCancelHandler : DataPackageHandlerBase
+    {
+        public ITcpSocketClient? Socket { get; set; }
+
         public override async ValueTask<ReadOnlyMemory<byte>> SendAsync(ReadOnlyMemory<byte> data, CancellationToken token = default)
         {
             if (Socket != null)