def _gen_ctx(self):
_ctx = MockCloudifyContext(
deployment_id="deployment_id"
)
_graph = Mock()
_graph.id = 'i_am_graph'
_graph.tasks_iter = Mock(return_value=['task1'])
_graph.remove_task = Mock(return_value=None)
_ctx.graph_mode = Mock(return_value=_graph)
_ctx.deployment.scaling_groups = {
'one_scale': {
'members': ['one'],
'properties': {'current_instances': 10}
},
'alfa_types': {
'members': ['a_type', 'b_type'],
'properties': {'current_instances': 55}
}
}
modification = Mock()
modification.id = "transaction_id"
modification.added.node_instances = []
modification.removed.node_instances = []
modification.rollback = Mock()
modification.finish = Mock()
_ctx.deployment.start_modification = Mock(return_value=modification)
_ctx._get_modification = modification
current_ctx.set(_ctx)
return _ctx
def _gen_ctx(self):
_ctx = MockCloudifyContext(deployment_id="deployment_id")
_ctx._subgraph = []
def _subgraph(instance_id):
_subgraph = Mock()
_subgraph.instance_id = "subgraph" + instance_id
sequence = Mock()
_subgraph.sequence = sequence
_ctx._subgraph.append(_subgraph)
return _subgraph
_graph = Mock()
_graph.id = u'test_graph'
_graph.subgraph = _subgraph
_ctx.nodes = []
_ctx.node_instances = []
_ctx.graph_mode = Mock(return_value=_graph)
_ctx._graph = _graph
_ctx.refresh_node_instances = Mock(return_value=None)
current_ctx.set(_ctx)
return _ctx
