如何使用python mock覆盖lambda函数?

时间:2013-12-13 00:41:23

标签: lambda python-mock

我有一些代码使用Greenlet.spawn稍后调用一些代码。我刚刚发现在该代码中出现了异常。它会被我们的测试捕获,但是spawn使它在测试成功后运行。

我正在寻找一种修补Greenlet.spawn的方法,以便立即调用传入的函数。

我想修补测试类,而不必为测试添加参数。我已经在很多其他东西上使用了补丁,我觉得它必须能够做到这一点,但我的谷歌搜索都没有取得丰硕的成果。

有什么想法吗?

1 个答案:

答案 0 :(得分:0)

想出来:

@patch('skynet.middleware.Greenlet.spawn', classmethod(lambda cls,f: f()))

回想起来似乎很明显,猜测我一直没有做到这一点。