本文介绍了Python:如何模拟 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()?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持跟版网！
本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除！