|
|
|
@ -27,12 +27,99 @@ namespace Grpc.Core.Internal.Tests |
|
|
|
|
public class SliceBufferSafeHandleTest |
|
|
|
|
{ |
|
|
|
|
[TestCase] |
|
|
|
|
public void BasicTest() |
|
|
|
|
public void Complete_EmptyBuffer() |
|
|
|
|
{ |
|
|
|
|
|
|
|
|
|
using (var sliceBuffer = SliceBufferSafeHandle.Create()) |
|
|
|
|
{ |
|
|
|
|
sliceBuffer.Complete(); |
|
|
|
|
CollectionAssert.AreEqual(new byte[0], sliceBuffer.ToByteArray()); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// other ideas: |
|
|
|
|
// AdjustTailSpace(0) if previous tail size is 0... (better for SliceBufferSafeHandle) |
|
|
|
|
[TestCase] |
|
|
|
|
public void Complete_TailSizeZero() |
|
|
|
|
{ |
|
|
|
|
using (var sliceBuffer = SliceBufferSafeHandle.Create()) |
|
|
|
|
{ |
|
|
|
|
var origPayload = GetTestBuffer(10); |
|
|
|
|
origPayload.AsSpan().CopyTo(sliceBuffer.GetSpan(origPayload.Length)); |
|
|
|
|
sliceBuffer.Advance(origPayload.Length); |
|
|
|
|
// call complete where tail space size == 0 |
|
|
|
|
sliceBuffer.Complete(); |
|
|
|
|
CollectionAssert.AreEqual(origPayload, sliceBuffer.ToByteArray()); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
[TestCase] |
|
|
|
|
public void Complete_TruncateTailSpace() |
|
|
|
|
{ |
|
|
|
|
using (var sliceBuffer = SliceBufferSafeHandle.Create()) |
|
|
|
|
{ |
|
|
|
|
var origPayload = GetTestBuffer(10); |
|
|
|
|
var dest = sliceBuffer.GetSpan(origPayload.Length + 10); |
|
|
|
|
origPayload.AsSpan().CopyTo(dest); |
|
|
|
|
sliceBuffer.Advance(origPayload.Length); |
|
|
|
|
// call complete where tail space needs to be truncated |
|
|
|
|
sliceBuffer.Complete(); |
|
|
|
|
CollectionAssert.AreEqual(origPayload, sliceBuffer.ToByteArray()); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
[TestCase] |
|
|
|
|
public void SliceBufferIsReusable() |
|
|
|
|
{ |
|
|
|
|
using (var sliceBuffer = SliceBufferSafeHandle.Create()) |
|
|
|
|
{ |
|
|
|
|
var origPayload = GetTestBuffer(10); |
|
|
|
|
origPayload.AsSpan().CopyTo(sliceBuffer.GetSpan(origPayload.Length)); |
|
|
|
|
sliceBuffer.Advance(origPayload.Length); |
|
|
|
|
sliceBuffer.Complete(); |
|
|
|
|
CollectionAssert.AreEqual(origPayload, sliceBuffer.ToByteArray()); |
|
|
|
|
|
|
|
|
|
sliceBuffer.Reset(); |
|
|
|
|
|
|
|
|
|
var origPayload2 = GetTestBuffer(20); |
|
|
|
|
origPayload2.AsSpan().CopyTo(sliceBuffer.GetSpan(origPayload2.Length)); |
|
|
|
|
sliceBuffer.Advance(origPayload2.Length); |
|
|
|
|
sliceBuffer.Complete(); |
|
|
|
|
CollectionAssert.AreEqual(origPayload2, sliceBuffer.ToByteArray()); |
|
|
|
|
|
|
|
|
|
sliceBuffer.Reset(); |
|
|
|
|
|
|
|
|
|
CollectionAssert.AreEqual(new byte[0], sliceBuffer.ToByteArray()); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
[TestCase] |
|
|
|
|
public void SliceBuffer_SizeHintZero() |
|
|
|
|
{ |
|
|
|
|
using (var sliceBuffer = SliceBufferSafeHandle.Create()) |
|
|
|
|
{ |
|
|
|
|
var destSpan = sliceBuffer.GetSpan(0); |
|
|
|
|
Assert.IsTrue(destSpan.Length > 0); // some non-zero size memory is made available |
|
|
|
|
|
|
|
|
|
sliceBuffer.Reset(); |
|
|
|
|
|
|
|
|
|
var destMemory = sliceBuffer.GetMemory(0); |
|
|
|
|
Assert.IsTrue(destMemory.Length > 0); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Advance() - multiple chunks... |
|
|
|
|
|
|
|
|
|
// other TODOS: |
|
|
|
|
|
|
|
|
|
// -- provide a null instance of SliceBufferSafeHandle |
|
|
|
|
//-- order of UnsafeSerialize in AsyncCall methods... |
|
|
|
|
|
|
|
|
|
private byte[] GetTestBuffer(int length) |
|
|
|
|
{ |
|
|
|
|
var testBuffer = new byte[length]; |
|
|
|
|
for (int i = 0; i < testBuffer.Length; i++) |
|
|
|
|
{ |
|
|
|
|
testBuffer[i] = (byte) i; |
|
|
|
|
} |
|
|
|
|
return testBuffer; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|