test: add socket test snapshots

Author: ikesnowy

src/Cnblogs.DashScope.Core/DashScopeClientWebSocket.cs

@@ -107,7 +107,7 @@ public void ResetOutput()
     public Task SendMessageAsync<TInput, TParameter>(
         DashScopeWebSocketRequest<TInput, TParameter> request,
         CancellationToken cancellationToken = default)
-        where TInput : class
+        where TInput : class, new()
         where TParameter : class
     {
         if (State == DashScopeWebSocketState.Closed)
@@ -222,6 +222,10 @@ public async Task CloseAsync(CancellationToken cancellationToken = default)
     {
         await _socket.CloseAsync(WebSocketCloseStatus.NormalClosure, "Closing", cancellationToken);
         State = DashScopeWebSocketState.Closed;
+        if (_receiveTask != null)
+        {
+            await _receiveTask;
+        }
     }
 
     private void Dispose(bool disposing)

src/Cnblogs.DashScope.Core/DashScopeClientWebSocketPool.cs

@@ -20,6 +20,15 @@ public DashScopeClientWebSocketPool(DashScopeOptions options)
         _options = options;
     }
 
+    internal DashScopeClientWebSocketPool(IEnumerable<DashScopeClientWebSocket> sockets)
+    {
+        _options = new DashScopeOptions();
+        foreach (var socket in sockets)
+        {
+            _available.Add(socket);
+        }
+    }
+
     internal void ReturnSocketAsync(DashScopeClientWebSocket socket)
     {
         if (socket.State != DashScopeWebSocketState.Ready)

src/Cnblogs.DashScope.Core/DashScopeClientWebSocketWrapper.cs

@@ -37,7 +37,7 @@ public sealed record DashScopeClientWebSocketWrapper(DashScopeClientWebSocket So
     public Task SendMessageAsync<TInput, TParameter>(
         DashScopeWebSocketRequest<TInput, TParameter> request,
         CancellationToken cancellationToken = default)
-        where TInput : class
+        where TInput : class, new()
         where TParameter : class
         => Socket.SendMessageAsync(request, cancellationToken);
 

src/Cnblogs.DashScope.Core/DashScopeWebSocketRequest.cs

@@ -6,7 +6,7 @@
 /// <typeparam name="TInput">Type of the input.</typeparam>
 /// <typeparam name="TParameter">Type of the parameter.</typeparam>
 public class DashScopeWebSocketRequest<TInput, TParameter>
-    where TInput : class
+    where TInput : class, new()
     where TParameter : class
 {
     /// <summary>

src/Cnblogs.DashScope.Core/DashScopeWebSocketRequestPayload.cs

@@ -6,7 +6,7 @@
 /// <typeparam name="TInput">Type of the input.</typeparam>
 /// <typeparam name="TParameter">Type of the parameter.</typeparam>
 public class DashScopeWebSocketRequestPayload<TInput, TParameter>
-    where TInput : class
+    where TInput : class, new() // Input's default value must be empty object(not null or omitted).
     where TParameter : class
 {
     /// <summary>
@@ -37,5 +37,5 @@ public class DashScopeWebSocketRequestPayload<TInput, TParameter>
     /// <summary>
     /// The input of the request.
     /// </summary>
-    public TInput Input { get; set; } = null!;
+    public TInput Input { get; set; } = new();
 }

src/Cnblogs.DashScope.Core/Internals/Assembly.cs

@@ -1,4 +1,5 @@
 using System.Runtime.CompilerServices;
 
 [assembly: InternalsVisibleTo("Cnblogs.DashScope.Sdk.UnitTests")]
+[assembly: InternalsVisibleTo("Cnblogs.DashScope.Tests.Shared")]
 [assembly: InternalsVisibleTo("DynamicProxyGenAssembly2")]

src/Cnblogs.DashScope.Core/SpeechSynthesizerParameters.cs

@@ -39,4 +39,9 @@ public class SpeechSynthesizerParameters
     /// Pitch of the voice, range between 0.5~2, defaults to 1.0.
     /// </summary>
     public float? Pitch { get; set; }
+
+    /// <summary>
+    /// Enable SSML, you can only send text once if enabled.
+    /// </summary>
+    public bool? EnableSsml { get; set; }
 }

test/Cnblogs.DashScope.Sdk.UnitTests/DashScopeClientWebSocketTests.cs

@@ -3,6 +3,7 @@
 using System.Reflection;
 using Cnblogs.DashScope.Core;
 using Cnblogs.DashScope.Core.Internals;
+using Cnblogs.DashScope.Tests.Shared.Utils;
 using NSubstitute;
 
 namespace Cnblogs.DashScope.Sdk.UnitTests;
@@ -102,6 +103,56 @@ public async Task ResetOutput_WithInitialOutput_CompleteThenCreateNewOutputAsync
         Assert.NotSame(oldSignal, client.TaskStarted);
     }
 
+    [Fact]
+    public async Task SendMessageAsync_SocketClosed_ThrowAsync()
+    {
+        // Arrange
+        var socket = Substitute.For<IClientWebSocket>();
+        var client = new DashScopeClientWebSocket(socket);
+        var snapshot = Snapshots.SpeechSynthesizer.RunTask;
+        await client.CloseAsync();
+
+        // Act
+        var act = () => client.SendMessageAsync(snapshot.Message);
+
+        // Assert
+        await Assert.ThrowsAsync<InvalidOperationException>(act);
+    }
+
+    [Fact]
+    public async Task SendMessageAsync_Connected_SendAsync()
+    {
+        // Arrange
+        var socket = Substitute.For<IClientWebSocket>();
+        var client = new DashScopeClientWebSocket(socket);
+        var snapshot = Snapshots.SpeechSynthesizer.RunTask;
+
+        // Act
+        await client.ConnectAsync<SpeechSynthesizerOutput>(new Uri(DashScopeDefaults.WebsocketApiBaseAddress));
+        await client.SendMessageAsync(snapshot.Message);
+
+        // Assert
+        await socket.Received().SendAsync(
+            Arg.Is<ArraySegment<byte>>(s => Checkers.IsJsonEquivalent(s, snapshot.GetRequestJson())),
+            WebSocketMessageType.Text,
+            true,
+            Arg.Any<CancellationToken>());
+    }
+
+    [Fact]
+    public async Task ReceiveMessageAsync_ServerClosed_CloseAsync()
+    {
+        // Arrange
+        var (_, dashScopeClientWebSocket, server) = await Sut.GetSocketTestClientAsync<SpeechSynthesizerOutput>();
+
+        // Act
+        await server.WriteServerCloseAsync();
+
+        // Assert
+        Assert.Equal(DashScopeWebSocketState.Closed, dashScopeClientWebSocket.State);
+        Assert.Equal(WebSocketCloseStatus.NormalClosure, server.CloseStatus);
+    }
+
     private static WebHeaderCollection ExtractHeaders(DashScopeClientWebSocket socket)
     {
         var obj = InnerSocketInfo.GetValue(socket);

test/Cnblogs.DashScope.Tests.Shared/Assembly.cs

@@ -0,0 +1,4 @@
+using System.Runtime.CompilerServices;
+
+[assembly: InternalsVisibleTo("Cnblogs.DashScope.Sdk.UnitTests")]
+[assembly: InternalsVisibleTo("Cnblogs.DashScope.AI.UnitTests")]

test/Cnblogs.DashScope.Tests.Shared/RawHttpData/socket-speech-synthesizer.continue-task.json

@@ -0,0 +1,11 @@
+{
+  "header": {
+    "action": "continue-task",
+    "task_id": "439e0616-2f5b-44e0-8872-0002a066a49c"
+  },
+  "payload": {
+    "input": {
+      "text": "代码改变世界"
+    }
+  }
+}