Python补丁实战指南（手把手教你使用patch进行单元测试）

安装与导入

从 Python 3.3 开始，unittest.mock 已经是标准库的一部分，无需额外安装。如果你使用的是更早版本，可以通过 pip 安装：

pip install mock

在代码中导入方式如下（适用于 Python 3.3+）：

from unittest.mock import patch, Mock

使用场景示例

假设我们有一个函数 get_user_name(user_id)，它会调用外部 API 获取用户名：

import requestsdef get_user_name(user_id):    response = requests.get(f"https://api.example.com/users/{user_id}")    if response.status_code == 200:        return response.json()['name']    else:        return None

如果我们直接测试这个函数，每次都会发起真实网络请求，这显然不理想。这时就可以用 patch 来模拟 requests.get。

方法一：作为装饰器使用

import unittestfrom unittest.mock import patch# 假设上面的 get_user_name 函数在 mymodule.py 中from mymodule import get_user_nameclass TestGetUserName(unittest.TestCase):    @patch('mymodule.requests.get')    def test_get_user_name_success(self, mock_get):        # 配置 mock 对象的行为        mock_response = Mock()        mock_response.status_code = 200        mock_response.json.return_value = {'name': 'Alice'}        mock_get.return_value = mock_response        # 调用被测函数        result = get_user_name(123)        # 断言        self.assertEqual(result, 'Alice')        mock_get.assert_called_once_with('https://api.example.com/users/123')

注意：@patch('mymodule.requests.get') 中的路径必须是 被测试模块中导入的位置，而不是 requests 模块本身的位置。这是初学者常犯的错误！

方法二：作为上下文管理器使用

def test_get_user_name_with_context():    with patch('mymodule.requests.get') as mock_get:        mock_response = Mock()        mock_response.status_code = 200        mock_response.json.return_value = {'name': 'Bob'}        mock_get.return_value = mock_response        result = get_user_name(456)        assert result == 'Bob'

上下文管理器的方式适合在单个测试函数中临时打补丁，不影响其他测试。

常见技巧与注意事项

• 路径要写对：patch 的目标是“被使用的地方”，不是“定义的地方”。例如，如果 mymodule.py 中写了 from requests import get，那么你应该 patch 'mymodule.get'。
• 多个 patch 可以叠加：可以同时 mock 多个依赖，装饰器从下往上应用。
• 使用 side_effect：可以模拟异常或动态返回值。

@patch('mymodule.requests.get')def test_api_error(self, mock_get):    mock_get.side_effect = ConnectionError("Network down")    result = get_user_name(789)    self.assertIsNone(result)

总结

通过本教程，你已经掌握了 Python补丁 的基本用法。无论是使用装饰器还是上下文管理器，patch 都能帮助你在单元测试中隔离外部依赖，写出更快、更稳定的测试用例。

记住关键词：Python mock patch、单元测试patch、patch使用教程，它们是你深入学习和搜索相关资料的好帮手。

现在，快去给你的项目加上可靠的单元测试吧！