PythonMock测试教程_依赖隔离与接口模拟

Mock是测试中替代真实对象以隔离外部依赖的技术,用于避免数据库、HTTP请求等慢且不稳定的副作用,专注验证逻辑正确性;Python用unittest.mock的Mock和patch实现模拟、断言与自动还原。

什么是Mock,为什么需要它

Mock是一种在测试中替代真实对象的技术,用来隔离被测代码的外部依赖。比如你的函数调用数据库、发HTTP请求、读写文件,这些操作慢、不稳定、有副作用,不适合在单元测试里真实执行。Mock能让你“假装”这些依赖存在并按预期返回数据,从而专注验证逻辑本身是否正确。

用unittest.mock快速模拟对象和方法

Python内置unittest.mock模块,最常用的是Mockpatch

  • 直接替换属性或返回值:给对象的某个方法设return_value,调用时就返回你指定的内容,不执行原逻辑。
  • patch装饰器或上下文管理器:在测试前后自动替换目标(如requests.get),结束后自动还原,避免污染其他测试。
  • 检查是否被调用:通过mock_obj.assert_called()assert_called_with(...)确认方法是否按预期被调用。

模拟第三方接口:以requests为例

假设有个函数fetch_user(user_id)内部调用requests.get获取用户信息。测试时不想真发请求,就可以这样写:

from unittest.mock import patch
import requests

def test_fetch_user():
    mock_response = Mock()
    mock_response.json.return_value = {"id": 123, "name": "Alice"}
    mock_response.status_code = 200

    with patch("requests.get", return_value=mock_response):
        result = fetch_user(123)
        assert result == {"id": 123, "name": "Alice"}

注意:patch的目标路径必须是“被测试代码中导入的位置”,不是定义位置。比如你在myapp.api里写了import requests,就要patch"myapp.api.requests.get"或直接"requests.get"(取决于导入方式)。

立即学习“Python免费学习笔记(深入)”;

处理复杂依赖:嵌套对象与链式调用

有些对象需要多层访问,比如config.db.urlresponse.data.items()[0].name。这时可用PropertyMockMockconfigure_mock方法预设结构:

  • mock_obj = Mock()后调configure_mock(db=Mock(url="sqlite:///test.db"))
  • 对链式返回,可设mock_obj.data.items.return_value = [{"name": "foo"}]
  • 需要抛异常?设side_effect = ValueError("timeout"),调用时就会触发。