test: 增加 DefaultSocketClient 单元测试

Author: ArgoZhang

src/BootstrapBlazor/Services/TcpSocket/SocketClientBase.cs renamed to src/BootstrapBlazor/Services/TcpSocket/DefaultSocketClient.cs

@@ -13,7 +13,7 @@ namespace BootstrapBlazor.Components;
 /// TcpSocket 客户端默认实现
 /// </summary>
 [UnsupportedOSPlatform("browser")]
-public class SocketClientBase(IPEndPoint localEndPoint) : ISocketClient
+class DefaultSocketClient(IPEndPoint localEndPoint) : ISocketClient
 {
     private TcpClient? _client;
 
@@ -25,15 +25,22 @@ public class SocketClientBase(IPEndPoint localEndPoint) : ISocketClient
     /// <summary>
     /// <inheritdoc/>
     /// </summary>
-    public IPEndPoint LocalEndPoint { get; set; } = new IPEndPoint(IPAddress.Loopback, 0);
+    public IPEndPoint LocalEndPoint { get; set; } = localEndPoint;
 
     /// <summary>
     /// <inheritdoc/>
     /// </summary>
     public async ValueTask<bool> ConnectAsync(IPEndPoint endPoint, CancellationToken token = default)
     {
-        _client = new TcpClient(localEndPoint);
+        _client = new TcpClient(LocalEndPoint);
         await _client.ConnectAsync(endPoint, token);
+        if (_client.Connected)
+        {
+            if (_client.Client.LocalEndPoint is IPEndPoint localEndPoint)
+            {
+                LocalEndPoint = localEndPoint;
+            }
+        }
         return _client.Connected;
     }
 

src/BootstrapBlazor/Services/TcpSocket/DefaultTcpSocketClient.cs

@@ -9,7 +9,7 @@
 namespace BootstrapBlazor.Components;
 
 [UnsupportedOSPlatform("browser")]
-sealed class DefaultTcpSocketClient : TcpSocketClientBase<SocketClientBase>
+sealed class DefaultTcpSocketClient : TcpSocketClientBase<DefaultSocketClient>
 {
     public DefaultTcpSocketClient(SocketClientOptions options)
     {
@@ -27,8 +27,8 @@ public DefaultTcpSocketClient(SocketClientOptions options)
     /// <param name="localEndPoint"></param>
     /// <returns></returns>
     /// <exception cref="NotImplementedException"></exception>
-    protected override SocketClientBase CreateSocketClient(IPEndPoint localEndPoint)
+    protected override DefaultSocketClient CreateSocketClient(IPEndPoint localEndPoint)
     {
-        return new SocketClientBase(localEndPoint);
+        return new DefaultSocketClient(localEndPoint);
     }
 }

test/UnitTest/Services/TcpSocketFactoryTest.cs

@@ -6,6 +6,8 @@
 using Microsoft.Extensions.Logging;
 using System.Net;
 using System.Net.Sockets;
+using System.Reflection;
+using System.Security.AccessControl;
 using System.Text;
 
 namespace UnitTest.Services;
@@ -193,7 +195,7 @@ public async Task ReceiveAsync_Cancel()
         var baseType = client.GetType().BaseType;
         Assert.NotNull(baseType);
 
-        var fieldInfo = baseType.GetField("_receiveCancellationTokenSource", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance);
+        var fieldInfo = baseType.GetField("_receiveCancellationTokenSource", BindingFlags.NonPublic | BindingFlags.Instance);
         Assert.NotNull(fieldInfo);
         var tokenSource = fieldInfo.GetValue(client) as CancellationTokenSource;
         Assert.NotNull(tokenSource);
@@ -249,7 +251,7 @@ public async Task ReceiveAsync_Error()
         var baseType = client.GetType().BaseType;
         Assert.NotNull(baseType);
 
-        var methodInfo = baseType.GetMethod("AutoReceiveAsync", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance);
+        var methodInfo = baseType.GetMethod("AutoReceiveAsync", BindingFlags.NonPublic | BindingFlags.Instance);
         Assert.NotNull(methodInfo);
 
         var task = (ValueTask)methodInfo.Invoke(client, null)!;
@@ -462,9 +464,51 @@ public async Task DelimiterDataPackageHandler_Ok()
     public void Logger_Null()
     {
         // 测试 Logger 为 null 的情况
-        var client = new MockTcpSocketClient();
-        client.TestLog();
-        Assert.Null(client.Logger);
+        var client = CreateClient();
+        var baseType = client.GetType().BaseType;
+        Assert.NotNull(baseType);
+
+        // 获取 Logger 字段设置为 null 测试 Log 不会抛出异常
+        var propertyInfo = baseType.GetProperty("Logger", BindingFlags.Public | BindingFlags.Instance);
+        Assert.NotNull(propertyInfo);
+
+        propertyInfo.SetValue(client, null);
+
+        var methodInfo = baseType.GetMethod("Log", BindingFlags.NonPublic | BindingFlags.Instance);
+        Assert.NotNull(methodInfo);
+        methodInfo.Invoke(client, [LogLevel.Information, null!, "Test log message"]);
+    }
+
+    [Fact]
+    public async Task DefaultSocketClient_Ok()
+    {
+        var port = 8894;
+        var server = StartTcpServer(port, MockDelimiterPackageAsync);
+        var client = CreateClient();
+
+        // 获得 Client 泛型属性
+        var baseType = client.GetType().BaseType;
+        Assert.NotNull(baseType);
+
+        // 建立连接
+        var connect = await client.ConnectAsync("localhost", port);
+        Assert.True(connect);
+
+        var propertyInfo = baseType.GetProperty("Client", BindingFlags.NonPublic | BindingFlags.Instance);
+        Assert.NotNull(propertyInfo);
+        var instance = propertyInfo.GetValue(client);
+        Assert.NotNull(instance);
+
+        ISocketClient socketClient = (ISocketClient)instance;
+        Assert.NotNull(socketClient);
+        Assert.True(socketClient.IsConnected);
+
+        await socketClient.CloseAsync();
+        Assert.False(socketClient.IsConnected);
+
+        var buffer = new byte[10];
+        var len = await socketClient.ReceiveAsync(buffer);
+        Assert.Equal(0, len);
     }
 
     private static TcpListener StartTcpServer(int port, Func<TcpClient, Task> handler)
@@ -677,21 +721,4 @@ public override async ValueTask ReceiveAsync(ReadOnlyMemory<byte> data, Cancella
             await base.ReceiveAsync(data, token);
         }
     }
-
-    class MockSoketClient(IPEndPoint localEndPoint) : SocketClientBase(localEndPoint)
-    {
-    }
-
-    class MockTcpSocketClient : TcpSocketClientBase<MockSoketClient>
-    {
-        protected override MockSoketClient CreateSocketClient(IPEndPoint localEndPoint)
-        {
-            return new MockSoketClient(localEndPoint);
-        }
-
-        public void TestLog()
-        {
-            Log(LogLevel.Information, null, "test");
-        }
-    }
 }