如何模拟请求和响应?

我试图使用python模拟包来模拟python请求模块。让我在下面的场景中工作的基本调用是什么?

在views.py中，我有一个函数，它每次都以不同的响应进行各种request .get()调用

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

在我的测试类中，我想做类似的事情，但不能确定确切的方法调用

步骤1:

# Mock the requests module
# when mockedRequests.get('aurl') is called then return 'a response'
# when mockedRequests.get('burl') is called then return 'b response'
# when mockedRequests.get('curl') is called then return 'c response'

步骤2:

调用我的视图

步骤3:

验证响应包含'a response'， 'b response'， 'c response'

我如何完成第1步(模拟请求模块)?

当前回答

如果你想模拟一个假响应，另一种方法是简单地实例化一个基本HttpResponse类的实例，如下所示:

from django.http.response import HttpResponseBase

self.fake_response = HttpResponseBase()

2016-03-31 18:37:51

其他回答

这就是模拟请求的方法。Post，将其更改为HTTP方法

@patch.object(requests, 'post')
def your_test_method(self, mockpost):
    mockresponse = Mock()
    mockpost.return_value = mockresponse
    mockresponse.text = 'mock return'

    #call your target method now

2017-02-17 23:02:59

下面是一个带有请求响应类的解决方案。恕我直言，它更干净。

import json
from unittest.mock import patch
from requests.models import Response

def mocked_requests_get(*args, **kwargs):
    response_content = None
    request_url = kwargs.get('url', None)
    if request_url == 'aurl':
        response_content = json.dumps('a response')
    elif request_url == 'burl':
        response_content = json.dumps('b response')
    elif request_url == 'curl':
        response_content = json.dumps('c response')
    response = Response()
    response.status_code = 200
    response._content = str.encode(response_content)
    return response

@mock.patch('requests.get', side_effect=mocked_requests_get)
def test_fetch(self, mock_get):
     response = requests.get(url='aurl')
     assert ...

2020-12-24 11:38:26

只是一个有用的提示给那些仍然挣扎，从urllib或urllib2/urllib3转换到请求，并试图模拟一个响应-我在实现我的模拟时得到了一个稍微令人困惑的错误:

与请求。get(path, auth=HTTPBasicAuth('user'， 'pass')， verify=False) as url:

AttributeError: __enter__

当然，如果我知道with如何工作(我不知道)，我就会知道这是一个残留的、不必要的上下文(来自PEP 343)。在使用请求库时没有必要，因为它在底层为您做了基本相同的事情。只需删除with并使用bare requests.get(…)，就万事大吉了。

2018-12-27 22:31:36

如果你想模拟一个假响应，另一种方法是简单地实例化一个基本HttpResponse类的实例，如下所示:

from django.http.response import HttpResponseBase

self.fake_response = HttpResponseBase()

2016-03-31 18:37:51

使用requests_mock可以很容易地修补任何请求

pip install requests-mock

from unittest import TestCase
import requests_mock
from <yourmodule> import <method> (auth)

class TestApi(TestCase):
  @requests_mock.Mocker()
  def test_01_authentication(self, m):
        """Successful authentication using username password"""
        token = 'token'
        m.post(f'http://localhost/auth', json= {'token': token})
        act_token =auth("user", "pass")
        self.assertEqual(act_token, token)

2022-06-22 11:37:59

如何模拟请求和响应?

推荐文章

最新文章

标签