python - 如何覆盖当前时间？

Question

我正在编写一个使用当前时间的应用程序(通过 datetime.datetime.now())

有没有一种方法可以出于调试目的覆盖此调用，以便它返回特定的时间戳？

作为回退，我可以编写一个函数来代替 datetime.datetime.now() 并返回所需的任何内容(生产中的实际当前时间和调试时所需的测试时间) 但可能有更 pythonic 的方式来执行这些类型的操作？

Answer 1

大体上，您的选择是:

• 使用unittest.mock library ，它可以用一个总是给出相同结果的虚拟函数(或使用另一个做同样事情的模拟库)来动态地替换函数。这意味着您不必修改您的功能；然而，对于用 mock 进行 monkey patching 是否是一种好的做法，即使是为了调试，理性的人也会持不同意见。我认为这是 Python 中针对此问题使用最广泛的解决方案。
• 修改您的函数以根据其环境(系统上的实际环境变量、全局状态或其他)执行不同的操作。这是最简单的方式，也是最粗暴和最脆弱的方式，所以一定要在调试完成后改回来。
• 修改您的函数以接受一个函数本身作为参数，并在正常操作中将 datetime.datetime.now 作为该函数传递，但传递一些不同的东西(例如 stub )用于测试。