试图模拟 datetime.date.today()，但不工作

小编典典

试图模拟 datetime.date.today()，但不工作

all

谁能告诉我为什么这不起作用？

>>> import mock
>>> @mock.patch('datetime.date.today')
... def today(cls):
...  return date(2010, 1, 1)
...
>>> from datetime import date
>>> date.today()
datetime.date(2010, 12, 19)

也许有人可以提出更好的方法？

阅读 66

2022-06-18

共1个答案

小编典典

有几个问题。

首先，您使用的方式mock.patch不太正确。当用作装饰器时，它会将给定的函数/类（在这种情况下，datetime.date.today）替换为
仅在装饰函数内的Mock对象。所以，只有在你的意志中有一个不同的功能，这似乎不是你想要的。
__today()``datetime.date.today

你真正想要的似乎更像是这样的：

@mock.patch('datetime.date.today')
def test():
    datetime.date.today.return_value = date(2010, 1, 1)
    print datetime.date.today()

不幸的是，这不起作用：

>>> test()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "build/bdist.macosx-10.6-universal/egg/mock.py", line 557, in patched
  File "build/bdist.macosx-10.6-universal/egg/mock.py", line 620, in __enter__
TypeError: can't set attributes of built-in/extension type 'datetime.date'

这失败了，因为 Python 内置类型是不可变的 -
有关更多详细信息，请参阅此答案。

使用 CASE、IIF 和 UDF 的解决方案是足够的，但在使用超过 2 个比较值将问题扩展到一般情况时是不切实际的。SQL Server 2008+ 中的通用解决方案利用了 VALUES 子句的一个奇怪应用：

SELECT
PaidForPast=(SELECT MIN(x) FROM (VALUES (PaidThisMonth),(OwedPast)) AS value(x))

在这种情况下，我将自己继承 datetime.date 并创建正确的函数：

import datetime
class NewDate(datetime.date):
    @classmethod
    def today(cls):
        return cls(2010, 1, 1)
datetime.date = NewDate

现在你可以这样做：

>>> datetime.date.today()
NewDate(2010, 1, 1)

2022-06-18