python – 如何使用datetime对方法进行单元测试？

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 如何使用datetime对方法进行单元测试？，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含2001字，纯文字阅读大概需要3分钟。

内容图文

我有以下类和方法：

class DateTimeHelper(object):

    @staticmethod
    def get_utc_millisecond_timestamp():
        (dt, micro) = datetime.datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S.%f').split('.')
        return "%s.%03d" % (dt, int(micro) / 1000)  # UTC time with millisecond

我怎样才能对它进行单元测试？尽管这很简单,但我完全被难倒了.这是我的第一次单元测试.

解决方法:

使用unittest.mock library(Python 3.3及更高版本,向后移植为mock)来替换对被测代码外部任何代码的调用.

在这里,我不仅要模拟utcnow()而且还要模拟strftime(),只返回一个字符串对象：

with mock.patch('datetime.datetime') as dt_mock:
    dt_mock.utcnow.return_value.strftime.return_value = '2016-08-04 12:22:44.123456'
    result = DateTimeHelper.get_utc_millisecond_timestamp()

如果您认为测试strftime()参数很重要,请为dt_mock.utcnow.return_value指定一个显式的datetime对象来返回;你必须在模拟之前创建该测试对象,因为你不能只模拟datetime.datetime.utcnow类方法：

testdt = datetime.datetime(2016, 8, 4, 12, 22, 44, 123456)
with mock.patch('datetime.datetime') as dt_mock:
    dt_mock.utcnow.return_value = testdt
    result = DateTimeHelper.get_utc_millisecond_timestamp()

或者,在您的单元测试中,使用datetime import datetime来保留对未模拟的类的引用.

演示：

>>> from unittest import mock
>>> import datetime
>>> class DateTimeHelper(object):
...     @staticmethod
...     def get_utc_millisecond_timestamp():
...         (dt, micro) = datetime.datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S.%f').split('.')
...         return "%s.%03d" % (dt, int(micro) / 1000)  # UTC time with millisecond
...
>>> with mock.patch('datetime.datetime') as dt_mock:
...     dt_mock.utcnow.return_value.strftime.return_value = '2016-08-04 12:22:44.123456'
...     result = DateTimeHelper.get_utc_millisecond_timestamp()
...
>>> result
'2016-08-04 12:22:44.123'
>>> testdt = datetime.datetime(2016, 8, 4, 12, 22, 44, 123456)
>>> with mock.patch('datetime.datetime') as dt_mock:
...     dt_mock.utcnow.return_value = testdt
...     result = DateTimeHelper.get_utc_millisecond_timestamp()
...
>>> result
'2016-08-04 12:22:44.123'

内容总结

以上是互联网集市为您收集整理的python – 如何使用datetime对方法进行单元测试？全部内容，希望文章能够帮你解决python – 如何使用datetime对方法进行单元测试？所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端