mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-09 01:03:24 +00:00
7caf29735a
Format with line length 110 to match flake8 make black part of linting check Update travis spec for updated black requirements Add diff output for black on failure update changelog
91 lines
3.4 KiB
Python
91 lines
3.4 KiB
Python
"""Stubs for tornado HTTP clients"""
|
|
from __future__ import absolute_import
|
|
|
|
import functools
|
|
from six import BytesIO
|
|
|
|
from tornado import httputil
|
|
from tornado.httpclient import HTTPResponse
|
|
|
|
from vcr.errors import CannotOverwriteExistingCassetteException
|
|
from vcr.request import Request
|
|
|
|
|
|
def vcr_fetch_impl(cassette, real_fetch_impl):
|
|
@functools.wraps(real_fetch_impl)
|
|
def new_fetch_impl(self, request, callback):
|
|
headers = request.headers.copy()
|
|
if request.user_agent:
|
|
headers.setdefault("User-Agent", request.user_agent)
|
|
|
|
# TODO body_producer, header_callback, and streaming_callback are not
|
|
# yet supported.
|
|
|
|
unsupported_call = (
|
|
getattr(request, "body_producer", None) is not None
|
|
or request.header_callback is not None
|
|
or request.streaming_callback is not None
|
|
)
|
|
if unsupported_call:
|
|
response = HTTPResponse(
|
|
request,
|
|
599,
|
|
error=Exception(
|
|
"The request (%s) uses AsyncHTTPClient functionality "
|
|
"that is not yet supported by VCR.py. Please make the "
|
|
"request outside a VCR.py context." % repr(request)
|
|
),
|
|
request_time=self.io_loop.time() - request.start_time,
|
|
)
|
|
return callback(response)
|
|
|
|
vcr_request = Request(request.method, request.url, request.body, headers)
|
|
|
|
if cassette.can_play_response_for(vcr_request):
|
|
vcr_response = cassette.play_response(vcr_request)
|
|
headers = httputil.HTTPHeaders()
|
|
|
|
recorded_headers = vcr_response["headers"]
|
|
if isinstance(recorded_headers, dict):
|
|
recorded_headers = recorded_headers.items()
|
|
for k, vs in recorded_headers:
|
|
for v in vs:
|
|
headers.add(k, v)
|
|
response = HTTPResponse(
|
|
request,
|
|
code=vcr_response["status"]["code"],
|
|
reason=vcr_response["status"]["message"],
|
|
headers=headers,
|
|
buffer=BytesIO(vcr_response["body"]["string"]),
|
|
effective_url=vcr_response.get("url"),
|
|
request_time=self.io_loop.time() - request.start_time,
|
|
)
|
|
return callback(response)
|
|
else:
|
|
if cassette.write_protected and cassette.filter_request(vcr_request):
|
|
response = HTTPResponse(
|
|
request,
|
|
599,
|
|
error=CannotOverwriteExistingCassetteException(
|
|
cassette=cassette, failed_request=vcr_request
|
|
),
|
|
request_time=self.io_loop.time() - request.start_time,
|
|
)
|
|
return callback(response)
|
|
|
|
def new_callback(response):
|
|
headers = [(k, response.headers.get_list(k)) for k in response.headers.keys()]
|
|
|
|
vcr_response = {
|
|
"status": {"code": response.code, "message": response.reason},
|
|
"headers": headers,
|
|
"body": {"string": response.body},
|
|
"url": response.effective_url,
|
|
}
|
|
cassette.append(vcr_request, vcr_response)
|
|
return callback(response)
|
|
|
|
real_fetch_impl(self, request, new_callback)
|
|
|
|
return new_fetch_impl
|