如何在Python中用unittest.mock模拟邮件发送函数避免真实发送邮件？

嘿，我之前也踩过这个mock邮件发送的坑！明明写了模拟代码，结果真实邮件还是发出去了——大概率是你模拟的位置不对，这是unittest.mock最容易踩的雷之一。下面我把常见原因和解决方法给你理清楚：

1. 你模拟的不是代码实际调用的函数引用 #

这是最常见的问题。举个例子：
假设你的业务代码在app.py里，是这么导入邮件发送函数的：

# app.py
from email_utils import send_email

def process_order(user_email):
    send_email(user_email, "订单已确认")

如果你的测试代码里模拟的是email_utils.send_email，但实际上app.py已经在导入时拿到了send_email的引用，此时模拟原模块的函数不会影响app.py里已经存在的引用。

正确做法：模拟被测试代码所在模块中引用的那个函数，也就是app.send_email，而不是原模块的email_utils.send_email。

2. 没有正确使用mock的装饰器/上下文管理器 #

检查你的mock语法是否正确，比如：

• 使用@mock.patch时，装饰器的路径是否正确
• 上下文管理器的范围是否覆盖了调用邮件函数的代码
• 是否忘记在测试函数中接收mock对象参数

给你一个正确的测试示例：

# test_app.py
from unittest import mock
from app import process_order

def test_process_order():
    # 模拟app模块里的send_email引用
    with mock.patch("app.send_email") as mock_send:
        process_order("user@test.com")
        
        # 验证mock被正确调用，确认真实函数没执行
        mock_send.assert_called_once_with("user@test.com", "订单已确认")

3. 检查是否有未被mock覆盖的代码分支 #

比如你的代码里可能在异常处理逻辑中也调用了邮件发送，而你的测试没有覆盖到那个分支，导致真实邮件被触发。这种情况下，要确保所有调用邮件函数的路径都被mock覆盖。

4. 验证mock是否真的生效 #

如果不确定模拟是否生效，可以在测试里打印mock_send.called或者直接用断言验证。如果断言失败，说明你的mock路径肯定错了，得重新检查代码中函数的引用位置。

完整可运行示例 #

假设项目结构：

my_project/
├── email_utils.py  # 真实邮件发送逻辑
├── app.py          # 业务代码
└── test_app.py     # 测试代码

email_utils.py：

def send_email(to, subject):
    # 真实的邮件发送逻辑，比如调用smtplib
    print(f"⚠️ 真实邮件已发送到 {to}")

app.py：

from email_utils import send_email

def handle_signup(email):
    send_email(email, "欢迎注册！")

test_app.py：

import unittest
from unittest import mock
from app import handle_signup

class TestSignup(unittest.TestCase):
    @mock.patch("app.send_email")
    def test_signup_triggers_email(self, mock_send):
        handle_signup("new_user@example.com")
        
        # 验证mock被调用
        mock_send.assert_called_once_with("new_user@example.com", "欢迎注册！")
        # 运行测试时，不会看到"真实邮件已发送"的打印，说明mock生效了

if __name__ == "__main__":
    unittest.main()

核心总结：找到代码中实际调用邮件函数的那个引用位置，然后patch它，而不是原模块的函数。另外一定要用断言验证mock的调用情况，这样能快速确认模拟是否生效。

内容的提问来源于stack exchange，提问作者Bharat Bittu