扭曲的 deferToThread，不适用于 Mock.patch()

Question

我有 2 个函数都用 @defer.inlineCallbacks 装饰器包装。

为了测试，我正在模拟各种东西，包括 save_to_db() 函数。

logic.py

@defer.inlineCallbacks
def save_to_db(obj):  # mocked for test.
    raise Exception('Oh Noe!')


@defer.inlineCallbacks
def create_contact():
    xero = yield get_xero_client()

    data = {
        # <contact info>
    }

    # Create a new contact
    response = yield deferToThread(xero.contacts.put, data)

    obj = {
        # some data extracted from response 
    }
    yield save_to_db(obj)

tests.py

import mock

from twisted.internet import defer
from twisted.trial import unittest

from .logic import create_contact


class TestContactCreation(unittest.TestCase):

    @mock.patch('logic.save_to_db')
    @mock.patch('logic.get_xero_client')
    @defer.inlineCallbacks
    def test_get_xero_client_is_called(self, mocked_xero_client, mocked_save_method):
        yield create_contact()
        mocked_get_xero_client.assert_called()

然而当我运行:

$ trial tests.TestContactCreation

save_to_db() 实际上被调用了，正如预期的那样它引发了一个异常。

Traceback (most recent call last):

  File "<file_path>/logic.py", line 93, in save_to_db
    raise Exception('Oh Noe!')

exceptions.Exception: Oh Noe!

而且我不确定为什么！我尝试使用 pdb.

进行调试

import pdb; pdb.set_trace()

看起来 save_to_db() 在我们使用 deferToThread()

之前被正确模拟了

(Pdb) save_to_db
<MagicMock name='save_to_db' id='4404276240'>

但是在我使用 deferToThread()

的行之后

(Pdb) save_to_db
<function save_to_db at 0x111c6f488>

save_to_db()不再被嘲笑！解决这个问题的唯一方法是我也模拟 deferToThread()

还有更好的选择吗？任何提示将不胜感激。非常感谢。

Answer 1

我遇到了同样的问题； @mock.patch(...) 装饰器在与 @inlineCallbacks 装饰器一起使用时不起作用（即，它实际上并没有模拟所需的东西）。

对我有用的是通过上下文管理器进行模拟：

@defer.inlineCallbacks
def test_get_xero_client_is_called(self):
    with mock.patch('logic.save_to_db') as mocked_save_method, \
            mock.patch('logic.get_xero_client') as mocked_xero_client:
        yield create_contact()
        mocked_get_xero_client.assert_called()

这对你有用吗？

扭曲的 deferToThread，不适用于 Mock.patch()

Twisted deferToThread, not working with Mock.patch()

python

twisted

logic.py

tests.py