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); }