本文介绍了正在修补日期时间.timedelta.Total_Second的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！
问题描述
我为Web应用程序编写单元测试，我应该更改函数等待时间TIME_TO_WAIT来测试一些模块。 代码示例：import time
from datetime import datetime as dt
def function_under_test():
TIME_TO_WAIT = 300
start_time = dt.now()
while True:
if (dt.now() - start_time).total_seconds() > TIME_TO_WAIT:
break
time.sleep(1)
我看到了一种使用日期时间.timedelta.Total_Second()的补丁来解决此问题的方法，但我不知道如何正确执行此操作。
谢谢。
推荐答案
正如我在评论中写的-我将修补dt和time，以便控制测试执行的速度，如下所示：
from unittest import TestCase
from mock import patch
from datetime import datetime
from tested.module import function_under_test
class FunctionTester(TestCase):
@patch('tested.module.time')
@patch('tested.module.dt')
def test_info_query(self, datetime_mock, time_mock):
datetime_mock.now.side_effect = [
datetime(year=2000, month=1, day=1, hour=0, minute=0, second=0),
datetime(year=2000, month=1, day=1, hour=0, minute=5, second=0),
# this should be over the threshold
datetime(year=2000, month=1, day=1, hour=0, minute=5, second=1),
]
value = function_under_test()
# self.assertEquals(value, ??)
self.assertEqual(datetime_mock.now.call_count, 3)
这篇关于正在修补日期时间.timedelta.Total_Second的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！
本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除！