1 from logilab.common.testlib import TestCase, unittest_main
2
3 from logilab.common.tasksqueue import *
4
6
8 self.failIf(Task('t1') == Task('t2'))
9 self.failUnless(Task('t1') == Task('t1'))
10
16
17
19
21 queue = PrioritizedTasksQueue()
22 queue.put(Task('t1'))
23 queue.put(Task('t2', MEDIUM))
24 queue.put(Task('t3', HIGH))
25 queue.put(Task('t4', LOW))
26 self.assertEquals(queue.get().id, 't3')
27 self.assertEquals(queue.get().id, 't2')
28 self.assertEquals(queue.get().id, 't1')
29 self.assertEquals(queue.get().id, 't4')
30
32 queue = PrioritizedTasksQueue()
33 queue.put(Task('t1'))
34 queue.put(Task('t2', MEDIUM))
35 queue.put(Task('t1', HIGH))
36 queue.put(Task('t3', MEDIUM))
37 queue.put(Task('t2', MEDIUM))
38 self.assertEquals(queue.qsize(), 3)
39 self.assertEquals(queue.get().id, 't1')
40 self.assertEquals(queue.get().id, 't2')
41 self.assertEquals(queue.get().id, 't3')
42 self.assertEquals(queue.qsize(), 0)
43
45 queue = PrioritizedTasksQueue()
46 queue.put(Task('t1'))
47 queue.put(Task('t2'))
48 queue.put(Task('t3'))
49 queue.remove('t2')
50 self.assertEquals([t.id for t in queue], ['t3', 't1'])
51 self.assertRaises(ValueError, queue.remove, 't4')
52
53 if __name__ == '__main__':
54 unittest_main()
55