public async Task HandleFinalHandshakeFailureCorrectly()
{
var initialHandshakeResponse = "96:0{ \"sid\":\"i4VXx68U4C_w6qiFAAAm\",\"upgrades\":[\"websocket\"],\"pingInterval\":25000,\"pingTimeout\":5000}2:40";
var initialHandshakeClient = new MockHttpClient(initialHandshakeResponse, HttpStatusCode.OK);
var secondaryHandshakeResponse = "ok";
var secondaryHandshakeClient = new MockHttpClient(secondaryHandshakeResponse, HttpStatusCode.OK);
var protocolUpgradeResponse = "";
var protocolUpgradeClient = new MockHttpClient(protocolUpgradeResponse, HttpStatusCode.SwitchingProtocols);
protocolUpgradeClient.AddHeader("Sec-Websocket-Accept", "RWZIQcMHYHEyvemvvKIkivC1Tvo=");
var finalHandshakeResponse = "1:6";
var finalHandshakeClient = new MockHttpClient(finalHandshakeResponse, HttpStatusCode.RequestTimeout);
var factoryMock = new Mock <IHttpClientFactory>();
factoryMock.SetupSequence(f => f.CreateHttpClient())
.Returns(initialHandshakeClient)
.Returns(secondaryHandshakeClient)
.Returns(protocolUpgradeClient)
.Returns(finalHandshakeClient);
var socketFactoryMock = new Mock <IClientSocketFactory>();
var mediator = new SocketMediator("ws://localhost:7200/collections");
await AsyncAssert.Throws <HttpRequestException>(
() => mediator.InitConnection(factoryMock.Object, socketFactoryMock.Object));
Mock.VerifyAll(factoryMock);
}
public async Task Handles2ProbeSendFailureCorrectly()
{
// 1. Initial handshake
var initialHandshakeResponse = "96:0{ \"sid\":\"i4VXx68U4C_w6qiFAAAm\",\"upgrades\":[\"websocket\"],\"pingInterval\":25000,\"pingTimeout\":5000}2:40";
var initialHandshakeClient = new MockHttpClient(initialHandshakeResponse, HttpStatusCode.OK);
var secondaryHandshakeResponse = "ok";
var secondaryHandshakeClient = new MockHttpClient(secondaryHandshakeResponse, HttpStatusCode.OK);
var protocolUpgradeResponse = "";
var protocolUpgradeClient = new MockHttpClient(protocolUpgradeResponse, HttpStatusCode.SwitchingProtocols);
protocolUpgradeClient.AddHeader("Sec-Websocket-Accept", "RWZIQcMHYHEyvemvvKIkivC1Tvo=");
var finalHandshakeResponse = "15:40/collections,";
var finalHandshakeClient = new MockHttpClient(finalHandshakeResponse, HttpStatusCode.OK);
var factoryMock = new Mock <IHttpClientFactory>();
factoryMock.SetupSequence(f => f.CreateHttpClient())
.Returns(initialHandshakeClient)
.Returns(secondaryHandshakeClient)
.Returns(protocolUpgradeClient)
.Returns(finalHandshakeClient);
// 2. Connection upgrade and handshake
var socketMock = new Mock <IClientSocket>();
socketMock.Setup(s => s.ConnectAsync(It.IsAny <Uri>(), It.IsAny <CancellationToken>()))
.Returns(Task.CompletedTask);
// The ws client handshake sends "2probe" and the server returns "3probe"
socketMock.Setup(s => s.SendAsync(It.IsAny <ArraySegment <byte> >(),
It.IsAny <WebSocketMessageType>(), It.IsAny <bool>(), It.IsAny <CancellationToken>()))
.Returns(Task.FromException(new HttpRequestException()));
var socketFactoryMock = new Mock <IClientSocketFactory>();
socketFactoryMock.Setup(f => f.CreateSocketClient()).Returns(socketMock.Object);
var mediator = new SocketMediator("ws://localhost:7200/collections");
await AsyncAssert.Throws <HttpRequestException>(
() => mediator.InitConnection(factoryMock.Object, socketFactoryMock.Object));
Mock.VerifyAll(socketMock, socketFactoryMock, factoryMock);
}
public async Task LoadsAndStreamsDataCorrectly()
{
// 1. Initial handshake
var initialHandshakeResponse = "96:0{ \"sid\":\"i4VXx68U4C_w6qiFAAAm\",\"upgrades\":[\"websocket\"],\"pingInterval\":25000,\"pingTimeout\":5000}2:40";
var initialHandshakeClient = new MockHttpClient(initialHandshakeResponse, HttpStatusCode.OK);
var secondaryHandshakeResponse = "ok";
var secondaryHandshakeClient = new MockHttpClient(secondaryHandshakeResponse, HttpStatusCode.OK);
var protocolUpgradeResponse = "";
var protocolUpgradeClient = new MockHttpClient(protocolUpgradeResponse, HttpStatusCode.SwitchingProtocols);
protocolUpgradeClient.AddHeader("Sec-Websocket-Accept", "RWZIQcMHYHEyvemvvKIkivC1Tvo=");
var finalHandshakeResponse = "15:40/collections,";
var finalHandshakeClient = new MockHttpClient(finalHandshakeResponse, HttpStatusCode.OK);
var factoryMock = new Mock <IHttpClientFactory>();
factoryMock.SetupSequence(f => f.CreateHttpClient())
.Returns(initialHandshakeClient)
.Returns(secondaryHandshakeClient)
.Returns(protocolUpgradeClient)
.Returns(finalHandshakeClient);
var initialProbeResponse = Encoding.UTF8.GetBytes("3probe");
var secondaryProbeResponse = Encoding.UTF8.GetBytes("5");
// 2. Connection upgrade and handshake
var socketMock = new Mock <IClientSocket>();
socketMock.Setup(s => s.ConnectAsync(It.IsAny <Uri>(), It.IsAny <CancellationToken>()))
.Returns(Task.CompletedTask);
// The ws client handshake sends "2probe" and the server returns "3probe"
socketMock.Setup(s => s.SendAsync(It.IsAny <ArraySegment <byte> >(),
It.IsAny <WebSocketMessageType>(), It.IsAny <bool>(), It.IsAny <CancellationToken>()))
.Returns(Task.CompletedTask);
// "3probe" response
List <byte> initialProbeBytes = new List <byte>(initialProbeResponse);
var originalBytes = initialProbeBytes.Count;
var dataProvider = new MockSocketDataProvider(initialProbeBytes);
socketMock.Setup(s => s.ReceiveAsync(It.IsAny <ArraySegment <byte> >(), It.IsAny <CancellationToken>()))
.Returns((ArraySegment <byte> buffer, CancellationToken token) =>
{
var writeResult = dataProvider.GetDataChunk(buffer.Array.Length);
writeResult.buffer.CopyTo(buffer.Array, 0);
return(MockSocketTaskFactory.CreateTask(initialProbeBytes.Count, writeResult.bytesWritten));
});
// This captures the mediator sending the value "5", can return the same thing
socketMock.Setup(s => s.SendAsync(It.IsAny <ArraySegment <byte> >(),
It.IsAny <WebSocketMessageType>(), It.IsAny <bool>(), It.IsAny <CancellationToken>()))
.Returns(Task.CompletedTask);
var socketFactoryMock = new Mock <IClientSocketFactory>();
socketFactoryMock.Setup(f => f.CreateSocketClient()).Returns(socketMock.Object);
var mediator = new SocketMediator("ws://localhost:7200/collections");
await mediator.InitConnection(factoryMock.Object, socketFactoryMock.Object);
Mock.VerifyAll(socketMock, socketFactoryMock, factoryMock);
}