1
0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-09 01:03:24 +00:00

Add support for configurable record modes

This support will let you select one of four different behaviors
for VCR's cassettes.  Closes #23
This commit is contained in:
Kevin McCarthy
2013-09-15 16:44:50 -10:00
parent 5ce67dc023
commit 03c22d79dd
7 changed files with 169 additions and 6 deletions

View File

@@ -4,13 +4,17 @@ from .serializers import yamlserializer, jsonserializer
class VCR(object):
def __init__(self, serializer='yaml', cassette_library_dir=None):
def __init__(self,
serializer='yaml',
cassette_library_dir=None,
record_mode="once"):
self.serializer = serializer
self.cassette_library_dir = cassette_library_dir
self.serializers = {
'yaml': yamlserializer,
'json': jsonserializer,
}
self.record_mode = record_mode
def _get_serializer(self, serializer_name):
try:
@@ -34,6 +38,7 @@ class VCR(object):
merged_config = {
"serializer": self._get_serializer(serializer_name),
"record_mode": kwargs.get('record_mode', self.record_mode),
}
return Cassette.load(path, **merged_config)