Mirroring/vcrpy
1
0
Fork 0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-09 01:03:24 +00:00
Code Releases Activity

Merge pull request #187 from abhinav/master

Browse Source 
Set request_time on Tornadoo HTTPResponses
This commit is contained in:
Ivan 'Goat' Malison
2015-08-18 16:10:42 -07:00
parent bb8f563135 1e1c093b3c
commit 1e414826e7
2 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/integration/test_tornado.py
View File

@@ -337,3 +337,19 @@ def test_existing_instances_get_patched(get_client, tmpdir):
with vcr.use_cassette(str(tmpdir.join('data.yaml'))) as cass:
yield get(client, 'http://httpbin.org/get')
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

3
vcr/stubs/tornado_stubs.py
View File

@@ -36,6 +36,7 @@ def vcr_fetch_impl(cassette, real_fetch_impl):
"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)
@@ -63,6 +64,7 @@ def vcr_fetch_impl(cassette, real_fetch_impl):
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:
@@ -78,6 +80,7 @@ def vcr_fetch_impl(cassette, real_fetch_impl):
"your current record mode (%r)."
% (vcr_request, cassette._path, cassette.record_mode)
),
request_time=self.io_loop.time() - request.start_time,
)
return callback(response)