def test_prompt_for_user_input():
"""Make sure we prompt the user with the string provided."""
error = IOError("totes fake")
if sys.version_info > (3,):
mock_input = Mock("builtins.input", return_value="words")
input_patch = patch.object(builtins, "input", new=mock_input)
else:
mock_input = Mock("__builtin__.raw_input", return_value="words")
input_patch = patch.object(__builtin__, "raw_input", new=mock_input)
with input_patch as patched:
formatting._prompt_for_input_on_error("prompted with:", error)
patched.assert_called_once_with("prompted with:")
def test_prompt_for_user_input():
"""Make sure we prompt the user with the string provided."""
error = IOError("totes fake")
if sys.version_info > (3, ):
mock_input = Mock("builtins.input", return_value="words")
input_patch = patch.object(builtins, "input", new=mock_input)
else:
mock_input = Mock("__builtin__.raw_input", return_value="words")
input_patch = patch.object(__builtin__, "raw_input", new=mock_input)
with input_patch as patched:
formatting._prompt_for_input_on_error("prompted with:", error)
patched.assert_called_once_with("prompted with:")
def test_raise_error_during_prompt():
"""Ensure the original exception is raised if the user sends ^C."""
error = OSError("totes fake")
if sys.version_info > (3,):
mock_input = Mock("builtins.input", side_effect=KeyboardInterrupt)
input_patch = patch.object(builtins, "input", new=mock_input)
else:
mock_input = Mock(
"__builtin__.raw_input",
side_effect=KeyboardInterrupt,
)
input_patch = patch.object(__builtin__, "raw_input", new=mock_input)
with pytest.raises(OSError) as raised_error:
with input_patch as patched:
formatting._prompt_for_input_on_error("prompted with:", error)
patched.assert_called_once_with("prompted with:")
assert "OSError: totes fake" == raised_error.exconly()
def test_raise_error_during_prompt():
"""Ensure the original exception is raised if the user sends ^C."""
error = OSError("totes fake")
if sys.version_info > (3, ):
mock_input = Mock("builtins.input", side_effect=KeyboardInterrupt)
input_patch = patch.object(builtins, "input", new=mock_input)
else:
mock_input = Mock(
"__builtin__.raw_input",
side_effect=KeyboardInterrupt,
)
input_patch = patch.object(__builtin__, "raw_input", new=mock_input)
with pytest.raises(OSError) as raised_error:
with input_patch as patched:
formatting._prompt_for_input_on_error("prompted with:", error)
patched.assert_called_once_with("prompted with:")
assert "OSError: totes fake" == raised_error.exconly()
