test: 增加单元测试

Author: ArgoZhang

src/BootstrapBlazor/Services/TcpSocket/TcpSocketClientBase.cs

@@ -39,7 +39,6 @@ public abstract class TcpSocketClientBase(SocketClientOptions options) : ITcpSoc
     /// <summary>
     /// Gets or sets the service provider used to resolve dependencies.
     /// </summary>
-    [NotNull]
     public IServiceProvider? ServiceProvider { get; set; }
 
     /// <summary>
@@ -88,13 +87,15 @@ public virtual void SetDataHandler(IDataPackageHandler handler)
     public async ValueTask<bool> ConnectAsync(IPEndPoint endPoint, CancellationToken token = default)
     {
         var ret = false;
+        SocketClientProvider = ServiceProvider?.GetRequiredService<ISocketClientProvider>()
+            ?? throw new InvalidOperationException("SocketClientProvider is not registered in the service provider.");
+
         try
         {
             // 释放资源
             await CloseAsync();
 
             // 创建新的 TcpClient 实例
-            SocketClientProvider = ServiceProvider.GetRequiredService<ISocketClientProvider>();
             SocketClientProvider.LocalEndPoint = Options.LocalEndPoint;
 
             _localEndPoint = Options.LocalEndPoint;
@@ -123,10 +124,10 @@ public async ValueTask<bool> ConnectAsync(IPEndPoint endPoint, CancellationToken
         }
         catch (OperationCanceledException ex)
         {
-            //var message = token.IsCancellationRequested
-            //    ? $"TCP Socket connect operation was canceled from {LocalEndPoint} to {endPoint}"
-            //    : $"TCP Socket connect operation timed out from {LocalEndPoint} to {endPoint}";
-            //Log(LogLevel.Warning, ex, message);
+            var message = token.IsCancellationRequested
+                ? $"TCP Socket connect operation was canceled from {LocalEndPoint} to {endPoint}"
+                : $"TCP Socket connect operation timed out from {LocalEndPoint} to {endPoint}";
+            Log(LogLevel.Warning, ex, message);
         }
         catch (Exception ex)
         {
@@ -275,8 +276,8 @@ private async ValueTask<int> ReceiveCoreAsync(ISocketClientProvider client, Memo
     /// </summary>
     protected void Log(LogLevel logLevel, Exception? ex, string? message)
     {
-        Logger ??= ServiceProvider.GetRequiredService<ILogger<TcpSocketClientBase>>();
-        Logger.Log(logLevel, ex, "{Message}", message);
+        Logger ??= ServiceProvider?.GetRequiredService<ILogger<TcpSocketClientBase>>();
+        Logger?.Log(logLevel, ex, "{Message}", message);
     }
 
     /// <summary>

test/UnitTest/Services/TcpSocketFactoryTest.cs

@@ -77,6 +77,26 @@ public async Task ConnectAsync_Failed()
         Assert.False(connect);
     }
 
+    [Fact]
+    public async Task ConnectAsync_Error()
+    {
+        var client = CreateClient();
+
+        // 反射设置 SocketClientProvider 为空
+        var propertyInfo = client.GetType().GetProperty("ServiceProvider", BindingFlags.Public | BindingFlags.Instance);
+        Assert.NotNull(propertyInfo);
+        propertyInfo.SetValue(client, null);
+
+        // 测试 ConnectAsync 方法连接失败
+        var ex = await Assert.ThrowsAsync<InvalidOperationException>(async () => await client.ConnectAsync("localhost", 9999));
+        Assert.NotNull(ex);
+
+        // 反射测试 Log 方法
+        var methodInfo = client.GetType().GetMethod("Log", BindingFlags.NonPublic | BindingFlags.Instance);
+        Assert.NotNull(methodInfo);
+        methodInfo.Invoke(client, [LogLevel.Error, null!, "Test error log"]);
+    }
+
     [Fact]
     public async Task Send_Timeout()
     {