问题描述
我是Python的新手，我了解测试，但是，我无法全神贯注地使用模拟对象和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)
我究竟做错了什么？
1楼
不要调用Session()或query() ;
请改用遍历调用图：
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类。
这完全取决于您的代码如何产生record_click的session参数。
有关更多详细信息，请参见程序。
如果@sessionized装饰产生这样的说法，它不生活在在audience.jobs模块，你没有打补丁的正确位置。