Set __wrapped__ on MockAWS decorator to be the test function. cc #41.

This commit is contained in:
Steve Pulec 2013-09-10 08:27:28 -04:00
commit dc2802a238
2 changed files with 9 additions and 0 deletions

View file

@ -47,6 +47,7 @@ class MockAWS(object):
result = func(*args, **kwargs)
return result
functools.update_wrapper(wrapper, func)
wrapper.__wrapped__ = func
return wrapper