public void TestPing()
{
ObjectPool <byte[]> pool = new ObjectPool <byte[]>(() => { return(new byte[1024]); });
Random rand = new Random(this.GetHashCode() ^ DateTime.Now.Millisecond);
uint streamId = (uint)rand.Next(0, (int)(Math.Pow(2, 24) - 1));
GdsFrame frame = GdsFrame.NewPingFrame(streamId);
Assert.AreEqual(true, frame.IsComplete);
Assert.AreEqual(GdsFrame.GdsFrameType.Ping, frame.Type);
Assert.AreEqual(streamId, frame.StreamId);
MemoryStream stream = new MemoryStream();
frame.Write(stream);
Assert.AreEqual(4, stream.Position);
stream.Position = 0;
GdsFrame readFrame = GdsFrame.ParseFrame(stream, pool);
Assert.AreEqual(true, readFrame.IsComplete);
Assert.AreEqual(GdsFrame.GdsFrameType.Ping, readFrame.Type);
Assert.AreEqual(streamId, readFrame.StreamId);
}
public ChunkedBuffer ToBuffer(GdsFrame frame)
{
ObjectPool <byte[]> pool = new ObjectPool <byte[]>(() => { return(new byte[1024]); });
ChunkedBuffer buffer = new ChunkedBuffer(pool);
frame.Write(buffer.Stream);
return(buffer);
}
public void TestFullCompressed()
{
ObjectPool <byte[]> pool = new ObjectPool <byte[]>(() => { return(new byte[1024]); });
Random rand = new Random(this.GetHashCode() ^ DateTime.Now.Millisecond);
uint streamId = (uint)rand.Next(0, (int)(Math.Pow(2, 24) - 1));
string header1Key = "the first key";
byte[] header1Value = new byte[rand.Next(32, 1024 * 64)];
rand.NextBytes(header1Value);
string header2Key = "the second key";
byte[] header2Value = new byte[rand.Next(32, 1024 * 64)];
rand.NextBytes(header2Value);
byte[] bodyBuffer = new byte[rand.Next(1024, 1024 * 64)];
rand.NextBytes(bodyBuffer);
ChunkedBuffer body = new ChunkedBuffer(pool);
body.OfferRaw(bodyBuffer, 0, bodyBuffer.Length);
GdsFrame frame = GdsFrame.NewContentFrame(streamId, new Dictionary <string, byte[]>()
{
{ header1Key, header1Value },
{ header2Key, header2Value }
}, true, body, true);
Assert.AreEqual(true, frame.IsComplete);
Assert.AreEqual(GdsFrame.GdsFrameType.Full, frame.Type);
Assert.AreEqual(streamId, frame.StreamId);
Assert.AreEqual(2, frame.Headers.Count);
MemoryStream stream = new MemoryStream();
frame.Write(stream);
stream.Position = 0;
GdsFrame readFrame = GdsFrame.ParseFrame(stream, pool);
Assert.AreEqual(true, readFrame.IsComplete);
Assert.AreEqual(GdsFrame.GdsFrameType.Full, readFrame.Type);
Assert.AreEqual(streamId, readFrame.StreamId);
Assert.AreEqual(2, readFrame.Headers.Count);
byte[] readBodyBuffer = new byte[readFrame.Body.AvailableBytesToRead];
readFrame.Body.Read(readBodyBuffer, 0, readBodyBuffer.Length);
AssertEquals(bodyBuffer, readBodyBuffer);
frame.Dispose();
readFrame.Dispose();
}
public void TestHeaderOnlyNotCompressed()
{
ObjectPool <byte[]> pool = new ObjectPool <byte[]>(() => { return(new byte[1024]); });
Random rand = new Random(this.GetHashCode() ^ DateTime.Now.Millisecond);
uint streamId = (uint)rand.Next(0, (int)(Math.Pow(2, 24) - 1));
string header1Key = "the first key";
byte[] header1Value = new byte[rand.Next(32, 1024 * 64)];
rand.NextBytes(header1Value);
string header2Key = "the second key";
byte[] header2Value = new byte[rand.Next(32, 1024 * 64)];
rand.NextBytes(header2Value);
GdsFrame frame = GdsFrame.NewContentFrame(streamId, new Dictionary <string, byte[]>()
{
{ header1Key, header1Value },
{ header2Key, header2Value }
},
false, null, true);
Assert.AreEqual(true, frame.IsComplete);
Assert.AreEqual(GdsFrame.GdsFrameType.HeadersOnly, frame.Type);
Assert.AreEqual(streamId, frame.StreamId);
Assert.AreEqual(2, frame.Headers.Count);
MemoryStream stream = new MemoryStream();
frame.Write(stream);
Assert.AreEqual(
4 // frame definition
+ 2 // header definition
+ (4 * 2) // header sizes for two headers
+ header1Key.Length + header1Value.Length + header2Key.Length + header2Value.Length
, stream.Position);
stream.Position = 0;
GdsFrame readFrame = GdsFrame.ParseFrame(stream, pool);
Assert.AreEqual(true, readFrame.IsComplete);
Assert.AreEqual(GdsFrame.GdsFrameType.HeadersOnly, readFrame.Type);
Assert.AreEqual(streamId, readFrame.StreamId);
Assert.AreEqual(2, readFrame.Headers.Count);
AssertEquals(header1Value, readFrame.Headers[header1Key]);
AssertEquals(header2Value, readFrame.Headers[header2Key]);
}
public void TestBodyOnly()
{
ObjectPool <byte[]> pool = new ObjectPool <byte[]>(() => { return(new byte[1024]); });
Random rand = new Random(this.GetHashCode() ^ DateTime.Now.Millisecond);
uint streamId = (uint)rand.Next(0, (int)(Math.Pow(2, 24) - 1));
byte[] bodyBuffer = new byte[rand.Next(1024, 1024 * 64)];
rand.NextBytes(bodyBuffer);
ChunkedBuffer body = new ChunkedBuffer(pool);
body.OfferRaw(bodyBuffer, 0, bodyBuffer.Length);
GdsFrame frame = GdsFrame.NewContentFrame(streamId, null, false, body, true);
Assert.AreEqual(true, frame.IsComplete);
Assert.AreEqual(GdsFrame.GdsFrameType.BodyOnly, frame.Type);
Assert.AreEqual(streamId, frame.StreamId);
Assert.AreEqual(0, frame.Headers.Count);
MemoryStream stream = new MemoryStream();
frame.Write(stream);
Assert.AreEqual(
4 // frame definition
+ 4 // body definition
+ bodyBuffer.Length
, stream.Position);
stream.Position = 0;
GdsFrame readFrame = GdsFrame.ParseFrame(stream, pool);
Assert.AreEqual(true, readFrame.IsComplete);
Assert.AreEqual(GdsFrame.GdsFrameType.BodyOnly, readFrame.Type);
Assert.AreEqual(streamId, readFrame.StreamId);
Assert.AreEqual(0, readFrame.Headers.Count);
byte[] readBodyBuffer = new byte[readFrame.Body.AvailableBytesToRead];
readFrame.Body.Read(readBodyBuffer, 0, readBodyBuffer.Length);
AssertEquals(bodyBuffer, readBodyBuffer);
frame.Dispose();
readFrame.Dispose();
}
/// <summary>
/// Handles an outgoing HttpRequest and converts it to a raw buffer.
/// </summary>
/// <param name="channel"></param>
/// <param name="obj"></param>
public void HandleOutgoing(ISockNetChannel channel, ref object obj)
{
if (!(obj is GdsFrame))
{
return;
}
GdsFrame gdsFrame = (GdsFrame)obj;
ChunkedBuffer buffer = new ChunkedBuffer(channel.BufferPool);
gdsFrame.Write(buffer.Stream);
gdsFrame.Dispose();
obj = buffer;
}
public void TestCompression()
{
Random rand = new Random(this.GetHashCode() ^ DateTime.Now.Millisecond);
uint streamId = (uint)rand.Next(0, (int)(Math.Pow(2, 24) - 1));
// deflate works great on text - it is horrible with random byte arrays
string header1Key = "Some key";
byte[] header1Value = Encoding.UTF8.GetBytes("Well here is a great value for some key. We're really great at making keys and value.");
string header2Key = "Another key";
byte[] header2Value = Encoding.UTF8.GetBytes("Yet another great value for another key. This is just getting absurd.");
GdsFrame frame = GdsFrame.NewContentFrame(streamId, new Dictionary <string, byte[]>()
{
{ header1Key, header1Value },
{ header2Key, header2Value }
},
true, null, true);
Assert.AreEqual(true, frame.IsComplete);
Assert.AreEqual(GdsFrame.GdsFrameType.HeadersOnly, frame.Type);
Assert.AreEqual(streamId, frame.StreamId);
Assert.AreEqual(2, frame.Headers.Count);
MemoryStream stream = new MemoryStream();
frame.Write(stream);
int uncompressedSize = (4 // frame definition
+ 2 // header definition
+ (4 * 2) // header sizes for two headers
+ header1Key.Length + header1Value.Length + header2Key.Length + header2Value.Length);
Console.WriteLine("Compressed: " + stream.Position + ", Uncompressed: " + uncompressedSize);
Assert.IsTrue(stream.Position < uncompressedSize);
}