def _gen_ctx(self): _ctx = MockCloudifyContext( deployment_id="deployment_id" ) _ctx._subgraph = [] def _subgraph(instance_id): _subgraph = Mock() _subgraph.instance_id = "subgraph" + instance_id _ctx._subgraph.append(_subgraph) return _subgraph _graph = Mock() _graph.id = 'i_am_graph' _graph.tasks_iter = Mock(return_value=['task1']) _graph.remove_task = Mock(return_value=None) _graph.subgraph = _subgraph _ctx.nodes = [] _ctx.graph_mode = Mock(return_value=_graph) _ctx._graph = _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