public Task <IStorageBlobResultSegment> ListBlobsSegmentedAsync(string prefix, bool useFlatBlobListing,
BlobListingDetails blobListingDetails, int?maxResults, BlobContinuationToken currentToken,
BlobRequestOptions options, OperationContext operationContext, CancellationToken cancellationToken)
{
if (options != null)
{
throw new NotImplementedException();
}
if (operationContext != null)
{
throw new NotImplementedException();
}
Func <string, IStorageBlobContainer> containerFactory = (n) => new FakeStorageBlobContainer(_store, n, this);
IStorageBlobResultSegment segment = _store.ListBlobsSegmented(containerFactory, prefix, useFlatBlobListing,
blobListingDetails, maxResults, currentToken);
return(Task.FromResult(segment));
}
public virtual Task <IStorageBlobResultSegment> ListBlobsSegmentedAsync(string prefix, bool useFlatBlobListing,
BlobListingDetails blobListingDetails, int?maxResults, BlobContinuationToken currentToken,
BlobRequestOptions options, OperationContext operationContext, CancellationToken cancellationToken)
{
if (options != null)
{
throw new NotImplementedException();
}
if (operationContext != null)
{
throw new NotImplementedException();
}
string fullPrefix;
if (!String.IsNullOrEmpty(prefix))
{
fullPrefix = _containerName + "/" + prefix;
}
else
{
fullPrefix = _containerName;
}
Func <string, IStorageBlobContainer> containerFactory = (name) =>
{
if (name != _containerName)
{
throw new InvalidOperationException();
}
return(this);
};
IStorageBlobResultSegment segment = _store.ListBlobsSegmented(containerFactory, fullPrefix,
useFlatBlobListing, blobListingDetails, maxResults, currentToken);
return(Task.FromResult(segment));
}
public async Task ListBlobsAsync_FollowsContinuationTokensToEnd(int blobCount)
{
Mock <IStorageBlobClient> mockClient = new Mock <IStorageBlobClient>(MockBehavior.Strict);
int maxResults = 5000;
List <IStorageListBlobItem> blobs = GetMockBlobs(blobCount);
int numPages = (int)Math.Ceiling(((decimal)blobCount / maxResults));
// create the test data pages with continuation tokens
List <IStorageBlobResultSegment> blobSegments = new List <IStorageBlobResultSegment>();
IStorageBlobResultSegment initialSegement = null;
for (int i = 0; i < numPages; i++)
{
BlobContinuationToken continuationToken = null;
if (i < (numPages - 1))
{
// add a token for all but the last page
continuationToken = new BlobContinuationToken()
{
NextMarker = i.ToString()
};
}
Mock <IStorageBlobResultSegment> mockSegment = new Mock <IStorageBlobResultSegment>(MockBehavior.Strict);
mockSegment.SetupGet(p => p.Results).Returns(blobs.Skip(i * maxResults).Take(maxResults).ToArray());
mockSegment.SetupGet(p => p.ContinuationToken).Returns(continuationToken);
if (i == 0)
{
initialSegement = mockSegment.Object;
}
else
{
blobSegments.Add(mockSegment.Object);
}
}
// Mock the List function to return the correct blob page
HashSet <BlobContinuationToken> seenTokens = new HashSet <BlobContinuationToken>();
IStorageBlobResultSegment resultSegment = null;
mockClient.Setup(p => p.ListBlobsSegmentedAsync("test", true, BlobListingDetails.None, null, It.IsAny <BlobContinuationToken>(), null, null, CancellationToken.None))
.Callback <string, bool, BlobListingDetails, int?, BlobContinuationToken, BlobRequestOptions, OperationContext, CancellationToken>(
(prefix, useFlatBlobListing, blobListingDetails, maxResultsValue, currentToken, options, operationContext, cancellationToken) =>
{
// Previously this is where a bug existed - ListBlobsAsync wasn't handling
// continuation tokens properly and kept sending the same initial token
Assert.DoesNotContain(currentToken, seenTokens);
seenTokens.Add(currentToken);
if (currentToken == null)
{
resultSegment = initialSegement;
}
else
{
int idx = int.Parse(currentToken.NextMarker);
resultSegment = blobSegments[idx];
}
})
.Returns(() => { return(Task.FromResult(resultSegment)); });
IEnumerable <IStorageListBlobItem> results = await mockClient.Object.ListBlobsAsync("test", true, BlobListingDetails.None, CancellationToken.None);
Assert.Equal(blobCount, results.Count());
}
