public void ShouldReturnIdsAcrossMultipleGenerators()
{
// Arrange
var account = CloudStorageAccount.DevelopmentStorageAccount;
using (var testScope = new TestScope(account))
{
var store1 = new BlobOptimisticDataStore(account, testScope.ContainerName);
var generator1 = new UniqueIdGenerator(store1) { BatchSize = 3 };
var store2 = new BlobOptimisticDataStore(account, testScope.ContainerName);
var generator2 = new UniqueIdGenerator(store2) { BatchSize = 3 };
// Act
var generatedIds = new[]
{
generator1.NextId(testScope.IdScopeName), //1
generator1.NextId(testScope.IdScopeName), //2
generator1.NextId(testScope.IdScopeName), //3
generator2.NextId(testScope.IdScopeName), //4
generator1.NextId(testScope.IdScopeName), //7
generator2.NextId(testScope.IdScopeName), //5
generator2.NextId(testScope.IdScopeName), //6
generator2.NextId(testScope.IdScopeName), //10
generator1.NextId(testScope.IdScopeName), //8
generator1.NextId(testScope.IdScopeName) //9
};
// Assert
CollectionAssert.AreEqual(
new[] { 1, 2, 3, 4, 7, 5, 6, 10, 8, 9 },
generatedIds);
}
}
public void ShouldInitializeBlobForFirstIdInNewScope()
{
// Arrange
var account = CloudStorageAccount.DevelopmentStorageAccount;
using (var testScope = new TestScope(account))
{
var store = new BlobOptimisticDataStore(account, testScope.ContainerName);
var generator = new UniqueIdGenerator(store) {BatchSize = 3};
// Act
generator.NextId(testScope.IdScopeName); //1
// Assert
Assert.AreEqual("4", testScope.ReadCurrentBlobValue());
}
}
public void ShouldReturnIdsFromThirdBatchIfSecondBatchTakenByAnotherGenerator()
{
// Arrange
var account = CloudStorageAccount.DevelopmentStorageAccount;
using (var testScope = new TestScope(account))
{
var store1 = new BlobOptimisticDataStore(account, testScope.ContainerName);
var generator1 = new UniqueIdGenerator(store1) { BatchSize = 3 };
var store2 = new BlobOptimisticDataStore(account, testScope.ContainerName);
var generator2 = new UniqueIdGenerator(store2) { BatchSize = 3 };
// Act
generator1.NextId(testScope.IdScopeName); //1
generator1.NextId(testScope.IdScopeName); //2
generator1.NextId(testScope.IdScopeName); //3
generator2.NextId(testScope.IdScopeName); //4
var lastId = generator1.NextId(testScope.IdScopeName); //7
// Assert
Assert.AreEqual(7, lastId);
}
}
public void ShouldUpdateBlobWhenGeneratingNextIdAfterEndOfBatch()
{
// Arrange
var account = CloudStorageAccount.DevelopmentStorageAccount;
using (var testScope = new TestScope(account))
{
var store = new BlobOptimisticDataStore(account, testScope.ContainerName);
var generator = new UniqueIdGenerator(store) { BatchSize = 3 };
// Act
generator.NextId(testScope.IdScopeName); //1
generator.NextId(testScope.IdScopeName); //2
generator.NextId(testScope.IdScopeName); //3
generator.NextId(testScope.IdScopeName); //4
// Assert
Assert.AreEqual("7", testScope.ReadCurrentBlobValue());
}
}
public void ShouldSupportUsingOneGeneratorFromMultipleThreads()
{
// Arrange
var account = CloudStorageAccount.DevelopmentStorageAccount;
using (var testScope = new TestScope(account))
{
var store = new BlobOptimisticDataStore(account, testScope.ContainerName);
var generator = new UniqueIdGenerator(store) { BatchSize = 1000 };
const int testLength = 10000;
// Act
var generatedIds = new ConcurrentQueue<long>();
var threadIds = new ConcurrentQueue<int>();
var scopeName = testScope.IdScopeName;
Parallel.For(
0,
testLength,
new ParallelOptions { MaxDegreeOfParallelism = 10 },
i =>
{
generatedIds.Enqueue(generator.NextId(scopeName));
threadIds.Enqueue(Thread.CurrentThread.ManagedThreadId);
});
// Assert we generated the right count of ids
Assert.AreEqual(testLength, generatedIds.Count);
// Assert there were no duplicates
Assert.IsFalse(generatedIds.GroupBy(n => n).Where(g => g.Count() != 1).Any());
// Assert we used multiple threads
var uniqueThreadsUsed = threadIds.Distinct().Count();
if (uniqueThreadsUsed == 1)
Assert.Inconclusive("The test failed to actually utilize multiple threads");
}
}
public void ShouldReturnOneForFirstIdInNewScope()
{
// Arrange
var account = CloudStorageAccount.DevelopmentStorageAccount;
using (var testScope = new TestScope(account))
{
var store = new BlobOptimisticDataStore(account, testScope.ContainerName);
var generator = new UniqueIdGenerator(store) {BatchSize = 3};
// Act
var generatedId = generator.NextId(testScope.IdScopeName);
// Assert
Assert.AreEqual(1, generatedId);
}
}