public void TestLargeMessage()
{
ObjectPool <byte[]> pool = new ObjectPool <byte[]>(() => { return(new byte[1024]); });
DummySockNetChannel channel = new DummySockNetChannel()
{
State = null,
IsActive = true,
BufferPool = pool
};
channel.Pipe = new SockNetChannelPipe(channel);
WebSocketClientSockNetChannelModule module = new WebSocketClientSockNetChannelModule("/test", "test", null);
channel.AddModule(module);
channel.Connect();
object sent = null;
channel.outgoing.TryTake(out sent, 5000);
Assert.IsTrue(sent is HttpRequest);
HttpResponse handshakeResponse = null;
using (HttpRequest request = (HttpRequest)sent)
{
handshakeResponse = new HttpResponse(null)
{
Code = "200", Reason = "OK", Version = "HTTP/1.1"
};
handshakeResponse.Header[WebSocketUtil.WebSocketAcceptHeader] = WebSocketUtil.GenerateAccept(request.Header[WebSocketUtil.WebSocketKeyHeader]);
}
object receiveResponse = handshakeResponse;
channel.Receive(ref receiveResponse);
Random random = new Random(this.GetHashCode() ^ (int)DateTime.Now.Subtract(new DateTime(2000, 1, 1)).TotalMilliseconds);
for (int n = 0; n < 100; n++)
{
byte[] data = new byte[random.Next(50000, 150000)];
random.NextBytes(data);
WebSocketFrame frame = WebSocketFrame.CreateBinaryFrame(data, false);
using (ChunkedBuffer buffer = ToBuffer(frame))
{
receiveResponse = buffer;
channel.Receive(ref receiveResponse);
Assert.IsTrue(receiveResponse is WebSocketFrame);
Assert.AreEqual(data.Length, ((WebSocketFrame)receiveResponse).Data.Length);
for (int i = 0; i < data.Length; i++)
{
Assert.AreEqual(data[i], ((WebSocketFrame)receiveResponse).Data[i]);
}
buffer.Close();
}
}
Console.WriteLine("Pool stats: " + pool.ObjectsInPool + "/" + pool.TotalNumberOfObjects);
}
public void TestContinuation()
{
ObjectPool <byte[]> pool = new ObjectPool <byte[]>(() => { return(new byte[1024]); });
DummySockNetChannel channel = new DummySockNetChannel()
{
State = null,
IsActive = true,
BufferPool = pool
};
channel.Pipe = new SockNetChannelPipe(channel);
WebSocketClientSockNetChannelModule module = new WebSocketClientSockNetChannelModule("/test", "test", null);
channel.AddModule(module);
channel.Connect();
object sent = null;
channel.outgoing.TryTake(out sent, 5000);
Assert.IsTrue(sent is HttpRequest);
HttpRequest request = (HttpRequest)sent;
HttpResponse handshakeResponse = new HttpResponse(channel.BufferPool)
{
Code = "200", Reason = "OK", Version = "HTTP/1.1"
};
handshakeResponse.Header[WebSocketUtil.WebSocketAcceptHeader] = WebSocketUtil.GenerateAccept(request.Header[WebSocketUtil.WebSocketKeyHeader]);
object receiveResponse = handshakeResponse;
channel.Receive(ref receiveResponse);
WebSocketFrame continuation1 = WebSocketFrame.CreateTextFrame("This ", false, false, false);
ChunkedBuffer buffer = ToBuffer(continuation1);
receiveResponse = buffer;
channel.Receive(ref receiveResponse);
buffer.Close();
Assert.IsTrue(receiveResponse is ChunkedBuffer);
WebSocketFrame continuation2 = WebSocketFrame.CreateTextFrame("is ", false, true, false);
buffer = ToBuffer(continuation2);
receiveResponse = buffer;
channel.Receive(ref receiveResponse);
buffer.Close();
Assert.IsTrue(receiveResponse is ChunkedBuffer);
WebSocketFrame continuation3 = WebSocketFrame.CreateTextFrame("awesome!", false, true, true);
buffer = ToBuffer(continuation3);
receiveResponse = buffer;
channel.Receive(ref receiveResponse);
buffer.Close();
Assert.IsTrue(receiveResponse is WebSocketFrame);
Assert.AreEqual("This is awesome!", ((WebSocketFrame)receiveResponse).DataAsString);
Console.WriteLine("Pool stats: " + pool.ObjectsInPool + "/" + pool.TotalNumberOfObjects);
}