def test_task_queue(self):
task_queue = TaskQueue()
task_queue.processing_timeout = 0.1
task_queue.put('a3', 3, time.time()+0.1)
task_queue.put('a1', 1)
task_queue.put('a2', 2)
# priority queue
self.assertEqual(task_queue.get(), 'a2')
# time queue
time.sleep(0.1)
task_queue._check_time_queue()
self.assertEqual(task_queue.get(), 'a3')
self.assertEqual(task_queue.get(), 'a1')
# processing queue
task_queue._check_processing()
self.assertEqual(task_queue.get(), 'a2')
self.assertEqual(len(task_queue), 0)
# done
task_queue.done('a2')
task_queue.done('a1')
time.sleep(0.1)
task_queue._check_processing()
task_queue._check_time_queue()
self.assertEqual(task_queue.get(), 'a3')
self.assertEqual(task_queue.get(), None)
class TestTaskQueue(unittest.TestCase):
@classmethod
def setUpClass(self):
self.task_queue = TaskQueue()
self.task_queue.rate = 100000
self.task_queue.burst = 100000
self.task_queue.processing_timeout = 0.2
self.task_queue.put('a3', 2, time.time() + 0.1)
self.task_queue.put('a1', 1)
self.task_queue.put('a2', 3)
def test_1_priority_queue(self):
self.assertEqual(self.task_queue.get(), 'a2')
def test_2_time_queue(self):
time.sleep(0.1)
self.task_queue.check_update()
self.assertEqual(self.task_queue.get(), 'a3')
self.assertEqual(self.task_queue.get(), 'a1')
def test_3_processing_queue(self):
time.sleep(0.1)
self.task_queue.check_update()
self.assertEqual(self.task_queue.get(), 'a2')
self.assertEqual(len(self.task_queue), 0)
def test_4_done(self):
self.task_queue.done('a2')
self.task_queue.done('a1')
time.sleep(0.1)
self.task_queue.check_update()
self.assertEqual(self.task_queue.get(), 'a3')
self.assertEqual(self.task_queue.get(), None)
class TestTaskQueue(unittest.TestCase):
@classmethod
def setUpClass(self):
self.task_queue = TaskQueue()
self.task_queue.rate = 100000
self.task_queue.burst = 100000
self.task_queue.processing_timeout = 0.2
self.task_queue.put('a3', 2, time.time()+0.1)
self.task_queue.put('a1', 1)
self.task_queue.put('a2', 3)
def test_1_priority_queue(self):
self.assertEqual(self.task_queue.get(), 'a2')
def test_2_time_queue(self):
time.sleep(0.1)
self.task_queue.check_update()
self.assertEqual(self.task_queue.get(), 'a3')
self.assertEqual(self.task_queue.get(), 'a1')
def test_3_processing_queue(self):
time.sleep(0.1)
self.task_queue.check_update()
self.assertEqual(self.task_queue.get(), 'a2')
self.assertEqual(len(self.task_queue), 0)
def test_4_done(self):
self.task_queue.done('a2')
self.task_queue.done('a1')
time.sleep(0.1)
self.task_queue.check_update()
self.assertEqual(self.task_queue.get(), 'a3')
self.assertEqual(self.task_queue.get(), None)
def test_task_queue(self):
task_queue = TaskQueue()
task_queue.processing_timeout = 0.1
task_queue.put('a3', 3, time.time() + 0.1)
task_queue.put('a1', 1)
task_queue.put('a2', 2)
# priority queue
self.assertEqual(task_queue.get(), 'a2')
# time queue
time.sleep(0.1)
task_queue._check_time_queue()
self.assertEqual(task_queue.get(), 'a3')
self.assertEqual(task_queue.get(), 'a1')
# processing queue
task_queue._check_processing()
self.assertEqual(task_queue.get(), 'a2')
self.assertEqual(len(task_queue), 0)
# done
task_queue.done('a2')
task_queue.done('a1')
time.sleep(0.1)
task_queue._check_processing()
task_queue._check_time_queue()
self.assertEqual(task_queue.get(), 'a3')
self.assertEqual(task_queue.get(), None)
