模拟请求来控制响应
在Python中，mock包提供了一种强大的方法来模拟外部模块或类，允许您操作行为并验证交互。在 HTTP 请求的上下文中，模拟请求模块对于测试依赖于外部服务的代码特别有用。
第 1 步：模拟请求模块
至模拟 Requests 模块时，您需要使用返回所需响应的自定义函数来修补 get() 函数。您可以像这样定义一个模拟方法：
<code class="python">def mocked_requests_get(url, **kwargs):
if url == "aurl":
return MockResponse("a response")
elif url == "burl":
return MockResponse("b response")
else:
raise Exception("URL not mocked")</code>请注意，此方法需要一个有效的 URL 并返回一个 MockResponse 对象，该对象表示具有预定义内容的假装响应。
步骤 2：修补原始请求模块
定义了模拟方法后，您可以使用 @mock.patch 装饰器来修补原始 requests.get() 。这将替换您正在使用模拟行为测试的代码中对 requests.get() 的所有调用。
<code class="python">@mock.patch("requests.get", side_effect=mocked_requests_get)
def test_myview(self, mock_get):
# Your test goes here</code>第 3 步：调用查看并验证响应
现在您可以照常调用您的函数并验证是否获得了预期的响应。可以检查模拟对象以断言 get() 函数是使用特定参数调用的并返回所需的值。
示例代码：
<code class="python">import requests
from unittest import mock
class MyViewTest(unittest.TestCase):
# ...
def test_myview(self, mock_get):
self.assertEqual(res1.text, "a response")
self.assertEqual(res2.text, "b response")
self.assertEqual(res3.text, "c response")
# Verify mock calls
mock_get.assert_called_with('aurl')
mock_get.assert_called_with('burl')
mock_get.assert_called_with('curl')</code>记住验证响应的文本内容以及传递给模拟方法的调用计数和参数。这使您可以确保发生预期的交互并实现预期的行为。
以上是如何在 Python 中模拟请求模块响应的详细内容。更多信息请关注PHP中文网其他相关文章！
