def test_create_run_handler(self): States.workers = None data = {"id": 1, "is_managed": True, "pipeline_id": None} event = MagicMock(data=data, instance=MagicMock(meta_info=None)) handle_run_created(DummyWorkers, event=event) assert States.workers["task"] == CoreSchedulerCeleryTasks.RUNS_PREPARE States.workers = None event = MagicMock(data=data, instance=MagicMock(meta_info={})) handle_run_created(DummyWorkers, event=event) assert States.workers["task"] == CoreSchedulerCeleryTasks.RUNS_PREPARE States.workers = None event = MagicMock(data=data, instance=MagicMock(meta_info={"is_approved": False})) handle_run_created(DummyWorkers, event=event) assert States.workers["task"] == CoreSchedulerCeleryTasks.RUNS_PREPARE States.workers = None event = MagicMock(data=data, instance=MagicMock(meta_info={"is_approved": True})) handle_run_created(DummyWorkers, event=event) assert States.workers["task"] == CoreSchedulerCeleryTasks.RUNS_PREPARE States.workers = None event = MagicMock(data=data, instance=MagicMock(meta_info={"eager": False})) handle_run_created(DummyWorkers, event=event) assert States.workers["task"] == CoreSchedulerCeleryTasks.RUNS_PREPARE States.workers = None event = MagicMock(data=data, instance=MagicMock(meta_info={"eager": True})) handle_run_created(DummyWorkers, event=event) assert States.workers is None
def test_create_run_handler_non_managed_run(self): States.workers = None event = MagicMock(data={"is_managed": False}) handle_run_created(None, event=event) assert States.workers is None
def test_create_run_handler_pipeline_run(self): States.workers = None data = {"is_managed": True, "pipeline_id": 1} event = MagicMock(data=data) handle_run_created(None, event=event) assert States.workers is None
def test_create_run_handler(self): States.workers = None data = {"id": 1, "is_managed": True, "pipeline_id": None} event = MagicMock(data=data) handle_run_created(DummyWorkers, event=event) assert States.workers["task"] == CoreSchedulerCeleryTasks.RUNS_PREPARE