python - 如何验证一个函数是否被 pytest 命中

Question

我试过这个:

def test_send_confirm_hit(monkeypatch):
    hit = False
    def called():
        global hit
        hit = True

    monkeypatch.setattr("web.email.send_confirm", called)

    # ... some event that will cause web.email.send_confirm to be hit

    assert hit  # verify send_confirm was hit

尽管我宁愿不使用全局变量，但这似乎有效。做这个的最好方式是什么？

Answer 1

如果您使用“适当的”模拟，它会带有 .assert_called() method :

import unittest.mock


def test_send_confirm_hit(monkeypatch):
    mock_send_confirm = unittest.mock.Mock()

    monkeypatch.setattr("web.email.send_confirm", mock_send_confirm)

    # ... some event that will cause web.email.send_confirm to be hit

    mock_send_confirm.assert_called()