public async Task GetBucket_MultipleTimes_ReturnsFirstOpenBucket()
{
// Arrange
var bucket1 = new Mock <IBucket>();
var cluster = new Mock <ICluster>();
cluster
.Setup(m => m.BucketAsync("bucket1"))
.ReturnsAsync(bucket1.Object);
var clusterProvider = new Mock <IClusterProvider>();
clusterProvider
.Setup(m => m.GetClusterAsync())
.ReturnsAsync(cluster.Object);
var provider = new BucketProvider(clusterProvider.Object);
// Act
await provider.GetBucketAsync("bucket1");
var result = await provider.GetBucketAsync("bucket1");
// Assert
Assert.Equal(bucket1.Object, result);
cluster.Verify(m => m.BucketAsync("bucket1"), Times.Once);
}
public async Task GetBucket_FirstTime_OpensBucket()
{
// Arrange
var bucket1 = new Mock <IBucket>();
var cluster = new Mock <ICluster>();
cluster
.Setup(m => m.BucketAsync("bucket1"))
.ReturnsAsync(bucket1.Object);
var clusterProvider = new Mock <IClusterProvider>();
clusterProvider
.Setup(m => m.GetClusterAsync())
.ReturnsAsync(cluster.Object);
var provider = new BucketProvider(clusterProvider.Object);
// Act
var result = await provider.GetBucketAsync("bucket1");
// Assert
Assert.Equal(bucket1.Object, result);
}
public async Task GetBucket_FirstTimeFails_OpensAgain()
{
// Arrange
var bucket1 = new Mock <IBucket>();
var cluster = new Mock <ICluster>();
cluster
.SetupSequence(m => m.BucketAsync("bucket1"))
.Returns(() => new ValueTask <IBucket>(Task.Run(async Task <IBucket> () =>
{
await Task.Delay(10);
throw new InvalidOperationException("test");
})))
.ReturnsAsync(bucket1.Object);
var clusterProvider = new Mock <IClusterProvider>();
clusterProvider
.Setup(m => m.GetClusterAsync())
.ReturnsAsync(cluster.Object);
var provider = new BucketProvider(clusterProvider.Object);
var firstEx = await Assert.ThrowsAsync <InvalidOperationException>(() =>
{
return(provider.GetBucketAsync("bucket1").AsTask());
});
Assert.Equal("test", firstEx.Message);
// Act
var result = await provider.GetBucketAsync("bucket1");
// Assert
Assert.Equal(bucket1.Object, result);
cluster.Verify(m => m.BucketAsync("bucket1"), Times.Exactly(2));
}
public async Task DisposeAsync_DisposesBuckets()
{
// Arrange
var bucket1 = new Mock <IBucket>();
bucket1.Setup(m => m.Dispose());
var bucket2 = new Mock <IBucket>();
bucket2.Setup(m => m.Dispose());
var cluster = new Mock <ICluster>();
cluster
.Setup(m => m.BucketAsync("bucket1"))
.ReturnsAsync(bucket1.Object);
cluster
.Setup(m => m.BucketAsync("bucket2"))
.ReturnsAsync(bucket2.Object);
var clusterProvider = new Mock <IClusterProvider>();
clusterProvider
.Setup(m => m.GetClusterAsync())
.ReturnsAsync(cluster.Object);
var provider = new BucketProvider(clusterProvider.Object);
await provider.GetBucketAsync("bucket1");
await provider.GetBucketAsync("bucket2");
// Act
await provider.DisposeAsync();
// Assert
bucket1.Verify(m => m.DisposeAsync(), Times.AtLeastOnce);
bucket2.Verify(m => m.DisposeAsync(), Times.AtLeastOnce);
}
public async Task GetBucketAsync_NullBucketName_Exception()
{
// Arrange
var clusterProvider = new Mock <IClusterProvider>();
var provider = new BucketProvider(clusterProvider.Object);
// Act/Assert
var ex = await Assert.ThrowsAsync <ArgumentNullException>(() => provider.GetBucketAsync(null).AsTask());
Assert.Equal("bucketName", ex.ParamName);
}
public async Task GetBucketAsync_Disposed_Exception()
{
// Arrange
var clusterProvider = new Mock <IClusterProvider>();
var provider = new BucketProvider(clusterProvider.Object);
provider.Dispose();
// Act/Assert
var ex = await Assert.ThrowsAsync <ObjectDisposedException>(() => provider.GetBucketAsync("bucket1").AsTask());
Assert.Equal(nameof(BucketProvider), ex.ObjectName);
}
