python - 如何在所有应用程序函数中模拟 Django 的 now 函数

Question

我试图模拟 Django 的 now() 函数以欺骗我的应用程序中使用的时间。我可以轻松地在我的测试文件中模拟 now() 函数，但模拟替换似乎并没有递归地渗透到我的应用程序函数中。这是我正在使用的代码:

# file - tests.py:

import datetime
import pytz
import mock

from django.test import TestCase

# this is the fake time I am using
TESTING_DJNOW = pytz.timezone('US/Central').localize(datetime.datetime(2016, 6, 14, 8, 0))


# This is the function that replaces django.utils.timezone.now()
def mocked_djnow():
    return TESTING_DJNOW

@mock.patch('django.utils.timezone.now', side_effect=mocked_djnow)
class ViewsTestCase(TestCase):
    fixtures = ['users.json', 'views_data.json'] # our initial test data

    def setUp(self):
        self.client = Client()
        self.client.login(username='fred', password='secret')

    def test_view(self, *args):
        from django.utils.timezone import now
        tm = now() # returns datetime.datetime(2016, 6, 14, 8, 0, 
                   # tzinfo=<DstTzInfo 'US/Central' CDT-1 day, 19:00:00 DST>)

        resp = self.client.get(reverse('myapp:viewfunc1'))



# file - myapp.views.py:

from django.utils.timezone import localtime, now

@login_required
def viewfunc1(request):
    # returns datetime.datetime(2016, 7, 16, 1, 11, 6, 964624, tzinfo=<UTC>)
    tm = now() # returns current datetime

是否可以在整个应用程序中修补像 now() 这样的 Django 函数？如果是这样，我做错了什么？还有其他建议吗？

Answer 1

我个人不喜欢使用 mock 来约会，因为这会带来持续的痛苦。相反，我建议你给 FreezeGun一试。它具有您在测试和任何地方需要的各种日期时间调整实用程序。

带有时区的文档示例:

from freezegun import freeze_time

@freeze_time("2012-01-14 03:21:34", tz_offset=-4)
def test():
    assert datetime.datetime.utcnow() == datetime.datetime(2012, 1, 14, 3, 21, 34)
    assert datetime.datetime.now() == datetime.datetime(2012, 1, 13, 23, 21, 34)

    # datetime.date.today() uses local time
    assert datetime.date.today() == datetime.date(2012, 1, 13)

它还有很酷的功能，例如手动标记和上下文管理器。该库大约有 500 行代码，写得非常好，因此您可以在一个小时内从头到尾阅读它以了解其工作原理。