requests-mock 是一个 Python 库，用于模拟 requests 库的行为。这在测试中使用得非常广泛，尤其是当你想要控制外部 API 或服务的响应时。通过模拟 requests 的行为，你可以确保测试的可重复性和独立性，而不必依赖于外部服务或网络条件。
以下是如何使用 requests-mock 来构建 Python 测试架构的一个基本示例：
- 安装 requests-mock:
pip install requests-mock
- 使用 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 或其响应。这使得测试更加快速、可靠和可重复。
