def workflow_context(self, tmpdir): workflow_context = mock.context.simple( storage.get_sqlite_api_kwargs(str(tmpdir)), resources_dir=str(tmpdir.join('resources'))) workflow_context.states = [] workflow_context.exception = None yield workflow_context storage.release_sqlite_storage(workflow_context.model)
def storage(): api_kwargs = test_storage.get_sqlite_api_kwargs() workflow_storage = application_model_storage(SQLAlchemyModelAPI, api_kwargs=api_kwargs) workflow_storage.blueprint.put(models.get_blueprint()) blueprint = workflow_storage.blueprint.get_by_name(models.BLUEPRINT_NAME) workflow_storage.deployment.put(models.get_deployment(blueprint)) yield workflow_storage test_storage.release_sqlite_storage(workflow_storage)
def ctx(): """ Create the following graph in storage: dependency_node <------ dependent_node :return: """ simple_context = mock.context.simple(storage.get_sqlite_api_kwargs()) simple_context.model.execution.put(mock.models.get_execution(simple_context.deployment)) yield simple_context storage.release_sqlite_storage(simple_context.model)
def context(tmpdir): result = mock.context.simple(storage.get_sqlite_api_kwargs(str(tmpdir))) yield result storage.release_sqlite_storage(result.model)
def ctx(tmpdir): context = mock.context.simple(storage.get_sqlite_api_kwargs(str(tmpdir))) yield context storage.release_sqlite_storage(context.model)
def _empty_storage(): return application_model_storage(sql_mapi.SQLAlchemyModelAPI, api_kwargs=get_sqlite_api_kwargs())
def memory_model_storage(): result = aria.application_model_storage( SQLAlchemyModelAPI, api_kwargs=storage.get_sqlite_api_kwargs()) yield result storage.release_sqlite_storage(result)
def context(tmpdir): result = mock.context.simple(storage.get_sqlite_api_kwargs(str(tmpdir)), resources_dir=str(tmpdir.join('resources')), workdir=str(tmpdir.join('workdir'))) yield result storage.release_sqlite_storage(result.model)
def test_task_graph_into_execution_graph(): operation_name = 'tosca.interfaces.node.lifecycle.Standard.create' task_context = mock.context.simple(storage.get_sqlite_api_kwargs()) node_instance = \ task_context.model.node_instance.get_by_name(mock.models.DEPENDENCY_NODE_INSTANCE_NAME) def sub_workflow(name, **_): return api.task_graph.TaskGraph(name) with context.workflow.current.push(task_context): test_task_graph = api.task.WorkflowTask(sub_workflow, name='test_task_graph') simple_before_task = api.task.OperationTask.node_instance( instance=node_instance, name=operation_name) simple_after_task = api.task.OperationTask.node_instance( instance=node_instance, name=operation_name) inner_task_graph = api.task.WorkflowTask(sub_workflow, name='test_inner_task_graph') inner_task = api.task.OperationTask.node_instance( instance=node_instance, name=operation_name) inner_task_graph.add_tasks(inner_task) test_task_graph.add_tasks(simple_before_task) test_task_graph.add_tasks(simple_after_task) test_task_graph.add_tasks(inner_task_graph) test_task_graph.add_dependency(inner_task_graph, simple_before_task) test_task_graph.add_dependency(simple_after_task, inner_task_graph) # Direct check execution_graph = DiGraph() core.translation.build_execution_graph(task_graph=test_task_graph, execution_graph=execution_graph) execution_tasks = topological_sort(execution_graph) assert len(execution_tasks) == 7 expected_tasks_names = [ '{0}-Start'.format(test_task_graph.id), simple_before_task.id, '{0}-Start'.format(inner_task_graph.id), inner_task.id, '{0}-End'.format(inner_task_graph.id), simple_after_task.id, '{0}-End'.format(test_task_graph.id) ] assert expected_tasks_names == execution_tasks assert isinstance(_get_task_by_name(execution_tasks[0], execution_graph), core.task.StartWorkflowTask) _assert_execution_is_api_task( _get_task_by_name(execution_tasks[1], execution_graph), simple_before_task) assert isinstance(_get_task_by_name(execution_tasks[2], execution_graph), core.task.StartSubWorkflowTask) _assert_execution_is_api_task( _get_task_by_name(execution_tasks[3], execution_graph), inner_task) assert isinstance(_get_task_by_name(execution_tasks[4], execution_graph), core.task.EndSubWorkflowTask) _assert_execution_is_api_task( _get_task_by_name(execution_tasks[5], execution_graph), simple_after_task) assert isinstance(_get_task_by_name(execution_tasks[6], execution_graph), core.task.EndWorkflowTask) storage.release_sqlite_storage(task_context.model)
def ctx(tmpdir): context = mock.context.simple(storage.get_sqlite_api_kwargs(), resources_dir=str(tmpdir.join('resources'))) yield context storage.release_sqlite_storage(context.model)