我想模拟计时,以便能够在我的测试中设置某个时间DateTimeField
auto_now_add=True
字段,例如:
class MyModel:
...
created_at = models.DateTimeField(auto_now_add=True)
...
class TestMyModel(TestCase):
...
def test_something(self):
# mock current time so that `created_at` be something like 1800-02-09T020000
my_obj = MyModel.objects.create(<whatever>)
# and here my_obj.created_at == 1800-02-09T000000
我知道当前日期always用于此类字段,这就是为什么我正在寻找替代模拟系统时序的方法,但只是在背景。
我尝试了一些方法,例如,使用freeze_time
创建上下文但不起作用:
with freeze_now("1800-02-09"):
MyModel.objects.create(<whatever>)
# here the created_at doesn't fit 1800-02-09
Ofc我想,这是由于auto_now_add=True
时对象创建方式背后的机制。
我不想删除auto_now_add=True
并使用默认值。
有没有办法可以模拟时间,以便我们可以做到这一点,这种领域在某些情境下得到我想要的时间?
我正在使用Django 1.9.6
和Python 3.4
答案 0 :(得分:10)
好的,我找到了一个解决方案,它基于mock:
def mock_now():
return <mock time>
class TestMyModel(TestCase):
...
@mock.patch('django.utils.timezone.now', mock_now)
def test_as_decorator(self):
...
my_obj = MyModel.objects.create(<whatever>)
...
# here the created_at field has the mocking time :)
def test_as_context_manager(self):
mocked_dt = datetime.datetime(2015, 9, 3, 11, 15, 0)
with mock.patch('django.utils.timezone.now', mock.Mock(return_value=mocked_dt)):
my_obj = MyModel.objects.create(<whatever>)
# here the created_at field has the mocking time :)
答案 1 :(得分:0)
扩展@trinchets的答案,这里是上下文管理器,
from datetime import timedelta
from django.utils import timezone
from mock import patch, Mock
last_week = timezone.now() - timedelta(weeks=1)
with patch('django.utils.timezone.now', Mock(return_value=last_week)):
# Make your models/factories
# Carry on with normal time
答案 2 :(得分:0)
你可以试试这个
from datetime import datetime
from django.test import TestCase
from mock import patch
class TestMyModel(TestCase):
@patch('django.utils.timezone.now', )
def test_something(self, mock_now):
mock_now.return_value = datetime(2015, 9, 3, 11, 15, 0)