def test_prompt_user_not_yes(self):
"""Test output prompting user if they want to upload (assume yes was not specified)"""
test_videos = {"v1", "v2", "v3"}
expected = "Found videos:\n"
joined = "\n".join(test_videos)
expected += joined
try:
sys.stdout = self.out
with mock.patch('__builtin__.raw_input', return_value='no'):
youtube.prompt_user(test_videos, confirmation=False)
output = self.out.getvalue().strip()
self.assertEqual(expected, output)
finally:
sys.stdout = self.saved_stdout
def test_prompt_user_not_yes(self):
"""Test output prompting user if they want to upload (assume yes was not specified)"""
test_videos = {"v1", "v2", "v3"}
expected = "Found videos:\n"
joined = "\n".join(test_videos)
expected += joined
try:
sys.stdout = self.out
with mock.patch("__builtin__.raw_input", return_value="no"):
youtube.prompt_user(test_videos, confirmation=False)
output = self.out.getvalue().strip()
self.assertEqual(expected, output)
finally:
sys.stdout = self.saved_stdout
def test_prompt_user_yes(self):
"""Test assume yes (assume yes was not specified)"""
self.assertTrue(
youtube.prompt_user({"v1", "v2", "v3"}, confirmation=True))
def test_prompt_user_yes(self):
"""Test assume yes (assume yes was not specified)"""
self.assertTrue(youtube.prompt_user({"v1", "v2", "v3"}, confirmation=True))