v1.0

SmartThreadPool v1.0

STPTests/TestPriorityQueue.cs

@@ -0,0 +1,169 @@
+using System;
+
+using NUnit.Framework;
+
+using Amib.Threading;
+using Amib.Threading.Internal;
+using SmartThreadPoolTests;
+
+namespace PriorityQueueTests
+{
+	/// <summary>
+	/// Summary description for TestPriorityQueue.
+	/// </summary>
+	[TestFixture]
+	[Category("TestPriorityQueue")]
+	public class TestPriorityQueue
+	{
+		public TestPriorityQueue()
+		{
+		}
+	
+		[Test]
+		public void Init()
+		{
+			PriorityQueue pq = new PriorityQueue();
+
+			Assert.AreEqual(0, pq.Count);
+
+			Assert.IsNull(pq.Dequeue());
+
+			Assert.AreEqual(0, pq.Count);
+		}
+
+		[Test]
+		public void OneWorkItem()
+		{
+			WorkItemPriority [] priorities = Enum.GetValues(typeof(WorkItemPriority)) as WorkItemPriority [];
+			foreach(WorkItemPriority wip in priorities)
+			{
+				PriorityQueue pq = new PriorityQueue();
+
+				PriorityItem pi = new PriorityItem(wip);
+
+				pq.Enqueue(pi);
+
+				Assert.AreEqual(1, pq.Count, "Failed for priority {0}", wip);
+
+				PriorityItem pi2 = pq.Dequeue() as PriorityItem;
+
+				Assert.IsNotNull(pi2, "Failed for priority {0}", wip);
+
+				Assert.AreSame(pi, pi2, "Failed for priority {0}", wip);
+
+				Assert.AreEqual(0, pq.Count, "Failed for priority {0}", wip);
+			}
+		}
+
+		[Test]
+		public void MultipleWorkItemsOnePriority()
+		{
+			WorkItemPriority [] priorities = Enum.GetValues(typeof(WorkItemPriority)) as WorkItemPriority [];
+			foreach(WorkItemPriority wip in priorities)
+			{
+				PriorityQueue pq = new PriorityQueue();
+
+				PriorityItem [] priorityItems = new PriorityItem[10];
+
+				for(int i = 0; i < priorityItems.Length; ++i)
+				{
+					priorityItems[i] = new PriorityItem(wip);
+
+					pq.Enqueue(priorityItems[i]);
+
+					Assert.AreEqual(i+1, pq.Count, "Failed for priority {0} item count {1}", wip, i+1);
+				}
+
+				for(int i = 0; i < priorityItems.Length; ++i)
+				{
+					PriorityItem pi = pq.Dequeue() as PriorityItem;
+
+					Assert.AreEqual(priorityItems.Length-(i+1), pq.Count, "Failed for priority {0} item count {1}", wip, i+1);
+
+					Assert.IsNotNull(pi, "Failed for priority {0} item count {1}", wip, i+1);
+
+					Assert.AreSame(pi, priorityItems[i], "Failed for priority {0} item count {1}", wip, i+1);
+				}
+
+				Assert.AreEqual(0, pq.Count, "Failed for priority {0}", wip);
+
+				Assert.IsNull(pq.Dequeue());
+
+				Assert.AreEqual(0, pq.Count);
+			}
+		}
+
+		[Test]
+		public void MultipleWorkItemsMultiplePriorities()
+		{
+			// Get all the available priorities
+			WorkItemPriority [] priorities = Enum.GetValues(typeof(WorkItemPriority)) as WorkItemPriority [];
+
+			// Create an array of priority items
+			PriorityItem [] priorityItems = new PriorityItem[priorities.Length];
+
+			// Create a priority item for each priority 
+			int i = priorities.Length;
+			foreach(WorkItemPriority workItemPriority in priorities)
+			{
+				--i;
+				priorityItems[i] = new PriorityItem(workItemPriority);
+			}
+
+			// Create a PermutationGenerator for the priority items
+			PermutationGenerator permutations = new PermutationGenerator(priorityItems);
+			
+			int count = 0;
+			// Iterate over the permutations
+			foreach(object [] permutation in permutations)
+			{
+				++count;
+				Console.Write("Permutation #" + count + " : ");
+				for(int j = 0; j < permutation.Length; ++j)
+				{
+					PriorityItem pi = permutation[j] as PriorityItem;
+					Console.Write(pi.WorkItemPriority + ", ");
+				}
+				Console.WriteLine();
+				// Create a priority queue
+				PriorityQueue pq = new PriorityQueue();
+
+				// Enqueue each priority item according to the permutation
+				for(i = 0; i < permutation.Length; ++i)
+				{
+					PriorityItem priorityItem = permutation[i] as PriorityItem;
+					pq.Enqueue(priorityItem);
+				}
+
+				// Make sure all the priority items are in the queue
+				Assert.AreEqual(priorityItems.Length, pq.Count);
+
+				// Compare the order of the priority items
+				for(i = 0; i < priorityItems.Length; ++i)
+				{
+					PriorityItem priorityItem = pq.Dequeue() as PriorityItem;
+					Assert.AreSame(priorityItems[i], priorityItem);
+				}
+			}
+		}
+
+		private class PriorityItem : IHasWorkItemPriority
+		{
+			private WorkItemPriority _workItemPriority;
+
+			public PriorityItem(WorkItemPriority workItemPriority)
+			{
+				_workItemPriority = workItemPriority;
+			}
+			
+			public WorkItemPriority WorkItemPriority
+			{
+				get
+				{
+					return _workItemPriority;
+				}
+			}
+		}
+
+	}
+}