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

7
vcr/config.py
View File

@@ -36,7 +36,8 @@ class VCR(object):
match_on=('method', 'scheme', 'host', 'port', 'path', 'query'),
before_record=None, inject_cassette=False, serializer='yaml',
cassette_library_dir=None, func_path_generator=None,
decode_compressed_response=False):
decode_compressed_response=False, save_callback=None,
load_callback=None):
self.serializer = serializer
self.match_on = match_on
self.cassette_library_dir = cassette_library_dir
@@ -69,6 +70,8 @@ class VCR(object):
self.path_transformer = path_transformer
self.func_path_generator = func_path_generator
self.decode_compressed_response = decode_compressed_response
self.save_callback = save_callback
self.load_callback = load_callback
self._custom_patches = tuple(custom_patches)
def _get_serializer(self, serializer_name):
@@ -147,6 +150,8 @@ class VCR(object):
tuple(matcher_names) + tuple(additional_matchers)
),
'record_mode': kwargs.get('record_mode', self.record_mode),
'save_callback': kwargs.get('save_callback', self.save_callback),
'load_callback': kwargs.get('load_callback', self.load_callback),
'before_record_request': self._build_before_record_request(kwargs),
'before_record_response': self._build_before_record_response(kwargs),
'custom_patches': self._custom_patches + kwargs.get(