86 lines
3.4 KiB
Python
86 lines
3.4 KiB
Python
import datetime
|
|
from io import StringIO
|
|
from unittest import mock
|
|
|
|
from django.core.management.base import OutputWrapper
|
|
from django.db.migrations.questioner import (
|
|
InteractiveMigrationQuestioner, MigrationQuestioner,
|
|
)
|
|
from django.db.models import NOT_PROVIDED
|
|
from django.test import SimpleTestCase
|
|
from django.test.utils import override_settings
|
|
|
|
|
|
class QuestionerTests(SimpleTestCase):
|
|
@override_settings(
|
|
INSTALLED_APPS=['migrations'],
|
|
MIGRATION_MODULES={'migrations': None},
|
|
)
|
|
def test_ask_initial_with_disabled_migrations(self):
|
|
questioner = MigrationQuestioner()
|
|
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'))
|
|
|
|
@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')
|
|
self.assertEqual(question, NOT_PROVIDED)
|
|
|
|
|
|
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)
|