Question

如何解决这个问题？用另一个签名修补一个对象方法（例如一个额外的参数。我试图绑定可选参数，但这似乎不起作用。我不能在这里使用普通的猴子补丁，因为补丁类被调用在一个我不能以这种方式打补丁的地方。

任何帮助表示感谢。

import mock
import functools

# this class lives in another (unchangeable) module, __len__ method has to be patched
class ToOverride(object):
    def __len__(self):
        raise NotImplementedError()

# this code is changeable
def my_len(self, arg):
    return arg+1

my_len_bound = functools.partial(my_len, arg=1)

with mock.patch.object(ToOverride, '__len__', my_len_bound):
    inst = ToOverride()
    print len(inst) # expected output: 2

当调用上下文mock.patch.object时，我收到以下错误：

TypeError                                 Traceback (most recent call last)
<ipython-input-7-bfdb41d8628f> in <module>()
      1 with mock.patch.object(ToOverride, '__len__', my_len_bound):
      2     inst = ToOverride()
----> 3     print len(inst)

TypeError: my_len() takes exactly 2 arguments (1 given)

但是使用None作为第一个参数调用my_len按预期工作（打印出来2）。

假设可以简单地修补补丁，如果用实例作为第一个参数手动调用 len ，它就可以工作。但这当然是不受欢迎的：

ToOverride.__len__ = my_len_bound
inst = ToOverride()
print( inst.__len__(inst)) # 2

Answer 1

通过使用闭包模式，可以避免使用functools partial，这对类方法不起作用。

def bind_arg(arg):
    def my_len(self, arg):
        return arg+1
    return my_len

可以使用functool.partial绑定参数的Python mock.patch.object吗？

1 个答案: