logic.py

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()

这对你有用吗？

Twisted deferToThread，不使用Mock.patch（）

logic.py

tests.py

1 个答案: