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()