Python:如何模拟datetime.utcnow()? [英] Python: How do I mock datetime.utcnow()?

问题描述

我有以下内容:

from datetime import datetime

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

如何模拟datetime.utcnow()，以便可以对此功能进行单元测试?

How do I mock datetime.utcnow() so that I can write unit test on this function?

试图阅读此一个，但是我无法在utcnow()上使它正常工作

Tried reading this one but I am unable to get it working for me on utcnow()

推荐答案

:

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)

这篇关于Python:如何模拟datetime.utcnow()?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！