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

fix decoration when in _handle_function does not return the wrapped function return value

Browse Source
This commit is contained in:
Bocai Cen
2015-10-01 23:17:15 -04:00
parent 327797c4ff
commit 7990c549d1
2 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/unit/test_vcr.py
View File

@@ -301,6 +301,17 @@ def test_additional_matchers():
function_additional()
def test_decoration_should_respect_function_return_value():
vcr = VCR()
ret = 'a-return-value'
@vcr.use_cassette
def function_with_return():
return ret
assert ret == function_with_return()
class TestVCRClass(VCR().test_case()):
def no_decoration(self):

2
vcr/cassette.py
View File

@@ -127,7 +127,7 @@ class CassetteContextDecorator(object):
def _handle_function(self, function, args, kwargs):
with self as cassette:
self.__handle_function(cassette, function, args, kwargs)
return self.__handle_function(cassette, function, args, kwargs)
@staticmethod
def get_function_name(function):