def test_no_such_group(self): def get_node(name): raise NoSuchNode() def get_group(name): raise NoSuchGroup() repo = MagicMock() repo.get_group = get_group repo.get_node = get_node with self.assertRaises(UsageException): cmdline.get_target_nodes(repo, "node1")
def test_no_such_node(self): def get_node(name): raise NoSuchNode() def get_group(name): group = MagicMock() group.nodes = ["g" + name[-1]] return group repo = MagicMock() repo.get_group = get_group repo.get_node = get_node self.assertEqual( cmdline.get_target_nodes(repo, "node1"), ["g1"], )
def test_mixed(self): def get_node(name): if int(name[-1]) > 2: raise NoSuchNode() return "n" + name[-1] def get_group(name): group = MagicMock() group.nodes = ["g" + name[-1]] return group repo = MagicMock() repo.get_group = get_group repo.get_node = get_node self.assertEqual( set(cmdline.get_target_nodes(repo, "node1, node2,group3")), set(["n1", "n2", "g3"]), )