1
0
mirror of https://github.com/django/django.git synced 2025-10-24 06:06:09 +00:00

Fixed #29595 -- Allowed using timedelta in migrations questioner.

Refs #29600 -- Removed usage of django.utils.datetime_safe in migrations.
This commit is contained in:
Tim Graham
2018-07-26 15:04:58 -04:00
parent 3af695eda2
commit c72dde41e6
4 changed files with 29 additions and 53 deletions

View File

@@ -1,6 +1,11 @@
from django.db.migrations.questioner import MigrationQuestioner
import datetime
from unittest import mock
from django.db.migrations.questioner import (
InteractiveMigrationQuestioner, MigrationQuestioner,
)
from django.test import SimpleTestCase
from django.test.utils import override_settings
from django.test.utils import captured_stdout, override_settings
class QuestionerTests(SimpleTestCase):
@@ -11,3 +16,10 @@ class QuestionerTests(SimpleTestCase):
def test_ask_initial_with_disabled_migrations(self):
questioner = MigrationQuestioner()
self.assertIs(False, questioner.ask_initial('migrations'))
@mock.patch('builtins.input', return_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))