Pytest 中使用 Mock 测试含系统调用的网络接口创建函数-Python教程-PHP中文网

Pytest 中使用 Mock 测试含系统调用的网络接口创建函数

本文详解如何在 Pytest 环境下正确对依赖 subprocess.Popen、外部校验函数（如 validate_interface 和 ping_test）的 create_interface 方法进行单元测试，重点解决 Mock 配置遗漏导致的 UnexpectedMethodCallError 问题，并提供现代 pytest-mock 替代 mox 的推荐实践。

本文详解如何在 pytest 环境下正确对依赖 `subprocess.popen`、外部校验函数（如 `validate_interface` 和 `ping_test`）的 `create_interface` 方法进行单元测试，重点解决 mock 配置遗漏导致的 `unexpectedmethodcallerror` 问题，并提供现代 pytest-mock 替代 mox 的推荐实践。

create_interface 是一个典型的系统级操作函数：它先校验参数，执行 ping 检测 IP 冲突，再通过 sudo ifconfig 创建虚拟网卡。直接运行该函数会修改真实网络环境，因此必须全面 mock 外部依赖——但原测试代码存在关键缺陷：它试图调用 vi.create_interface(...)，却未提前声明该方法为可 mock 对象（即未 StubOutWithMock(vi, "create_interface")），导致 mox 将其视为“未预期的方法调用”，抛出 UnexpectedMethodCallError。

更根本的问题在于：mox 已是过时框架（2014 年后无维护），且与 pytest 原生生态不兼容。现代 Python 测试应优先使用 pytest-mock（基于 unittest.mock），它语法简洁、调试友好、社区活跃。以下是推荐的 pytest + mocker 实现：

import pytest
from unittest.mock import patch, MagicMock

# 假设 vi 是包含 create_interface 的模块（如 your_module.py）
import your_module as vi

def test_create_interface_success(mocker):
    # Mock 依赖函数
    mock_validate = mocker.patch("your_module.validate_interface")
    mock_ping = mocker.patch("your_module.ping_test")
    mock_popen = mocker.patch("your_module.subprocess.Popen")

    # 构造模拟的 Popen 实例
    mock_proc = MagicMock()
    mock_proc.returncode = 0
    mock_proc.communicate.return_value = (b"", b"")
    mock_popen.return_value = mock_proc

    # 执行测试
    iface, address, netmask = "eth0:1", "1.2.3.4", "255.255.255.0"
    vi.create_interface(iface, address, netmask)

    # 断言调用行为
    mock_validate.assert_called_once_with(iface, address, netmask)
    mock_ping.assert_called_once_with(address)
    mock_popen.assert_called_once_with(
        f"sudo ifconfig {iface} {address} netmask {netmask} up",
        shell=True,
        stdout=vi.subprocess.PIPE,
        stderr=vi.subprocess.PIPE
    )
    assert mock_proc.communicate.called

⚠️ 关键注意事项：

不要 mock create_interface 自身：它是被测对象（SUT），应保持真实逻辑；只 mock 它的下游依赖（validate_interface, ping_test, subprocess.Popen）。

路径需精准：mocker.patch("your_module.validate_interface") 中的路径是 create_interface 内部导入/引用该函数的位置，而非定义位置（常见错误！）。

异常路径也需覆盖：例如 ping_test 返回 True 时应触发 Exception，可用 pytest.raises(Exception) 验证。

避免 shell=True 的安全风险：生产代码中建议改用 Popen(["ifconfig", ...]) 并显式传参，便于测试和安全加固。