Question

我是Python的新手，我理解测试，但是，我无法解决使用Mocked Objects和side_effects的问题。

这是我的方法：

@retry(every=RETRY_EVERY, until=RETRY_UNTIL)
@unique()
@sessionized(0)
def record_click(session, queue, mailing_id, member_id, link_id, timestamp, user_agent):
    message = session.query(Message).get((mailing_id, member_id))
    mailing = session.query(Mailing).get(mailing_id)
    # More code here

这是我的测试：

@mock.patch("audience.jobs.EventProvider")
@mock.patch("audience.jobs.enqueue_webhook")
@mock.patch("logging.exception")
@mock.patch("audience.jobs.audience_queues")
@mock.patch("audience.jobs.Session")
@mock.patch("audience.jobs.DatabaseConnector")
def test_track_click_publishes_event_to_sns(self, DatabaseConnector, Session, audience_queues, logger, enqueue_webhook, EventProvider):
    message_mock = mock.Mock(account_id=77)
    message_mock.record_open.return_value = True
    mailing_mock = mock.Mock(mailing_id=123)
    mailing_mock.recipient_groups.return_value = [111]
    session_query = Session.return_value.query.return_value
    session_query.side_effect = lambda arg: message_mock if isinstance(arg, tuple) else mailing_mock

    result = jobs.record_click(
        888,
        9999,
        2048,
        datetime.datetime(1999, 12, 31, 23, 59, 59, 999999).isoformat(),
        "Mozilla/5.0")

    self.assertIsNone(result)
    self.assertListEqual(EventProvider.mock_calls, [
        mock.call(),
        mock.call().publish_link_clicked(
            headers={'User-Agent': 'Mozilla/5.0'},
            mailing_id=888,
            account_id=77,
            contact_id=9999,
            link_id=2048,
            group_ids=[111]
        )
    ])
    self.assertListEqual(logger.mock_calls, [])

我一直收到的错误是：

而不是
call().publish_link_clicked(group_ids=[111], account_id=77, **etc)

这就是UnitTest中所谓的内容 call().publish_link_clicked(group_ids=<MagicMock name='Session().query().get().recipient_groups' id='4557662736'>, account_id=<MagicMock name='Session().query().get().account_id' id='4557652048'>, **etc)

我做错了什么？

Answer 1

请勿致电Session()或query();使用Mock.return_value attribute代替遍历调用图：

Session.return_value.query.return_value.side_effect = lambda arg: message_mock if isinstance(arg, tuple) else mailing_mock

我通常使用中间名来保存返回值：

session_query = Session.return_value.query.return_value
session_query.side_effect = lambda arg: message_mock if isinstance(arg, tuple) else mailing_mock

您还需要修补正确的Session课程;这完全取决于您的代码如何为session生成record_click参数。有关详细信息，请参阅Where to Patch。如果@sessionized装饰器产生此参数，并且不存在于audience.jobs模块中，则表示您没有修补正确的位置。

了解模拟和副作用

1 个答案: