class QuestionerHelperMethodsTests(SimpleTestCase): def setUp(self): self.prompt = OutputWrapper(StringIO()) self.questioner = InteractiveMigrationQuestioner( prompt_output=self.prompt) @mock.patch('builtins.input', return_value='datetime.timedelta(days=1)') def test_questioner_default_timedelta(self, mock_input): value = self.questioner._ask_default() self.assertEqual(value, datetime.timedelta(days=1)) @mock.patch('builtins.input', return_value='') def test_questioner_default_no_user_entry(self, mock_input): value = self.questioner._ask_default( default='datetime.timedelta(days=1)') self.assertEqual(value, datetime.timedelta(days=1)) @mock.patch('builtins.input', side_effect=['', 'exit']) def test_questioner_no_default_no_user_entry(self, mock_input): with self.assertRaises(SystemExit): self.questioner._ask_default() self.assertIn( "Please enter some code, or 'exit' (without quotes) to exit.", self.prompt.getvalue(), ) @mock.patch('builtins.input', side_effect=['bad code', 'exit']) def test_questioner_no_default_bad_user_entry_code(self, mock_input): with self.assertRaises(SystemExit): self.questioner._ask_default() self.assertIn('Invalid input: ', self.prompt.getvalue()) @mock.patch('builtins.input', side_effect=['', 'n']) def test_questioner_no_default_no_user_entry_boolean(self, mock_input): value = self.questioner._boolean_input('Proceed?') self.assertIs(value, False) @mock.patch('builtins.input', return_value='') def test_questioner_default_no_user_entry_boolean(self, mock_input): value = self.questioner._boolean_input('Proceed?', default=True) self.assertIs(value, True) @mock.patch('builtins.input', side_effect=[10, 'garbage', 1]) def test_questioner_bad_user_choice(self, mock_input): question = 'Make a choice:' value = self.questioner._choice_input(question, choices='abc') expected_msg = (f'{question}\n' f' 1) a\n' f' 2) b\n' f' 3) c\n') self.assertIn(expected_msg, self.prompt.getvalue()) self.assertEqual(value, 1)
class QuestionerHelperMethodsTests(SimpleTestCase): def setUp(self): self.prompt = OutputWrapper(StringIO()) self.questioner = InteractiveMigrationQuestioner(prompt_output=self.prompt) @mock.patch("builtins.input", return_value="datetime.timedelta(days=1)") def test_questioner_default_timedelta(self, mock_input): value = self.questioner._ask_default() self.assertEqual(value, datetime.timedelta(days=1)) @mock.patch("builtins.input", return_value="") def test_questioner_default_no_user_entry(self, mock_input): value = self.questioner._ask_default(default="datetime.timedelta(days=1)") self.assertEqual(value, datetime.timedelta(days=1)) @mock.patch("builtins.input", side_effect=["", "exit"]) def test_questioner_no_default_no_user_entry(self, mock_input): with self.assertRaises(SystemExit): self.questioner._ask_default() self.assertIn( "Please enter some code, or 'exit' (without quotes) to exit.", self.prompt.getvalue(), ) @mock.patch("builtins.input", side_effect=["bad code", "exit"]) def test_questioner_no_default_bad_user_entry_code(self, mock_input): with self.assertRaises(SystemExit): self.questioner._ask_default() self.assertIn("Invalid input: ", self.prompt.getvalue()) @mock.patch("builtins.input", side_effect=["", "n"]) def test_questioner_no_default_no_user_entry_boolean(self, mock_input): value = self.questioner._boolean_input("Proceed?") self.assertIs(value, False) @mock.patch("builtins.input", return_value="") def test_questioner_default_no_user_entry_boolean(self, mock_input): value = self.questioner._boolean_input("Proceed?", default=True) self.assertIs(value, True) @mock.patch("builtins.input", side_effect=[10, "garbage", 1]) def test_questioner_bad_user_choice(self, mock_input): question = "Make a choice:" value = self.questioner._choice_input(question, choices="abc") expected_msg = f"{question}\n" f" 1) a\n" f" 2) b\n" f" 3) c\n" self.assertIn(expected_msg, self.prompt.getvalue()) self.assertEqual(value, 1)
def test_timedelta_default(self, mock): questioner = InteractiveMigrationQuestioner() with captured_stdout(): value = questioner._ask_default() self.assertEqual(value, datetime.timedelta(days=1))
def test_timedelta_default(self, mock): questioner = InteractiveMigrationQuestioner() with captured_stdout(): value = questioner._ask_default() self.assertEqual(value, datetime.timedelta(days=1))