public void Remove_Should_ReturnFalseForSomeItemsWhenNoExternalLockIsUsed()
{
// Arrange
var concurrentCollection = new ConcurrentCollection <int>();
for (int i = 0; i < 500; i++)
{
concurrentCollection.Add(i);
}
// Act
var collectionConsumeTasks = new List <Task>();
int collectionTotalSize = concurrentCollection.Count;
int removedItems = 0;
for (int i = 0; i < collectionTotalSize; i++)
{
collectionConsumeTasks.Add(Task.Run(() =>
{
var item = concurrentCollection.FirstOrDefault();
var removed = concurrentCollection.Remove(item);
if (removed)
{
removedItems++;
}
}));
}
Task.WaitAll(collectionConsumeTasks.ToArray());
// Assert
Assert.Equal(collectionTotalSize - removedItems, concurrentCollection.Count);
}
public void Remove_Should_WorkWhenLockIsDoneBetweenIteratorAndModifier()
{
// Arrange
var concurrentCollection = new ConcurrentCollection <int>();
for (int i = 0; i < 500; i++)
{
concurrentCollection.Add(i);
}
// Act
var collectionConsumeTasks = new List <Task>();
int collectionTotalSize = concurrentCollection.Count;
for (int i = 0; i < collectionTotalSize; i++)
{
collectionConsumeTasks.Add(Task.Run(() =>
{
lock (_lock)
{
var item = concurrentCollection.FirstOrDefault();
concurrentCollection.Remove(item);
}
}));
}
Task.WaitAll(collectionConsumeTasks.ToArray());
// Assert
Assert.Empty(concurrentCollection);
}