aspnetcore/test/Microsoft.AspNetCore.WebSockets.Protocol.Test/SendReceiveTests.cs

using System;
using System.Linq;
using System.Net.WebSockets;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using Xunit;

namespace Microsoft.AspNetCore.WebSockets.Protocol.Test
{
    public class SendReceiveTests
    {
        [Fact]
        public async Task ClientToServerTextMessage()
        {
            const string message = "Hello, World!";

            var pair = WebSocketPair.Create();
            var sendBuffer = Encoding.UTF8.GetBytes(message);

            await pair.ClientSocket.SendAsync(new ArraySegment<byte>(sendBuffer), WebSocketMessageType.Text, endOfMessage: true, cancellationToken: CancellationToken.None);

            var receiveBuffer = new byte[32];
            var result = await pair.ServerSocket.ReceiveAsync(new ArraySegment<byte>(receiveBuffer), CancellationToken.None);

            Assert.Equal(WebSocketMessageType.Text, result.MessageType);
            Assert.Equal(message, Encoding.UTF8.GetString(receiveBuffer, 0, result.Count));
        }

        [Fact]
        public async Task ServerToClientTextMessage()
        {
            const string message = "Hello, World!";

            var pair = WebSocketPair.Create();
            var sendBuffer = Encoding.UTF8.GetBytes(message);

            await pair.ServerSocket.SendAsync(new ArraySegment<byte>(sendBuffer), WebSocketMessageType.Text, endOfMessage: true, cancellationToken: CancellationToken.None);

            var receiveBuffer = new byte[32];
            var result = await pair.ClientSocket.ReceiveAsync(new ArraySegment<byte>(receiveBuffer), CancellationToken.None);

            Assert.Equal(WebSocketMessageType.Text, result.MessageType);
            Assert.Equal(message, Encoding.UTF8.GetString(receiveBuffer, 0, result.Count));
        }

        [Fact]
        public async Task ClientToServerBinaryMessage()
        {
            var pair = WebSocketPair.Create();
            var sendBuffer = new byte[] { 0xde, 0xad, 0xbe, 0xef };

            await pair.ClientSocket.SendAsync(new ArraySegment<byte>(sendBuffer), WebSocketMessageType.Binary, endOfMessage: true, cancellationToken: CancellationToken.None);

            var receiveBuffer = new byte[32];
            var result = await pair.ServerSocket.ReceiveAsync(new ArraySegment<byte>(receiveBuffer), CancellationToken.None);

            Assert.Equal(WebSocketMessageType.Binary, result.MessageType);
            Assert.Equal(sendBuffer, receiveBuffer.Take(result.Count).ToArray());
        }

        [Fact]
        public async Task ServerToClientBinaryMessage()
        {
            var pair = WebSocketPair.Create();
            var sendBuffer = new byte[] { 0xde, 0xad, 0xbe, 0xef };

            await pair.ServerSocket.SendAsync(new ArraySegment<byte>(sendBuffer), WebSocketMessageType.Binary, endOfMessage: true, cancellationToken: CancellationToken.None);

            var receiveBuffer = new byte[32];
            var result = await pair.ClientSocket.ReceiveAsync(new ArraySegment<byte>(receiveBuffer), CancellationToken.None);

            Assert.Equal(WebSocketMessageType.Binary, result.MessageType);
            Assert.Equal(sendBuffer, receiveBuffer.Take(result.Count).ToArray());
        }
    }
}