模拟 API 交互的模拟 Python 请求模块

在我们全面测试与 API 交互的 Python 代码的过程中，有效地模拟 requests 模块至关重要。以下是使用自定义响应模拟 requests.get() 调用的分步方法：

第 1 步：模拟 Requests 模块

利用 Python 的模拟包，我们定义自定义函数（mocked_requests_get）来覆盖 requests.get()。此函数根据提供的 URL 返回模拟响应。在我们的示例中，不同的 URL 将收到特定的响应：

def mocked_requests_get(*args, **kwargs):
    if args[0] == 'aurl':
        return 'a response'
    elif args[0] == 'burl':
        return 'b response'
    elif args[0] == 'curl':
        return 'c response'

第2步：在测试类中进行模拟

在我们的测试类中，我们使用unittest.mock.patch将模拟应用到requests模块():

@mock.patch('requests.get', side_effect=mocked_requests_get)

这个装饰器包装了我们的测试方法，确保 requests.get 被我们的自定义函数模拟。

第 3 步：调用视图并验证响应

我们照常调用视图函数并验证预期响应。我们的模拟函数保证响应与我们之前定义的自定义响应匹配：

res1 = requests.get('aurl')
assert res1 == 'a response'
res2 = request.get('burl')
assert res2 == 'b response'
res3 = request.get('curl')
assert res3 == 'c response'

按照以下步骤，您可以在Python测试中有效地模拟requests模块，从而模拟各种API响应并彻底测试代码在不同场景下的行为。