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:
@@ -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))
|
||||
|
||||
Reference in New Issue
Block a user