def test_new_dir_no_input(self): with patch('builtins.input') as handle_input: tmp_dir = os.path.join(tempfile.gettempdir(), "am_mailroom_unittest") handle_input.side_effect = [tmp_dir, "n"] output = mailroom.get_user_output_path() assert output is None
def test_new_dir_yes_input(self): with patch('builtins.input') as handle_input: tmp_dir = os.path.join(tempfile.gettempdir(), "am_mailroom_unittest") with patch("os.makedirs") as make_dirs: handle_input.side_effect = [tmp_dir, "y"] output = mailroom.get_user_output_path() assert output == tmp_dir make_dirs.assert_called_once_with(tmp_dir)
def test_new_invalid_input(self): interceptor, hold_stdout = intercept_stdout() with patch('builtins.input') as handle_input: tmp_dir = os.path.join(tempfile.gettempdir(), "am_mailroom_unittest") handle_input.side_effect = [tmp_dir, "stuff", "n"] output = mailroom.get_user_output_path() assert output is None assert interceptor.getvalue().count( "Invalid choice. Please enter \"Yes\" or \"No\"") == 1 sys.stdout = hold_stdout
def test_existing_dir_input(self): with patch('builtins.input') as handle_input: handle_input.return_value = tempfile.gettempdir() output = mailroom.get_user_output_path() assert output == tempfile.gettempdir()
def test_cancel_input(self): with patch('builtins.input') as handle_input: handle_input.return_value = "cancel" output = mailroom.get_user_output_path() assert output == "cancel"
def test_empty_input(self): with patch('builtins.input') as handle_input: handle_input.return_value = "" output = mailroom.get_user_output_path() assert output is None