1
0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-10 09:35:34 +00:00

change response to response_of

This commit is contained in:
Kevin McCarthy
2013-08-10 13:00:23 -10:00
parent 4f0668471e
commit a50cebaf21
3 changed files with 6 additions and 6 deletions

View File

@@ -44,12 +44,12 @@ def test_cassette_contains():
a.append('foo','bar') a.append('foo','bar')
assert 'foo' in a assert 'foo' in a
def test_cassette_response(): def test_cassette_response_of():
a = Cassette('test') a = Cassette('test')
a.append('foo','bar') a.append('foo','bar')
assert a.response('foo') == 'bar' assert a.response_of('foo') == 'bar'
def test_cassette_missing_response(): def test_cassette_get_missing_response():
a = Cassette('test') a = Cassette('test')
with pytest.raises(KeyError): with pytest.raises(KeyError):
a.response('foo') a.response_of('foo')

View File

@@ -52,7 +52,7 @@ class Cassette(object):
'''Add a pair of request, response to this cassette''' '''Add a pair of request, response to this cassette'''
self.data[request] = response self.data[request] = response
def response(self, request): def response_of(self, request):
'''Find the response corresponding to a request''' '''Find the response corresponding to a request'''
return self.data[request] return self.data[request]

View File

@@ -69,7 +69,7 @@ class VCRConnectionMixin:
# 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._request in self.cassette:
response = self.cassette.response(self._request) response = self.cassette.response_of(self._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._request)