public void TestForceEnqueue()
{
BlockingQueue <int> col = new BlockingQueue <int>(100);
for (int i = 0; i < 100; i++)
{
col.Add(i);
}
Assert.AreEqual(100, col.Count);
Assert.IsFalse(col.TryAdd(int.MaxValue));
col.AddForced(100);
Assert.AreEqual(101, col.Count);
Assert.AreEqual(100, col.BoundedCapacity);
Assert.IsFalse(col.TryAdd(int.MaxValue));
Assert.AreEqual(0, col.Take());
Assert.AreEqual(100, col.Count);
Assert.AreEqual(100, col.BoundedCapacity);
Assert.IsFalse(col.TryAdd(int.MaxValue));
for (int i = 1; i < 101; i++)
{
Assert.AreEqual(i, col.Take());
}
}
public void TestNotAddAfterEnd()
{
BlockingQueue <int> queue = new BlockingQueue <int>(2);
queue.AddForced(1);
Assert.IsTrue(queue.TryAdd(2));
Assert.IsFalse(queue.TryAdd(3));
}
public override void AddForced(T item)
{
Assert.IsFalse(IsDeleted);
Queue.AddForced(item);
Interlocked.Increment(ref FilledItemCount);
}
private void RunComplexTest(BlockingQueue <int> q, int elemCount, int thCount)
{
int atomicRandom = 0;
int trackElemCount = elemCount;
int addFinished = 0;
Thread[] threadsTake = new Thread[thCount];
Thread[] threadsAdd = new Thread[thCount];
CancellationTokenSource tokSrc = new CancellationTokenSource();
List <int> global = new List <int>(elemCount);
Action addAction = () =>
{
Random rnd = new Random(Environment.TickCount + Interlocked.Increment(ref atomicRandom) * thCount * 2);
while (true)
{
int item = Interlocked.Decrement(ref trackElemCount);
if (item < 0)
{
break;
}
if (rnd.Next(100) == 0)
{
q.AddForced(item);
}
else
{
q.Add(item);
}
int sleepTime = rnd.Next(100);
int tmpItem = 0;
if (q.TryPeek(out tmpItem) && tmpItem == item)
{
sleepTime += 100;
}
if (sleepTime > 0)
{
Thread.SpinWait(sleepTime);
}
if (rnd.Next(100) == 0)
{
q.IncreaseBoundedCapacity(1);
}
if (rnd.Next(100) == 0 && q.BoundedCapacity > 20)
{
q.DecreaseBoundedCapacity(1);
}
}
Interlocked.Increment(ref addFinished);
};
Action takeAction = () =>
{
Random rnd = new Random(Environment.TickCount + Interlocked.Increment(ref atomicRandom) * thCount * 2);
List <int> data = new List <int>();
try
{
while (Volatile.Read(ref addFinished) < thCount)
{
int tmp = 0;
if (q.TryTake(out tmp, -1, tokSrc.Token))
{
data.Add((int)tmp);
}
int sleepTime = rnd.Next(100);
if (sleepTime > 0)
{
Thread.SpinWait(sleepTime);
}
}
}
catch (OperationCanceledException) { }
int tmp2;
while (q.TryTake(out tmp2))
{
data.Add((int)tmp2);
}
lock (global)
global.AddRange(data);
};
Task.Delay(1000).ContinueWith(t => q.IncreaseBoundedCapacity(50));
for (int i = 0; i < threadsTake.Length; i++)
{
threadsTake[i] = new Thread(new ThreadStart(takeAction));
}
for (int i = 0; i < threadsAdd.Length; i++)
{
threadsAdd[i] = new Thread(new ThreadStart(addAction));
}
for (int i = 0; i < threadsTake.Length; i++)
{
threadsTake[i].Start();
}
for (int i = 0; i < threadsAdd.Length; i++)
{
threadsAdd[i].Start();
}
for (int i = 0; i < threadsAdd.Length; i++)
{
threadsAdd[i].Join();
}
tokSrc.Cancel();
for (int i = 0; i < threadsTake.Length; i++)
{
threadsTake[i].Join();
}
Assert.AreEqual(elemCount, global.Count);
global.Sort();
for (int i = 0; i < elemCount; i++)
{
Assert.AreEqual(i, global[i]);
}
}