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

Use contextdecorator from contextlib2. add logging for entering context.

Browse Source
This commit is contained in:
Ivan Malison
2014-09-18 15:25:42 -07:00
parent 4868a63876
commit 0c19acd74f
2 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
vcr/patch.py
View File

@@ -79,11 +79,13 @@ class PatcherBuilder(object):
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._cassette_subclass(klass)
self._class_to_cassette_subclass[klass] = self._build_cassette_subclass(klass)
return self._class_to_cassette_subclass[klass]
def _cassette_subclass(self, base_class):
def _build_cassette_subclass(self, base_class):
bases = (base_class,)
if not issubclass(base_class, object): # Check for old style class
bases += (object,)
@@ -144,7 +146,6 @@ class PatcherBuilder(object):
yield cpool, 'CertValidatingHTTPSConnection', VCRCertValidatingHTTPSConnection
def reset_patchers():
yield mock.patch.object(httplib, 'HTTPConnection', _HTTPConnection)
yield mock.patch.object(httplib, 'HTTPSConnection', _HTTPSConnection)