private async Task <IByteBuffer> ReceiveBufferAsync()
{
try
{
var messageSizeBytes = new byte[4];
await _stream.ReadBytesAsync(messageSizeBytes, 0, 4, _backgroundTaskCancellationToken).ConfigureAwait(false);
var messageSize = BitConverter.ToInt32(messageSizeBytes, 0);
EnsureMessageSizeIsValid(messageSize);
var inputBufferChunkSource = new InputBufferChunkSource(BsonChunkPool.Default);
var buffer = ByteBufferFactory.Create(inputBufferChunkSource, messageSize);
buffer.Length = messageSize;
buffer.SetBytes(0, messageSizeBytes, 0, 4);
await _stream.ReadBytesAsync(buffer, 4, messageSize - 4, _backgroundTaskCancellationToken).ConfigureAwait(false);
_lastUsedAtUtc = DateTime.UtcNow;
buffer.MakeReadOnly();
return(buffer);
}
catch (Exception ex)
{
var wrappedException = WrapException(ex, "receiving a message from the server");
ConnectionFailed(wrappedException);
throw wrappedException;
}
}
public void constructor_should_use_default_value_for_minChunkSize()
{
var mockBaseSource = new Mock <IBsonChunkSource>();
var subject = new InputBufferChunkSource(mockBaseSource.Object);
subject.MinChunkSize.Should().Be(16 * 1024);
}
public void Dispose_can_be_called_more_than_once()
{
var baseSource = Substitute.For <IBsonChunkSource>();
var subject = new InputBufferChunkSource(baseSource);
subject.Dispose();
subject.Dispose();
}
public void Dispose_can_be_called_more_than_once()
{
var mockBaseSource = new Mock <IBsonChunkSource>();
var subject = new InputBufferChunkSource(mockBaseSource.Object);
subject.Dispose();
subject.Dispose();
}
public void constructor_should_use_default_value_for_minChunkSize()
{
var baseSource = Substitute.For <IBsonChunkSource>();
var subject = new InputBufferChunkSource(baseSource);
subject.MinChunkSize.Should().Be(16 * 1024);
}
public void BaseSource_get_should_return_expected_result()
{
var baseSource = Substitute.For <IBsonChunkSource>();
var subject = new InputBufferChunkSource(baseSource);
var result = subject.BaseSource;
result.Should().BeSameAs(baseSource);
}
public void BaseSource_get_should_return_expected_result()
{
var mockBaseSource = new Mock <IBsonChunkSource>();
var subject = new InputBufferChunkSource(mockBaseSource.Object);
var result = subject.BaseSource;
result.Should().BeSameAs(mockBaseSource.Object);
}
public void MinChunkSize_get_should_return_expected_result()
{
var mockBaseSource = new Mock <IBsonChunkSource>();
var minChunkSize = 8 * 1024;
var subject = new InputBufferChunkSource(mockBaseSource.Object, minChunkSize: minChunkSize);
var result = subject.MinChunkSize;
result.Should().Be(minChunkSize);
}
public void MinChunkSize_get_should_return_expected_result()
{
var baseSource = Substitute.For <IBsonChunkSource>();
var minChunkSize = 8 * 1024;
var subject = new InputBufferChunkSource(baseSource, minChunkSize: minChunkSize);
var result = subject.MinChunkSize;
result.Should().Be(minChunkSize);
}
public void GetChunk_should_throw_when_requestedSize_is_less_than_or_equal_to_zero(
[Values(-1, 0)]
int requestedSize)
{
var baseSource = Substitute.For <IBsonChunkSource>();
var subject = new InputBufferChunkSource(baseSource);
Action action = () => subject.GetChunk(requestedSize);
action.ShouldThrow <ArgumentOutOfRangeException>().And.ParamName.Should().Be("requestedSize");
}
public void GetChunk_should_round_requestedSize_to_power_of_2_without_wasting_too_much_space(int requestedSize, int roundedSize)
{
var baseSource = Substitute.For <IBsonChunkSource>();
var subject = new InputBufferChunkSource(baseSource, maxUnpooledChunkSize: 0, minChunkSize: 4, maxChunkSize: 16);
baseSource.GetChunk(Arg.Any <int>()).Returns(Substitute.For <IBsonChunk>());
subject.GetChunk(requestedSize);
baseSource.Received(1).GetChunk(roundedSize);
}
public void GetChunk_should_throw_when_subject_is_disposed()
{
var mockBaseSource = new Mock <IBsonChunkSource>();
var subject = new InputBufferChunkSource(mockBaseSource.Object);
subject.Dispose();
Action action = () => subject.GetChunk(1);
action.ShouldThrow <ObjectDisposedException>().And.ObjectName.Should().Be("InputBufferChunkSource");
}
public void GetChunk_should_round_requestedSize_to_power_of_2_without_wasting_too_much_space(int requestedSize, int roundedSize)
{
var mockBaseSource = new Mock <IBsonChunkSource>();
var subject = new InputBufferChunkSource(mockBaseSource.Object, maxUnpooledChunkSize: 0, minChunkSize: 4, maxChunkSize: 16);
mockBaseSource.Setup(s => s.GetChunk(It.IsAny <int>())).Returns(() => new Mock <IBsonChunk>().Object);
subject.GetChunk(requestedSize);
mockBaseSource.Verify(s => s.GetChunk(roundedSize), Times.Once);
}
public void Dispose_should_dispose_subject()
{
var baseSource = Substitute.For <IBsonChunkSource>();
var subject = new InputBufferChunkSource(baseSource);
subject.Dispose();
var reflector = new Reflector(subject);
reflector._disposed.Should().BeTrue();
}
public void Dispose_should_dispose_subject()
{
var mockBaseSource = new Mock <IBsonChunkSource>();
var subject = new InputBufferChunkSource(mockBaseSource.Object);
subject.Dispose();
var reflector = new Reflector(subject);
reflector._disposed.Should().BeTrue();
}
public void BaseSource_get_should_throw_when_subject_is_disposed()
{
var baseSource = Substitute.For <IBsonChunkSource>();
var subject = new InputBufferChunkSource(baseSource);
subject.Dispose();
Action action = () => { var _ = subject.BaseSource; };
action.ShouldThrow <ObjectDisposedException>().And.ObjectName.Should().Be("InputBufferChunkSource");
}
public void GetChunk_should_return_unpooled_chunk_when_requestedSize_is_less_than_or_equal_to_maxUnpooledChunkSize(
[Values(1, 2, 4 * 1024 - 1, 4 * 1024)]
int requestedSize)
{
var baseSource = Substitute.For <IBsonChunkSource>();
var subject = new InputBufferChunkSource(baseSource);
var result = subject.GetChunk(requestedSize);
result.Should().BeOfType <ByteArrayChunk>();
result.Bytes.Count.Should().Be(requestedSize);
baseSource.DidNotReceive().GetChunk(Arg.Any <int>());
}
public void GetChunk_should_return_unpooled_chunk_when_requestedSize_is_less_than_or_equal_to_maxUnpooledChunkSize(
[Values(1, 2, 4 * 1024 - 1, 4 * 1024)]
int requestedSize)
{
var mockBaseSource = new Mock <IBsonChunkSource>();
var subject = new InputBufferChunkSource(mockBaseSource.Object);
var result = subject.GetChunk(requestedSize);
result.Should().BeOfType <ByteArrayChunk>();
result.Bytes.Count.Should().Be(requestedSize);
mockBaseSource.Verify(s => s.GetChunk(It.IsAny <int>()), Times.Never);
}
public void constructor_should_initialize_subject()
{
var baseSource = Substitute.For <IBsonChunkSource>();
var maxUnpooledChunkSize = 2;
var minChunkSize = 4;
var maxChunkSize = 8;
var subject = new InputBufferChunkSource(baseSource, maxUnpooledChunkSize, minChunkSize, maxChunkSize);
var reflector = new Reflector(subject);
subject.BaseSource.Should().BeSameAs(baseSource);
subject.MaxChunkSize.Should().Be(maxChunkSize);
subject.MaxUnpooledChunkSize.Should().Be(maxUnpooledChunkSize);
subject.MinChunkSize.Should().Be(minChunkSize);
reflector._disposed.Should().BeFalse();
}
public Reflector(InputBufferChunkSource instance)
{
_instance = instance;
}
