python中mock的断言使用-Python教程-PHP中文网

python中mock的断言使用

舞夢輝影

发布： 2025-10-23 23:26:01

原创

575人浏览过

答案：Python中使用unittest.mock的断言方法验证模拟对象调用情况，如assert_called_once_with检查调用次数和参数。通过@mock.patch替换目标方法，结合call_count和assert_any_call可验证多次调用的参数，确保函数行为正确。

python中mock的断言使用

在Python中使用mock进行断言，主要是为了验证模拟对象的方法是否被正确调用。常用的方法来自unittest.mock模块，比如assert_called()、assert_called_once()、assert_called_with()等。这些断言方法帮助我们检查函数或方法的调用情况。

1. 常见的mock断言方法

以下是常用的mock断言方法及其用途：

assert_called()：确认方法至少被调用过一次。
assert_called_once()：确认方法只被调用了一次。
assert_called_with(*args, **kwargs)：确认最后一次调用时传入了指定的参数。
assert_any_call(*args, **kwargs)：确认在某次调用中使用了指定的参数（不管是不是最后一次）。
assert_not_called()：确认方法从未被调用。

2. 实际使用示例

假设有一个发送邮件的函数，我们想测试它是否正确调用了send_email方法。

from unittest import mock
import unittest
<p>def notify_user(email, message):
send_email(email, message)  # 假设这是要mock的方法</p><h1>测试类</h1><p>class TestNotification(unittest.TestCase):</p><p><span>立即学习</span>“<a href="https://pan.quark.cn/s/00968c3c2c15" style="text-decoration: underline !important; color: blue; font-weight: bolder;" rel="nofollow" target="_blank">Python免费学习笔记（深入）</a>”；</p>
                    <div class="aritcle_card">
                        <a class="aritcle_card_img" href="/ai/%E6%99%BA%E8%B0%B1%E6%B8%85%E8%A8%80-%E5%85%8D%E8%">
                            <img src="https://img.php.cn/upload/ai_manual/000/000/000/175679976181507.png" alt="智谱清言 - 免费全能的AI助手">
                        </a>
                        <div class="aritcle_card_info">
                            <a href="/ai/%E6%99%BA%E8%B0%B1%E6%B8%85%E8%A8%80-%E5%85%8D%E8%">智谱清言 - 免费全能的AI助手</a>
                            <p>智谱清言 - 免费全能的AI助手</p>
                            <div class="">
                                <img src="/static/images/card_xiazai.png" alt="智谱清言 - 免费全能的AI助手">
                                <span>2</span>
                            </div>
                        </div>
                        <a href="/ai/%E6%99%BA%E8%B0%B1%E6%B8%85%E8%A8%80-%E5%85%8D%E8%" class="aritcle_card_btn">
                            <span>查看详情</span>
                            <img src="/static/images/cardxiayige-3.png" alt="智谱清言 - 免费全能的AI助手">
                        </a>
                    </div>
                <pre class='brush:python;toolbar:false;'>@mock.patch('my_module.send_email')
def test_notify_user_calls_send_email(self, mock_send):
    notify_user('user@example.com', 'Hello!')

    mock_send.assert_called_once_with('user@example.com', 'Hello!')

登录后复制

在这个例子中，我们用@mock.patch替换了send_email，然后通过assert_called_once_with确保它被正确调用了一次，并且参数匹配。

3. 检查多次调用的情况

如果一个方法被调用多次，可以使用call_args_list来查看每次调用的参数。

def broadcast_message(emails, message):
    for email in emails:
        send_email(email, message)
<p>@mock.patch('my_module.send_email')
def test_broadcast_calls_multiple_times(mock_send):
emails = ['a@example.com', 'b@example.com']
broadcast_message(emails, 'Hi all!')</p><pre class='brush:python;toolbar:false;'>assert mock_send.call_count == 2
mock_send.assert_any_call('a@example.com', 'Hi all!')
mock_send.assert_any_call('b@example.com', 'Hi all!')

登录后复制

这里通过call_count判断调用次数，再用assert_any_call确认特定参数曾被使用。

基本上就这些。掌握这些断言方法能让你的单元测试更准确地验证行为。注意别忘了打patch的作用范围和mock对象的传递方式。不复杂但容易忽略细节。

以上就是python中mock的断言使用的详细内容，更多请关注php中文网其它相关文章！