starred/WireMock.Net-wiremock
1
0
Fork 0
mirror of https://github.com/wiremock/WireMock.Net.git synced 2026-03-25 10:31:01 +01:00
Code Issues 32 Packages Projects Releases 10 Wiki Activity

byte[]

Browse Source
This commit is contained in:
Stef Heyenrath
2026-02-11 22:39:41 +01:00
parent 547b5673c0
commit 1bac18aeb1
5 changed files with 40 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test/WireMock.Net.Tests/WebSockets/ClientWebSocketExtensions.cs
View File

@@ -1,5 +1,6 @@
// Copyright © WireMock.Net
using System.Buffers;
using System.Net.WebSockets;
using System.Text;
@@ -12,9 +13,9 @@ internal static class ClientWebSocketExtensions
return client.SendAsync(new ArraySegment<byte>(Encoding.UTF8.GetBytes(text)), WebSocketMessageType.Text, endOfMessage, cancellationToken);
}
internal static async Task<string> ReceiveAsTextAsync(this ClientWebSocket client, CancellationToken cancellationToken = default)
internal static async Task<string> ReceiveAsTextAsync(this ClientWebSocket client, int bufferSize = 1024, CancellationToken cancellationToken = default)
{
var receiveBuffer = new byte[1024];
using var receiveBuffer = ArrayPool<byte>.Shared.Lease(1024);
var result = await client.ReceiveAsync(new ArraySegment<byte>(receiveBuffer), cancellationToken);
if (result.MessageType != WebSocketMessageType.Text)
@@ -32,7 +33,7 @@ internal static class ClientWebSocketExtensions
internal static async Task<byte[]> ReceiveAsBytesAsync(this ClientWebSocket client, int bufferSize = 1024, CancellationToken cancellationToken = default)
{
var receiveBuffer = new byte[bufferSize];
using var receiveBuffer = ArrayPool<byte>.Shared.Lease(1024);
var result = await client.ReceiveAsync(new ArraySegment<byte>(receiveBuffer), cancellationToken);
if (result.MessageType != WebSocketMessageType.Binary)