def test_mark_node_failed(self): dgm = DeploymentGroupManager(yaml.safe_load(_GROUPS_YAML), node_lookup) dgm.mark_node_failed('node1') assert dgm.get_nodes(Stage.FAILED) == ['node1']
def test_get_nodes_all(self): dgm = DeploymentGroupManager(yaml.safe_load(_GROUPS_YAML), node_lookup) assert set(dgm.get_nodes()) == set( ['node1', 'node2', 'node3', 'node4', 'node5', 'node6', 'node7', 'node8', 'node9', 'node10', 'node11', 'node12'] )
def test_mark_node_prepared(self): dgm = DeploymentGroupManager(yaml.safe_load(GROUPS_YAML), node_lookup) dgm.mark_node_prepared('node1') assert dgm.get_nodes(Stage.PREPARED) == ['node1']