def test_suicide_with_invalid_method_log_warning(mocker): node = Node() mocker.patch.object(node.log, 'warning') node.suicide(method="invalid_split_action", delay=6) expected_warning_calls = [ call('invalid commit suicide method %s', 'invalid_split_action'), ] node.log.warning.assert_has_calls(expected_warning_calls)
def test_suicide_call_correct_node_method(mocker, split_action, expected_call): mocker.patch.object(Node, expected_call) node = Node() node.suicide(method=split_action, delay=6) getattr(node, expected_call).assert_called_once_with(6)