def test_load_pipeline(self): action, workflow, stage, pipeline = self._pipeline_helper() workflow_engine = WorkflowEngine(pipeline) loaded_pipeline = workflow_engine._load_pipeline() eq_(pipeline, workflow_engine.pipeline) eq_(pipeline, loaded_pipeline) eq_({1: stage}, workflow_engine.stage_mapper) eq_({1: workflow}, workflow_engine.workflow_mapper) eq_({1: action}, workflow_engine.action_mapper)
def test_init(self): workflow_engine = WorkflowEngine(Pipeline()) eq_(Pipeline, workflow_engine.pipeline_class) eq_(Stage, workflow_engine.stage_class) eq_(Workflow, workflow_engine.workflow_class) eq_(Action, workflow_engine.action_class)
def test_get_actions_by_workflow_id(self): action, workflow, stage, pipeline = self._pipeline_helper() workflow_engine = WorkflowEngine(pipeline) eq_([action], workflow_engine._get_actions_by_workflow_id(1))
def test_get_actions(self): action = Action() workflow = Workflow(actions=[action]) workflow_engine = WorkflowEngine(Mock()) eq_([action], workflow_engine._get_actions(workflow))
def test_get_workflows(self): workflow = Workflow() stage = Stage(workflows=[workflow]) workflow_engine = WorkflowEngine(Mock()) eq_([workflow], workflow_engine._get_workflows(stage))
def test_get_stages(self): stage = Stage() pipeline = Pipeline(stages=[stage]) workflow_engine = WorkflowEngine(pipeline) eq_([stage], workflow_engine._get_stages(pipeline))
def test_get_action(self): action, workflow, stage, pipeline = self._pipeline_helper() workflow_engine = WorkflowEngine(pipeline) eq_(action, workflow_engine._get_action(1))