Mirroring/vcrpy
1
0
Fork 0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-08 16:53:23 +00:00
Code Releases Activity

allow injection of persistence methods

Browse Source
This commit is contained in:
Julien Funk
2017-01-12 16:41:26 -05:00
parent 53c55b13e7
commit 6f8486e0a2
4 changed files with 83 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
vcr/cassette.py
View File

@@ -163,13 +163,16 @@ class Cassette(object):
def use(cls, **kwargs):
return CassetteContextDecorator.from_args(cls, **kwargs)
def __init__(self, path, serializer=yamlserializer, record_mode='once',
def __init__(self, path, serializer=yamlserializer, save_callback=None,
load_callback=None, record_mode='once',
match_on=(uri, method), before_record_request=None,
before_record_response=None, custom_patches=(),
inject=False):
self._path = path
self._serializer = serializer
self._save_callback = save_callback
self._load_callback = load_callback
self._match_on = match_on
self._before_record_request = before_record_request or (lambda x: x)
self._before_record_response = before_record_response or (lambda x: x)
@@ -274,7 +277,8 @@ class Cassette(object):
save_cassette(
self._path,
self._as_dict(),
serializer=self._serializer
serializer=self._serializer,
save_callback=self._save_callback
)
self.dirty = False
@@ -282,7 +286,8 @@ class Cassette(object):
try:
requests, responses = load_cassette(
self._path,
serializer=self._serializer
serializer=self._serializer,
load_callback=self._load_callback
)
for request, response in zip(requests, responses):
self.append(request, response)