关于mocking：如何在python中模拟“+”运算符(特别是datetime.date + datetime.timedelta)

How to mock the “+” operator in python (specifically datetime.date + datetime.timedelta)

我已经在Django中完成了一些日期模拟问题，并且最后的障碍(我希望)是以下情况。我有一个FakeDate类，它派生自datetime.date，它会模拟出来。

FakeDate类按预期工作，但是在向FakeDate添加datetime.timedelta时遇到问题，因为它返回真正的datetime.date，而不是模拟。这很重要，因为在第三方库的其他地方有一个isinstance(value, datetime.date)检查，在使用timedelta时总是会失败。

1
2
3
4
5
6
7
8
9
10
11
12
13
14

>>> import mock
>>> import datetime
>>>
>>> class FakeDate(datetime.date):
... @classmethod
... def today(cls):
... return cls(1999, 12, 31)
...
>>> FakeDate.today()
FakeDate(1999, 12, 31)
>>> FakeDate(2000, 1, 1)
FakeDate(2000, 1, 1)
>>> FakeDate(1999, 12, 31) + datetime.timedelta(days=1)
datetime.date(2000, 1, 1)

我希望FakeDate + timedelta添加返回一个FakeDate对象而不是datetime.date对象 - 我想这会涉及以某种方式修补timedelta - 但是我如何/在哪里可以做到这一点？

将__add__方法添加到FakeDate()类：

1
2
3
4
5
6
7

class FakeDate(datetime.date):
@classmethod
def today(cls):
return cls(1999, 12, 31)
def __add__(self, other):
res = super(FakeDate, self).__add__(other)
return type(self)(res.year, res.month, res.day)

演示：

1
2
3
4
5
6
7
8
9
10

>>> class FakeDate(datetime.date):
... @classmethod
... def today(cls):
... return cls(1999, 12, 31)
... def __add__(self, other):
... res = super(FakeDate, self).__add__(other)
... return type(self)(res.year, res.month, res.day)
...
>>> FakeDate.today() + datetime.timedelta(days=1)
FakeDate(2000, 1, 1)

请注意，您可以在此处简单地将实际添加委托给datetime.date类; 我们需要做的就是将结果转换回FakeDate()实例。