def test_interactive(self):
node1 = FakeNode()
repo = MagicMock()
repo.get_node.return_value = node1
args = MagicMock()
args.item_workers = 4
args.target = "node1"
verify.bw_verify(repo, args)
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_interactive(self):
node1 = FakeNode()
repo = MagicMock()
repo.get_node.return_value = node1
args = MagicMock()
args.force = False
args.interactive = True
args.item_workers = 4
args.target = "node1"
output = list(bw_apply(repo, args))
self.assertTrue(output[0].startswith("nodename: run started at "))
self.assertTrue(output[1].startswith("nodename: run completed after "))
self.assertTrue(output[1].endswith("(0 OK, 0 fixed, 0 skipped, 0 failed)\n"))
self.assertEqual(len(output), 2)
