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