Add WebSockets (#1423)

* Add WebSockets

* Add tests

* fix

* more tests

* Add tests

* ...

* remove IOwin

* -

* tests

* fluent

* ok

* match

* .

* byte[]

* x

* func

* func

* byte

* trans

* ...

* frameworks.........

* jmes

* xxx

* sc

test/WireMock.Net.Tests/WebSockets/ClientWebSocketExtensions.cs

@@ -0,0 +1,53 @@
+// Copyright © WireMock.Net
+
+using System.Buffers;
+using System.Net.WebSockets;
+using System.Text;
+
+namespace WireMock.Net.Tests.WebSockets;
+
+internal static class ClientWebSocketExtensions
+{
+    internal static Task SendAsync(this ClientWebSocket client, string text, bool endOfMessage = true, CancellationToken cancellationToken = default)
+    {
+        return client.SendAsync(new ArraySegment<byte>(Encoding.UTF8.GetBytes(text)), WebSocketMessageType.Text, endOfMessage, cancellationToken);
+    }
+
+    internal static async Task<string> ReceiveAsTextAsync(this ClientWebSocket client, int bufferSize = 1024, CancellationToken cancellationToken = default)
+    {
+        using var receiveBuffer = ArrayPool<byte>.Shared.Lease(1024);
+        var result = await client.ReceiveAsync(new ArraySegment<byte>(receiveBuffer), cancellationToken);
+
+        if (result.MessageType != WebSocketMessageType.Text)
+        {
+            throw new InvalidOperationException($"Expected a text message but received a {result.MessageType} message.");
+        }
+
+        if (!result.EndOfMessage)
+        {
+            throw new InvalidOperationException("Received message is too large for the buffer. Consider increasing the buffer size.");
+        }
+
+        return Encoding.UTF8.GetString(receiveBuffer, 0, result.Count);
+    }
+
+    internal static async Task<byte[]> ReceiveAsBytesAsync(this ClientWebSocket client, int bufferSize = 1024, CancellationToken cancellationToken = default)
+    {
+        using var receiveBuffer = ArrayPool<byte>.Shared.Lease(1024);
+        var result = await client.ReceiveAsync(new ArraySegment<byte>(receiveBuffer), cancellationToken);
+
+        if (result.MessageType != WebSocketMessageType.Binary)
+        {
+            throw new InvalidOperationException($"Expected a binary message but received a {result.MessageType} message.");
+        }
+
+        if (!result.EndOfMessage)
+        {
+            throw new InvalidOperationException("Received message is too large for the buffer. Consider increasing the buffer size.");
+        }
+
+        var receivedData = new byte[result.Count];
+        Array.Copy(receiveBuffer, receivedData, result.Count);
+        return receivedData;
+    }
+}