本文整理匯總了Python中mock.NonCallableMock.prepare_request方法的典型用法代碼示例。如果您正苦於以下問題:Python NonCallableMock.prepare_request方法的具體用法?Python NonCallableMock.prepare_request怎麽用?Python NonCallableMock.prepare_request使用的例子?那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類mock.NonCallableMock
的用法示例。
在下文中一共展示了NonCallableMock.prepare_request方法的1個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: NonCallableMock
# 需要導入模塊: from mock import NonCallableMock [as 別名]
# 或者: from mock.NonCallableMock import prepare_request [as 別名]
not_found_response.status_code = 404
error_response = NonCallableMock(spec=Response)
error_response.headers = CaseInsensitiveDict(data={"content-type": "application/json"})
error_response.content = json.dumps({"error": "internal error"})
error_response.status_code = 500
def prepare_request(request):
return requests.Session().prepare_request(request)
def return_response(prepared_request, proxies, verify, timeout):
if "404" in prepared_request.url:
return not_found_response
elif "500" in prepared_request.url:
return error_response
else:
return ok_response
MockRequests = NonCallableMock(spec=requests)
MockRequestsSend = Mock(side_effect=return_response)
MockRequests.send = MockRequestsSend
MockRequests.prepare_request = Mock(side_effect=prepare_request)
MockRequestsWithAgent = NonCallableMock(spec=requests)
MockRequestsSendWithAgent = Mock(return_value=agent_response)
MockRequestsWithAgent.send = MockRequestsSendWithAgent
MockRequestsWithAgent.prepare_request = Mock(side_effect=prepare_request)