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)