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

Revert "Fixed issue in test_nested_context_managers_with_session_created_before_first_nesting. by using a single class and patching cassette on that class. Not a great solution :\"

Browse Source 
This reverts commit 2bf23b2cdf.
This commit is contained in:
Ivan Malison
2014-09-19 14:32:21 -07:00
parent b6e96020c1
commit 1018867838
1 changed files with 23 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
vcr/patch.py
View File

@@ -57,22 +57,41 @@ class PatcherBuilder(object):
def __init__(self, cassette): def __init__(self, cassette):
self._cassette = cassette self._cassette = cassette
self._class_to_cassette_subclass = {}
def build_patchers(self): def build_patchers(self):
patcher_args = itertools.chain(self._httplib(), self._requests(), self._urllib3(), patcher_args = itertools.chain(self._httplib(), self._requests(), self._urllib3(),
self._httplib2(), self._boto()) self._httplib2(), self._boto())
for obj, patched_attribute, replacement_class in patcher_args: for args in patcher_args:
patcher = self._build_patcher(obj, patched_attribute, replacement_class) patcher = self._build_patcher(*args)
if patcher: if patcher:
yield patcher yield patcher
if hasattr(replacement_class, 'cassette'):
yield mock.patch.object(replacement_class, 'cassette', self._cassette)
def _build_patcher(self, obj, patched_attribute, replacement_class): def _build_patcher(self, obj, patched_attribute, replacement_class):
if not hasattr(obj, patched_attribute): if not hasattr(obj, patched_attribute):
return return
if isinstance(replacement_class, dict):
for key in replacement_class:
replacement_class[key] = self._get_cassette_subclass(replacement_class[key])
else:
replacement_class = self._get_cassette_subclass(replacement_class)
return mock.patch.object(obj, patched_attribute, replacement_class) return mock.patch.object(obj, patched_attribute, replacement_class)
def _get_cassette_subclass(self, klass):
if klass.cassette is not None:
return klass
if klass not in self._class_to_cassette_subclass:
self._class_to_cassette_subclass[klass] = self._build_cassette_subclass(klass)
return self._class_to_cassette_subclass[klass]
def _build_cassette_subclass(self, base_class):
bases = (base_class,)
if not issubclass(base_class, object): # Check for old style class
bases += (object,)
return type('{0}{1}'.format(base_class.__name__, self._cassette._path),
bases, dict(cassette=self._cassette))
def _httplib(self): def _httplib(self):
yield httplib, 'HTTPConnection', VCRHTTPConnection yield httplib, 'HTTPConnection', VCRHTTPConnection
yield httplib, 'HTTPSConnection', VCRHTTPSConnection yield httplib, 'HTTPSConnection', VCRHTTPSConnection