unit test for -[CWPriorityQueue dequeueAllObjectsOfNextPriorityLevel]

Machx · Jan 14, 2013 · 997257d · 997257d
1 parent d5241b7
commit 997257d
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/Source/CWPriorityQueueTests.m b/Source/CWPriorityQueueTests.m
@@ -75,4 +75,28 @@ -(void)testCountOfPriority
 	STAssertTrue([queue countofObjectsWithPriority:9] == 2, nil);
 }
 
+-(void)testDequeueAllObjectsOfNextPriorityLevel
+{
+	CWPriorityQueue *queue = [CWPriorityQueue new];
+
+	[queue addItem:@"1-1" withPriority:1];
+	[queue addItem:@"2-1" withPriority:2];
+	[queue addItem:@"2-2" withPriority:2];
+	[queue addItem:@"2-3" withPriority:2];
+	[queue addItem:@"2-4" withPriority:2];
+	[queue addItem:@"2-5" withPriority:2];
+	[queue addItem:@"3-1" withPriority:3];
+	[queue addItem:@"3-2" withPriority:3];
+	[queue addItem:@"3-3" withPriority:3];
+	[queue addItem:@"4-1" withPriority:4];
+	[queue addItem:@"4-2" withPriority:4];
+	[queue addItem:@"4-3" withPriority:4];
+	[queue addItem:@"4-4" withPriority:4];
+
+	STAssertTrue([queue dequeueAllObjectsOfNextPriorityLevel].count == 1, nil);
+	STAssertTrue([queue dequeueAllObjectsOfNextPriorityLevel].count == 5, nil);
+	STAssertTrue([queue dequeueAllObjectsOfNextPriorityLevel].count == 3, nil);
+	STAssertTrue([queue dequeueAllObjectsOfNextPriorityLevel].count == 4, nil);
+}
+
 @end