在 pytest 上禁用网络 - Python

在编写 Python 测试时，有时需要在测试运行期间禁用网络访问。这可以确保测试不会意外地与外部资源进行交互，并消除一些不稳定的因素。

在 pytest 上实现禁用网络访问的方法很简单。有两种方法可以做到这一点：

1. 使用 monkeypatch 修补网络连接函数

通过使用 monkeypatch 对 Python 的网络连接函数进行修补，我们可以将其替换为简单的占位符函数，以禁用实际网络访问。以下是一个使用 monkeypatch 来模拟禁用网络的例子：

import socket


def network_disabled(*args, **kwargs):
    raise RuntimeError("Network access disabled for testing")


def test_network_disabled(monkeypatch):
    monkeypatch.setattr(socket, "create_connection", network_disabled)
    
    try:
        # 这里是你的测试代码
        pass
    except RuntimeError as e:
        assert str(e) == "Network access disabled for testing"

在上面的例子中，我们使用了 monkeypatch 将 socket.create_connection 函数替换为了 network_disabled 函数。这个函数在被调用时，会抛出 RuntimeError 异常，表示网络访问已被禁用。

测试中使用的 try-except 块是为了确保测试代码不会因为预期得到的 RuntimeError 异常而失败。

2. 使用 pytest_mock 库

pytest_mock 库是一个专门用于在 pytest 测试中进行模拟和打桩的库。这个库包含了一些方便的函数和装饰器，可以帮助我们轻松地模拟各种函数和对象。

以下是一个使用 pytest_mock 库以禁用网络访问的例子：

import requests


def test_network_disabled(mocker):
    mock_create_connection = mocker.patch("socket.create_connection")
    mock_create_connection.side_effect = RuntimeError("Network access disabled for testing")
    
    # 这里是你的测试代码
    pass

在上面的例子中，我们使用了 pytest_mock 库来顶替 socket.create_connection 函数。使用 mocker.patch 函数替换函数，并使用 side_effect 参数来指定该函数在被调用时应该抛出的异常。在这个例子中，我们使用了 RuntimeError 异常来模拟网络访问被禁止。