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(repo.bw_repo_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(repo.bw_repo_test(repo_obj, args)) exit.assert_called_once_with(1)