Python:如何模拟 datetime.utcnow()？

Question

我有以下内容:

from datetime import datetime

def get_report_month_key():
    month_for_report = datetime.utcnow()
    return month_for_report.strftime("%Y%m") 

如何模拟 datetime.utcnow() 以便我可以对此函数编写单元测试？

尝试阅读此 one但我无法让它在 utcnow() 上为我工作

Answer 1

在你的测试文件中:

from yourfile import get_report_month_key
import mock
import unittest
from datetime import datetime

class TestCase(unittest.TestCase):

    @mock.patch('yourfile.datetime')
    def test_dt(self, mock_dt):
        mock_dt.utcnow = mock.Mock(return_value=datetime(1901, 12, 21))
        r = get_report_month_key()
        self.assertEqual('190112', r)