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

persist on _vcr_request

Browse Source
This commit is contained in:
Kevin McCarthy
2013-08-11 18:32:02 -10:00
parent d00c27e244
commit 5c0a1e1769
1 changed files with 7 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
vcr/stubs/__init__.py
View File

@@ -29,8 +29,6 @@ class VCRHTTPResponse(object):
# I have to add it to both. # I have to add it to both.
self.msg.headers.append("{0}:{1}".format(key, val)) self.msg.headers.append("{0}:{1}".format(key, val))
#self.msg.headers = ["{0}:{1}".format(k,v) for k, v in self.recorded_response['headers'].iteritems()]
self.length = self.msg.getheader('content-length') or None self.length = self.msg.getheader('content-length') or None
def read(self, *args, **kwargs): def read(self, *args, **kwargs):
@@ -55,8 +53,8 @@ class VCRConnectionMixin:
cassette = None cassette = None
def request(self, method, url, body=None, headers=None): def request(self, method, url, body=None, headers=None):
'''Persist the request metadata in self._vcr''' '''Persist the request metadata in self._vcr_request'''
self._request = Request( self._vcr_request = Request(
protocol = self._protocol, protocol = self._protocol,
host = self.host, host = self.host,
port = self.port, port = self.port,
@@ -68,7 +66,7 @@ class VCRConnectionMixin:
# Check if we have a cassette set, and if we have a response saved. # Check if we have a cassette set, and if we have a response saved.
# If so, there's no need to keep processing and we can bail # If so, there's no need to keep processing and we can bail
if self.cassette and self._request in self.cassette: if self.cassette and self._vcr_request in self.cassette:
return return
# Otherwise, we should submit the request # Otherwise, we should submit the request
@@ -79,11 +77,11 @@ class VCRConnectionMixin:
'''Retrieve a the response''' '''Retrieve a the response'''
# Check to see if the cassette has a response for this request. If so, # Check to see if the cassette has a response for this request. If so,
# then return it # then return it
if self._request in self.cassette: if self._vcr_request in self.cassette:
response = self.cassette.response_of(self._request) response = self.cassette.response_of(self._vcr_request)
# Alert the cassette to the fact that we've served another # Alert the cassette to the fact that we've served another
# response for the provided requests # response for the provided requests
self.cassette.mark_played(self._request) self.cassette.mark_played(self._vcr_request)
return VCRHTTPResponse(response) return VCRHTTPResponse(response)
else: else:
# Otherwise, we made an actual request, and should return the response # Otherwise, we made an actual request, and should return the response
@@ -94,7 +92,7 @@ class VCRConnectionMixin:
'headers': dict(response.getheaders()), 'headers': dict(response.getheaders()),
'body': {'string': response.read()}, 'body': {'string': response.read()},
} }
self.cassette.append(self._request, response) self.cassette.append(self._vcr_request, response)
return VCRHTTPResponse(response) return VCRHTTPResponse(response)