Python 测试的模拟请求和响应

在 Python 测试中，有必要模拟模块及其功能来控制执行流程并验证具体场景。其中，模拟 requests 模块通常用于测试依赖于 HTTP 请求的函数或方法。

考虑一个包含以下代码的views.py 文件：

def myview(request):
    res1 = requests.get('aurl')
    res2 = request.get('burl')
    res3 = request.get('curl')

要模拟此行为，您可以使用 Python 模拟包。以下是逐步实现它的方法：

第 1 步：定义模拟行为

要模拟 requests 模块，请定义一个函数来替代 requests.get ()。在此函数中，您可以为每个 URL 指定所需的响应。

def mocked_requests_get(*args, **kwargs):
    class MockResponse:
        def __init__(self, json_data, status_code):
            self.json_data = json_data
            self.status_code = status_code

        def json(self):
            return self.json_data

    if args[0] == 'aurl':
        return MockResponse({'a': 'a'}, 200)
    elif args[0] == 'burl':
        return MockResponse({'b': 'b'}, 200)
    elif args[0] == 'curl':
        return MockResponse({'c': 'c'}, 200)

    return MockResponse(None, 404)

第2步：修补请求模块

在您的测试用例中，使用mock.patch()装饰器替换实际请求带有您的模拟函数的模块。

@mock.patch('requests.get', side_effect=mocked_requests_get)
def test_myview(self, mock_get):
    # Call the function you want to test
    myview(None)
    # Assertions for expected responses
    ...

第 3 步：验证断言

在测试函数中，您可以使用断言来验证是否返回了预期的响应。

self.assertEqual(mock_get.call_args_list[0][0][0], 'aurl')
self.assertEqual(mock_get.call_args_list[1][0][0], 'burl')
self.assertEqual(mock_get.call_args_list[2][0][0], 'curl')

通过执行以下步骤，您可以有效地模拟 HTTP 请求并控制 Python 测试场景中的响应。这使您可以隔离和测试特定功能，同时确保外部依赖项不会干扰您的结果。