python - 修补函数的__call__

我需要在测试中修补当前日期时间。我正在使用此解决方案:

def _utcnow():
    return datetime.datetime.utcnow()


def utcnow():
    """A proxy which can be patched in tests.
    """
    # another level of indirection, because some modules import utcnow
    return _utcnow()

然后在我的测试中，我做类似的事情:

    with mock.patch('***.utils._utcnow', return_value=***):
        ...

但是今天有了一个主意，我可以通过修补函数__call__的utcnow而不是额外的_utcnow来简化实现。

这对我不起作用:

    from ***.utils import utcnow
    with mock.patch.object(utcnow, '__call__', return_value=***):
        ...

如何优雅地做到这一点？

最佳答案

修补函数的__call__时，要设置该实例的__call__属性。 Python实际上调用了在类上定义的__call__方法。

例如:

>>> class A(object):
...     def __call__(self):
...         print 'a'
...
>>> a = A()
>>> a()
a
>>> def b(): print 'b'
...
>>> b()
b
>>> a.__call__ = b
>>> a()
a
>>> a.__call__ = b.__call__
>>> a()
a

将任何东西分配给a.__call__是没有意义的。

然而:

>>> A.__call__ = b.__call__
>>> a()
b

TLDR；
a()不调用a.__call__。它调用type(a).__call__(a)。

链接

有一个很好的解释，为什么在answer to "Why type(x).__enter__(x) instead of x.__enter__() in Python standard contextlib?"中会发生这种情况。

在Special method lookup上的Python文档中记录了此行为。

关于python - 修补函数的__call__，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/34261111/