public async Task Should_connect_and_disconnect()
{
TcpTransport transport = CreateTcpTransport();
transport.State.Should().Be(TransportState.Disconnected);
transport.IsConnected.Should().BeFalse();
await transport.ConnectAsync();
Socket clientSocket = _serverSocket.Accept();
clientSocket.Should().NotBeNull();
clientSocket.IsConnected().Should().BeTrue();
transport.State.Should().Be(TransportState.Connected);
transport.IsConnected.Should().BeTrue();
await transport.DisconnectAsync();
transport.Dispose();
clientSocket.IsConnected().Should().BeFalse();
transport.State.Should().Be(TransportState.Disconnected);
transport.IsConnected.Should().BeFalse();
}
public async Task Should_process_lost_connection()
{
TcpTransport transport = CreateTcpTransport();
transport.State.Should().Be(TransportState.Disconnected);
await transport.ConnectAsync();
Socket clientSocket = _serverSocket.Accept();
clientSocket.Should().NotBeNull();
clientSocket.IsConnected().Should().BeTrue();
transport.State.Should().Be(TransportState.Connected);
clientSocket.Shutdown(SocketShutdown.Both);
clientSocket.Disconnect(false);
clientSocket.Close();
await Task.Delay(200);
transport.State.Should().Be(TransportState.Disconnected);
await transport.DisconnectAsync();
transport.State.Should().Be(TransportState.Disconnected);
}
public async Task Should_receive_multiple_packets()
{
TcpTransport transport = CreateTcpTransport();
var receivedMessages = new AsyncProducerConsumerQueue <byte[]>();
transport.Subscribe(receivedMessages.Enqueue);
await transport.ConnectAsync();
Socket clientSocket = _serverSocket.Accept();
byte[] payload1 = Enumerable.Range(0, 10).Select(i => (byte)i).ToArray();
var packet1 = new TcpTransportPacket(0, payload1);
byte[] payload2 = Enumerable.Range(11, 40).Select(i => (byte)i).ToArray();
var packet2 = new TcpTransportPacket(1, payload2);
byte[] payload3 = Enumerable.Range(51, 205).Select(i => (byte)i).ToArray();
var packet3 = new TcpTransportPacket(2, payload3);
byte[] allData = ArrayUtils.Combine(packet1.Data, packet2.Data, packet3.Data);
byte[] dataPart1;
byte[] dataPart2;
allData.Split(50, out dataPart1, out dataPart2);
clientSocket.Send(dataPart1);
await Task.Delay(100);
clientSocket.Send(dataPart2);
await Task.Delay(100);
byte[] receivedData1 = await receivedMessages.DequeueAsync(CancellationTokenHelpers.Timeout(1000).Token);
receivedData1.Should().BeEquivalentTo(payload1);
byte[] receivedData2 = await receivedMessages.DequeueAsync(CancellationTokenHelpers.Timeout(1000).Token);
receivedData2.Should().BeEquivalentTo(payload2);
byte[] receivedData3 = await receivedMessages.DequeueAsync(CancellationTokenHelpers.Timeout(1000).Token);
receivedData3.Should().BeEquivalentTo(payload3);
await transport.DisconnectAsync();
clientSocket.Close();
}
public async Task Should_send()
{
TcpTransport transport = CreateTcpTransport();
await transport.ConnectAsync();
Socket clientSocket = _serverSocket.Accept();
byte[] payload = "010203040506070809".HexToBytes();
await transport.SendAsync(payload);
var buffer = new byte[30];
var received = clientSocket.Receive(buffer);
var packet = new TcpTransportPacket(buffer, 0, received);
var receivedPayload = packet.GetPayloadCopy();
receivedPayload.ShouldAllBeEquivalentTo(payload);
}
public async Task Should_receive_small_parts_less_than_4_bytes()
{
TcpTransport transport = CreateTcpTransport();
Task <byte[]> receiveTask = transport.FirstAsync().Timeout(TimeSpan.FromMilliseconds(3000)).ToTask();
await transport.ConnectAsync();
Socket clientSocket = _serverSocket.Accept();
byte[] payload = "010203040506070809".HexToBytes();
var packet = new TcpTransportPacket(0x0ABBCCDD, payload);
byte[] part1 = packet.Data.Take(1).ToArray();
byte[] part2 = packet.Data.Skip(part1.Length).Take(2).ToArray();
byte[] part3 = packet.Data.Skip(part1.Length + part2.Length).Take(3).ToArray();
byte[] part4 = packet.Data.Skip(part1.Length + part2.Length + part3.Length).ToArray();
clientSocket.Send(part1);
await Task.Delay(100);
clientSocket.Send(part2);
await Task.Delay(200);
clientSocket.Send(part3);
await Task.Delay(50);
clientSocket.Send(part4);
await Task.Delay(50);
byte[] receivedData = await receiveTask;
receivedData.Should().BeEquivalentTo(payload);
await transport.DisconnectAsync();
clientSocket.Close();
}
public async Task Should_receive_big_payload()
{
TcpTransport transport = CreateTcpTransport();
Task <byte[]> receiveTask = transport.FirstAsync().Timeout(TimeSpan.FromMilliseconds(1000)).ToTask();
await transport.ConnectAsync();
Socket clientSocket = _serverSocket.Accept();
byte[] payload = Enumerable.Range(0, 255).Select(i => (byte)i).ToArray();
var packet = new TcpTransportPacket(0x0FA0B1C2, payload);
clientSocket.Send(packet.Data);
byte[] receivedData = await receiveTask;
receivedData.Should().BeEquivalentTo(payload);
await transport.DisconnectAsync();
clientSocket.Close();
}
public async Task Should_receive()
{
TcpTransport transport = CreateTcpTransport();
Task <byte[]> receiveTask = transport.FirstAsync().Timeout(TimeSpan.FromMilliseconds(1000)).ToTask();
await transport.ConnectAsync();
Socket clientSocket = _serverSocket.Accept();
byte[] payload = "010203040506070809".HexToBytes();
var packet = new TcpTransportPacket(0x0FA0B1C2, payload);
clientSocket.Send(packet.Data);
byte[] receivedData = await receiveTask;
receivedData.Should().BeEquivalentTo(payload);
await transport.DisconnectAsync();
clientSocket.Close();
}