public async Task Advance_VeryLargeData_SuccessWithRemainder()
{
// Arrange
var s = string.Create <object>(16384, null !, (s, o) =>
{
for (var i = 0; i < s.Length; i++)
{
s[i] = Convert.ToChar(i % 10);
}
});
var initialData = Encoding.UTF8.GetBytes(s);
var testPipe = new Pipe();
var w = new Base64PipeWriter(testPipe.Writer);
var innerBuffer = testPipe.Writer.GetMemory();
// Act
var buffer = w.GetMemory(initialData.Length);
initialData.CopyTo(buffer);
w.Advance(initialData.Length);
await w.CompleteAsync().AsTask().DefaultTimeout();
// Assert
var result = await testPipe.Reader.ReadAsync().AsTask().DefaultTimeout();
var resultData = result.Buffer.ToArray();
Assert.AreEqual(21848, result.Buffer.Length);
var base64Data = Encoding.UTF8.GetBytes(Convert.ToBase64String(initialData)).ToArray();
CollectionAssert.AreEqual(resultData, base64Data);
}
public async Task Complete_HasRemainder_WriteRemainder()
{
// Arrange
var initialData = Encoding.UTF8.GetBytes("Hello world");
var testPipe = new Pipe();
var w = new Base64PipeWriter(testPipe.Writer);
// Act
var buffer = w.GetMemory(initialData.Length);
initialData.CopyTo(buffer);
w.Advance(initialData.Length);
w.Complete();
// Assert
var result = await testPipe.Reader.ReadAsync().AsTask().DefaultTimeout();
Assert.AreEqual(16, result.Buffer.Length);
var base64Data = Encoding.UTF8.GetBytes(Convert.ToBase64String(initialData));
CollectionAssert.AreEqual(base64Data, result.Buffer.ToArray());
}
public async Task Advance_SmallDataIncompleteWrites_Success()
{
// Arrange
var initialData1 = Encoding.UTF8.GetBytes("Hello");
var initialData2 = Encoding.UTF8.GetBytes("world");
var testPipe = new Pipe();
var w = new Base64PipeWriter(testPipe.Writer);
// Act
var buffer = w.GetMemory(initialData1.Length);
initialData1.CopyTo(buffer);
w.Advance(initialData1.Length);
buffer = w.GetMemory(initialData2.Length);
initialData2.CopyTo(buffer);
w.Advance(initialData2.Length);
w.Complete();
// Assert
var result = await testPipe.Reader.ReadAsync().AsTask().DefaultTimeout();
Assert.Greater(result.Buffer.Length, 0);
var base64Data = Encoding.UTF8.GetBytes(Convert.ToBase64String(initialData1.Concat(initialData2).ToArray()));
CollectionAssert.AreEqual(base64Data, result.Buffer.ToArray());
}
public async Task Advance_SmallDataMultipleWrites_Success(int size)
{
// Arrange
var initialData = Encoding.UTF8.GetBytes("Hello world");
var testPipe = new Pipe();
var w = new Base64PipeWriter(testPipe.Writer);
// Act
foreach (var b in Split(initialData, size))
{
var buffer = w.GetMemory(b.Length);
for (var i = 0; i < b.Length; i++)
{
buffer.Span[i] = b[i];
}
w.Advance(b.Length);
}
w.Complete();
// Assert
var result = await testPipe.Reader.ReadAsync().AsTask().DefaultTimeout();
Assert.Greater(result.Buffer.Length, 0);
var base64Data = Encoding.UTF8.GetBytes(Convert.ToBase64String(initialData));
var resultData = result.Buffer.ToArray();
CollectionAssert.AreEqual(base64Data, resultData);
}
public async Task WriteMessageAsync_NoFlush_WriteNoData()
{
// Arrange
var ms = new MemoryStream();
var pipeWriter = new Base64PipeWriter(PipeWriter.Create(ms));
// Act
await pipeWriter.WriteMessageAsync(new EchoRequest { Message = "test" }, HttpContextServerCallContextHelper.CreateServerCallContext(), MarshallerEchoRequest.ContextualSerializer, canFlush : false);
// Assert
var messageData = ms.ToArray();
Assert.AreEqual(0, messageData.Length);
}
public async Task WriteMessageAsync_MultipleMessagesWithFlush_WriteMessagesAsSegments()
{
// Arrange
var ms = new MemoryStream();
var pipeWriter = new Base64PipeWriter(PipeWriter.Create(ms));
// Act
await pipeWriter.WriteMessageAsync(new EchoRequest { Message = "test" }, HttpContextServerCallContextHelper.CreateServerCallContext(), MarshallerEchoRequest.ContextualSerializer, canFlush : true);
await pipeWriter.WriteMessageAsync(new EchoRequest { Message = "test" }, HttpContextServerCallContextHelper.CreateServerCallContext(), MarshallerEchoRequest.ContextualSerializer, canFlush : true);
await pipeWriter.WriteMessageAsync(new EchoRequest { Message = "test" }, HttpContextServerCallContextHelper.CreateServerCallContext(), MarshallerEchoRequest.ContextualSerializer, canFlush : true);
// Assert
var base64 = Encoding.UTF8.GetString(ms.ToArray());
Assert.AreEqual("AAAAAAYKBHRlc3Q=AAAAAAYKBHRlc3Q=AAAAAAYKBHRlc3Q=", base64);
}
public async Task WriteStreamedMessageAsync_MultipleMessagesNoFlush_WriteMessages()
{
// Arrange
var ms = new MemoryStream();
var pipeWriter = new Base64PipeWriter(PipeWriter.Create(ms));
var writeOptions = new WriteOptions(WriteFlags.BufferHint);
// Act
await pipeWriter.WriteStreamedMessageAsync(new EchoRequest { Message = "test" }, HttpContextServerCallContextHelper.CreateServerCallContext(writeOptions: writeOptions), MarshallerEchoRequest.ContextualSerializer);
await pipeWriter.WriteStreamedMessageAsync(new EchoRequest { Message = "test" }, HttpContextServerCallContextHelper.CreateServerCallContext(writeOptions: writeOptions), MarshallerEchoRequest.ContextualSerializer);
await pipeWriter.WriteStreamedMessageAsync(new EchoRequest { Message = "test" }, HttpContextServerCallContextHelper.CreateServerCallContext(writeOptions: writeOptions), MarshallerEchoRequest.ContextualSerializer);
pipeWriter.Complete();
// Assert
var base64 = Encoding.UTF8.GetString(ms.ToArray());
Assert.AreEqual("AAAAAAYKBHRlc3QAAAAABgoEdGVzdAAAAAAGCgR0ZXN0", base64);
}
public void Advance_SmallData_SuccessWithRemainder()
{
// Arrange
var initialData = Encoding.UTF8.GetBytes("Hello world");
var testPipe = new Pipe();
var w = new Base64PipeWriter(testPipe.Writer);
var innerBuffer = testPipe.Writer.GetMemory();
// Act
var buffer = w.GetMemory(initialData.Length);
initialData.CopyTo(buffer);
w.Advance(initialData.Length);
// Assert
Assert.AreEqual((byte)'l', w._remainderByte0); // remaining bytes, end of "world"
Assert.AreEqual((byte)'d', w._remainderByte1); // remaining bytes, end of "world"
var base64Data = Encoding.UTF8.GetBytes(Convert.ToBase64String(initialData)).AsSpan(0, 12).ToArray();
CollectionAssert.AreEqual(innerBuffer.Slice(0, 12).ToArray(), base64Data);
}
public async Task WriteMessageAsync_EmptyMessage_WriteMessageWithNoData()
{
// Arrange
var ms = new MemoryStream();
var pipeWriter = new Base64PipeWriter(PipeWriter.Create(ms));
// Act
await pipeWriter.WriteMessageAsync(new EchoRequest(), HttpContextServerCallContextHelper.CreateServerCallContext(), MarshallerEchoRequest.ContextualSerializer, canFlush : true);
// Assert
var base64 = Encoding.UTF8.GetString(ms.ToArray());
var messageData = Convert.FromBase64String(base64);
CollectionAssert.AreEqual(
new byte[]
{
0x00, // compression = 0
0x00,
0x00,
0x00,
0x00, // length = 0
},
messageData);
}
