def test_add_task():
global imageA
global imageB
global test_taskA
global test_taskB
from corecluster.cache.task import Task
assert imageA.last_task == None
assert imageB.last_task == None
test_taskA = Task()
test_taskA.type = 'queue_a'
test_taskA.state = 'not active'
test_taskA.action = 'test_action'
test_taskA.append_to([imageA, imageB])
assert test_taskA.blockers == []
test_taskB = Task()
test_taskB.type = 'queue_b'
test_taskB.state = 'not active'
test_taskB.action = 'test_action'
test_taskB.append_to([imageA, imageB])
assert test_taskB.blockers == [
test_taskA.cache_key(), test_taskA.cache_key()
]
def test_append_third_task():
# Append next task to empty queue
global imageA
global imageB
global test_taskC
from corecluster.cache.task import Task
from corecluster.models.core import Image
test_taskC = Task()
test_taskC.type = 'test_task'
test_taskC.state = 'not active'
test_taskC.action = 'test_action'
# New task is assigned only to imageA
test_taskC.append_to([imageA])
img_a = Image.objects.get(pk=imageA.id)
assert img_a.last_task == test_taskC.cache_key()
