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()