python - 我可以出于测试目的更改今天的日期吗？

Question

这个问题已经有答案了:

Trying to mock datetime.date.today(), but not working (23 个回答)

已关闭 4 年前。

我有一个函数可以根据今天的日期执行一些逻辑:

def do_work()
  todays_date = datetime.datetime.today().date()
  #do something based on todays_date

我想编写一些单元测试来验证此函数在一年中的特定日期是否执行其应有的操作。有什么方法可以更改存储在 todays_date 中的值不改变功能？

我怀疑不是，但只是想问一下。

Answer 1

mock module在这种情况下可以派上用场。该文档有an example关于如何部分模拟固定日期。

>>> from datetime import date
>>> with patch('mymodule.date') as mock_date:
...     mock_date.today.return_value = date(2010, 10, 8)
...     mock_date.side_effect = lambda *args, **kw: date(*args, **kw)
...
...     assert mymodule.date.today() == date(2010, 10, 8)
...     assert mymodule.date(2009, 6, 8) == date(2009, 6, 8)
...