def test_evaluate_processing_limit(self):
wf = Workflow()
wf.set_max_processing_time(1200)
with mock.patch('celery_dyrygent.workflows.workflow.time') as mck:
mck.time.return_value = 100
wf.evaluate_processing_limit()
wf.processing_limit_ts == 1300
def test_processing_limit(self):
wf = Workflow()
assert wf.max_processing_time == 604800
wf.set_max_processing_time(1200)
assert wf.max_processing_time == 1200
wf2 = Workflow()
assert wf2.max_processing_time == 604800
def test_check_processing_time(self):
wf = Workflow()
wf.set_max_processing_time(0)
wf.evaluate_processing_limit()
with pytest.raises(WorkflowException, match='for too long'):
wf.check_processing_time()
