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

replace cached() with mark_played / play_count

This commit is contained in:
Kevin McCarthy
2013-08-05 22:25:45 -10:00
parent 9b665fef27
commit fb2aa89445
6 changed files with 42 additions and 49 deletions

View File

@@ -23,9 +23,9 @@ class Cassette(object):
def __init__(self, path, data=None):
self._path = path
self._cached = []
self._requests = []
self._responses = []
self.play_count = 0
if data:
self.deserialize(data)
@@ -45,18 +45,11 @@ class Cassette(object):
self._requests, self._responses = (
[r['request'] for r in source], [r['response'] for r in source])
def cached(self, request=None):
def mark_played(self, request=None):
'''
Alert the cassette of a request that's been cached, or get the
requests that we've cached. This is used mainly for
debugging purposes.
Alert the cassette of a request that's been played
'''
# TODO: maybe dependency injection for this method since
# it's only used in tests?
if request:
self._cached.append(request)
else:
return self._cached
self.play_count += 1
def append(self, request, response):
'''Add a pair of request, response pairs to this cassette'''

View File

@@ -67,9 +67,9 @@ class VCRConnectionMixin:
# then return it
response = self.cassette.response(self._request)
if response:
# Alert the cassette to the fact that we've served another cached
# Alert the cassette to the fact that we've served another
# response for the provided requests
self.cassette.cached(self._request)
self.cassette.mark_played()
return VCRHTTPResponse(response)
# Otherwise, we made an actual request, and should return the response