diff options
Diffstat (limited to 'tests/migrations/test_questioner.py')
| -rw-r--r-- | tests/migrations/test_questioner.py | 54 |
1 files changed, 27 insertions, 27 deletions
diff --git a/tests/migrations/test_questioner.py b/tests/migrations/test_questioner.py index 472a3f248f..c1aebcb224 100644 --- a/tests/migrations/test_questioner.py +++ b/tests/migrations/test_questioner.py @@ -4,7 +4,8 @@ from unittest import mock from django.core.management.base import OutputWrapper from django.db.migrations.questioner import ( - InteractiveMigrationQuestioner, MigrationQuestioner, + InteractiveMigrationQuestioner, + MigrationQuestioner, ) from django.db.models import NOT_PROVIDED from django.test import SimpleTestCase @@ -13,21 +14,25 @@ from django.test.utils import override_settings class QuestionerTests(SimpleTestCase): @override_settings( - INSTALLED_APPS=['migrations'], - MIGRATION_MODULES={'migrations': None}, + INSTALLED_APPS=["migrations"], + MIGRATION_MODULES={"migrations": None}, ) def test_ask_initial_with_disabled_migrations(self): questioner = MigrationQuestioner() - self.assertIs(False, questioner.ask_initial('migrations')) + self.assertIs(False, questioner.ask_initial("migrations")) def test_ask_not_null_alteration(self): questioner = MigrationQuestioner() - self.assertIsNone(questioner.ask_not_null_alteration('field_name', 'model_name')) + self.assertIsNone( + questioner.ask_not_null_alteration("field_name", "model_name") + ) - @mock.patch('builtins.input', return_value='2') + @mock.patch("builtins.input", return_value="2") def test_ask_not_null_alteration_not_provided(self, mock): - questioner = InteractiveMigrationQuestioner(prompt_output=OutputWrapper(StringIO())) - question = questioner.ask_not_null_alteration('field_name', 'model_name') + questioner = InteractiveMigrationQuestioner( + prompt_output=OutputWrapper(StringIO()) + ) + question = questioner.ask_not_null_alteration("field_name", "model_name") self.assertEqual(question, NOT_PROVIDED) @@ -36,17 +41,17 @@ class QuestionerHelperMethodsTests(SimpleTestCase): self.prompt = OutputWrapper(StringIO()) self.questioner = InteractiveMigrationQuestioner(prompt_output=self.prompt) - @mock.patch('builtins.input', return_value='datetime.timedelta(days=1)') + @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='') + @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)') + value = self.questioner._ask_default(default="datetime.timedelta(days=1)") self.assertEqual(value, datetime.timedelta(days=1)) - @mock.patch('builtins.input', side_effect=['', 'exit']) + @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() @@ -55,31 +60,26 @@ class QuestionerHelperMethodsTests(SimpleTestCase): self.prompt.getvalue(), ) - @mock.patch('builtins.input', side_effect=['bad code', 'exit']) + @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()) + self.assertIn("Invalid input: ", self.prompt.getvalue()) - @mock.patch('builtins.input', side_effect=['', 'n']) + @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?') + value = self.questioner._boolean_input("Proceed?") self.assertIs(value, False) - @mock.patch('builtins.input', return_value='') + @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) + value = self.questioner._boolean_input("Proceed?", default=True) self.assertIs(value, True) - @mock.patch('builtins.input', side_effect=[10, 'garbage', 1]) + @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' - ) + 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) |