mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-10 09:35:34 +00:00
Set request_time on Tornadoo HTTPResponses
This commit is contained in:
@@ -337,3 +337,19 @@ def test_existing_instances_get_patched(get_client, tmpdir):
|
|||||||
with vcr.use_cassette(str(tmpdir.join('data.yaml'))) as cass:
|
with vcr.use_cassette(str(tmpdir.join('data.yaml'))) as cass:
|
||||||
yield get(client, 'http://httpbin.org/get')
|
yield get(client, 'http://httpbin.org/get')
|
||||||
assert cass.play_count == 1
|
assert cass.play_count == 1
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.gen_test
|
||||||
|
def test_request_time_is_set(get_client, tmpdir):
|
||||||
|
'''Ensures that the request_time on HTTPResponses is set.'''
|
||||||
|
|
||||||
|
with vcr.use_cassette(str(tmpdir.join('data.yaml'))):
|
||||||
|
client = get_client()
|
||||||
|
response = yield get(client, 'http://httpbin.org/get')
|
||||||
|
assert response.request_time is not None
|
||||||
|
|
||||||
|
with vcr.use_cassette(str(tmpdir.join('data.yaml'))) as cass:
|
||||||
|
client = get_client()
|
||||||
|
response = yield get(client, 'http://httpbin.org/get')
|
||||||
|
assert response.request_time is not None
|
||||||
|
assert cass.play_count == 1
|
||||||
|
|||||||
@@ -36,6 +36,7 @@ def vcr_fetch_impl(cassette, real_fetch_impl):
|
|||||||
"that is not yet supported by VCR.py. Please make the "
|
"that is not yet supported by VCR.py. Please make the "
|
||||||
"request outside a VCR.py context." % repr(request)
|
"request outside a VCR.py context." % repr(request)
|
||||||
),
|
),
|
||||||
|
request_time=self.io_loop.time() - request.start_time,
|
||||||
)
|
)
|
||||||
return callback(response)
|
return callback(response)
|
||||||
|
|
||||||
@@ -63,6 +64,7 @@ def vcr_fetch_impl(cassette, real_fetch_impl):
|
|||||||
headers=headers,
|
headers=headers,
|
||||||
buffer=BytesIO(vcr_response['body']['string']),
|
buffer=BytesIO(vcr_response['body']['string']),
|
||||||
effective_url=vcr_response.get('url'),
|
effective_url=vcr_response.get('url'),
|
||||||
|
request_time=self.io_loop.time() - request.start_time,
|
||||||
)
|
)
|
||||||
return callback(response)
|
return callback(response)
|
||||||
else:
|
else:
|
||||||
@@ -78,6 +80,7 @@ def vcr_fetch_impl(cassette, real_fetch_impl):
|
|||||||
"your current record mode (%r)."
|
"your current record mode (%r)."
|
||||||
% (vcr_request, cassette._path, cassette.record_mode)
|
% (vcr_request, cassette._path, cassette.record_mode)
|
||||||
),
|
),
|
||||||
|
request_time=self.io_loop.time() - request.start_time,
|
||||||
)
|
)
|
||||||
return callback(response)
|
return callback(response)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user