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