def test_get_task(): global test_taskA global test_taskB from corecluster.cache.task import Task tasks = Task.get_task('queue_b', ['test_action']) assert len(tasks) == 0 tasks = Task.get_task('queue_a', ['test_action']) assert len(tasks) > 0 tasks[0].set_state('not active') tasks[0].save()
def test_empty_queue(): from corecluster.cache.task import Task while True: tasks = Task.get_task('test_task', ['test_action']) if len(tasks) > 0: tasks[0].delete() else: break
def test_empty_queue(): from corecluster.cache.task import Task global imageA global imageB imageA = None imageB = None while True: tasks = Task.get_task('queue_a', ['test_action']) if len(tasks) > 0: tasks[0].delete() else: break while True: tasks = Task.get_task('queue_b', ['test_action']) if len(tasks) > 0: tasks[0].delete() else: break
def test_get_task_c(): # Here, get_task should return only task C (A and B are removed) global test_taskC from corecluster.cache.task import Task tasks = Task.get_task('test_task', ['test_action']) assert len(tasks) > 0 assert tasks[0].cache_key() == test_taskC.cache_key() tasks[0].delete()
def test_get_task_with_another_in_queue(): # taskA is done, get_task should return taskB global test_taskB from corecluster.cache.task import Task tasks = Task.get_task('test_task', ['test_action']) assert len(tasks) > 0 assert tasks[0].cache_key() == test_taskB.cache_key() tasks[0].delete()
def test_get_task(): # We have only test_taskA in queue global test_taskA from corecluster.cache.task import Task tasks = Task.get_task('test_task', ['test_action']) assert len(tasks) > 0 assert tasks[0].cache_key() == test_taskA.cache_key() tasks[0].delete()
def test_get_task_with_another_in_queue(): # taskA is still not active and first in queue global test_taskA from corecluster.cache.task import Task tasks = Task.get_task('test_task', ['test_action']) assert len(tasks) > 0 assert tasks[0].cache_key() == test_taskA.cache_key() tasks[0].set_state('not active') tasks[0].save()
def test_get_task_empty(): # There is no more tasks in queue from corecluster.cache.task import Task tasks = Task.get_task('test_task', ['test_action']) assert len(tasks) == 0