def test_ok(self):
node1 = FakeNode()
repo_obj = MagicMock()
repo_obj.nodes = (node1,)
args = MagicMock()
args.item_workers = 4
args.node_workers = 1
args.target = None
list(test.bw_test(repo_obj, args))
def test_fail(self, exit):
node1 = FailNode()
repo_obj = MagicMock()
repo_obj.get_node.return_value = node1
args = MagicMock()
args.item_workers = 4
args.node_workers = 1
args.target = "node1"
list(test.bw_test(repo_obj, args))
exit.assert_called_once_with(1)
def test_single_node_fail(self, get_target_nodes):
args = MagicMock()
args.command = "foo"
args.may_fail = False
args.node_workers = 2
args.sudo = True
node = FakeNode("node1")
get_target_nodes.return_value = [node]
output = list(run.bw_run(MagicMock(), args))
self.assertTrue(output[0].startswith("[node1] ✘ failed after "))
self.assertTrue(output[0].endswith("s (return code 47)"))
def test_group_success(self, get_target_nodes):
args = MagicMock()
args.command = "foo"
args.may_fail = False
args.node_workers = 2
args.sudo = True
node1 = FakeNode("node1")
node1.result.return_code = 0
node2 = FakeNode("node2")
node2.result.return_code = 0
get_target_nodes.return_value = [node1, node2]
output = list(run.bw_run(MagicMock(), args))
self.assertTrue("completed successfully after" in output[0])
self.assertTrue("completed successfully after" in output[1])
self.assertEqual(len(output), 2)
