Python测试架构requests-mock

requests-mock 是一个 Python 库，用于模拟 requests 库的行为。这在测试中使用得非常广泛，尤其是当你想要控制外部 API 或服务的响应时。通过模拟 requests 的行为，你可以确保测试的可重复性和独立性，而不必依赖于外部服务或网络条件。

以下是如何使用 requests-mock 来构建 Python 测试架构的一个基本示例：

1. 安装 requests-mock:

pip install requests-mock

1. 使用 requests-mock 进行测试:

假设你有一个函数，它使用 requests 库来调用一个外部 API，并返回结果：

import requests

def get_data_from_api(url):
    response = requests.get(url)
    return response.json()

为了测试这个函数，你可以使用 requests-mock 来模拟 API 的响应：

from requests_mock import Mocker
import unittest

class TestApi(unittest.TestCase):

    def setUp(self):
        self.mock_requests = Mocker()

    def tearDown(self):
        self.mock_requests.stop()

    def test_get_data_from_api(self):
        # 设置模拟的响应
        self.mock_requests.get('http://example.com/api', json={
   
   'data': 'mocked data'})

        # 调用函数并检查结果
        result = get_data_from_api('http://example.com/api')
        self.assertEqual(result, {
   
   'data': 'mocked data'})

if __name__ == '__main__':
    unittest.main()

在上面的示例中：

• setUp 和 tearDown 方法分别用于在开始和结束每个测试时启动和停止模拟。
• 在 test_get_data_from_api 测试中，我们使用 self.mock_requests.get 方法来模拟对 http://example.com/api 的 GET 请求，并设置其响应为一个包含 'mocked data' 的 JSON 对象。
• 然后，我们调用 get_data_from_api 函数并检查结果是否与模拟的响应匹配。

通过这种方法，你可以确保 get_data_from_api 函数的测试不依赖于实际的外部 API 或其响应。这使得测试更加快速、可靠和可重复。