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
