From dc2802a2383a1799b75febb71f078d3a85136959 Mon Sep 17 00:00:00 2001 From: Steve Pulec Date: Tue, 10 Sep 2013 08:27:28 -0400 Subject: [PATCH] Set __wrapped__ on MockAWS decorator to be the test function. cc #41. --- moto/core/models.py | 1 + tests/test_core/test_decorator_calls.py | 8 ++++++++ 2 files changed, 9 insertions(+) diff --git a/moto/core/models.py b/moto/core/models.py index c451fb11d..f3e6ad701 100644 --- a/moto/core/models.py +++ b/moto/core/models.py @@ -47,6 +47,7 @@ class MockAWS(object): result = func(*args, **kwargs) return result functools.update_wrapper(wrapper, func) + wrapper.__wrapped__ = func return wrapper diff --git a/tests/test_core/test_decorator_calls.py b/tests/test_core/test_decorator_calls.py index f9bdbe1ac..444bae89d 100644 --- a/tests/test_core/test_decorator_calls.py +++ b/tests/test_core/test_decorator_calls.py @@ -42,3 +42,11 @@ def test_decorator_start_and_stop(): mock.stop() conn.get_all_instances.when.called_with().should.throw(EC2ResponseError) + + +@mock_ec2 +def test_decorater_wrapped_gets_set(): + """ + Moto decorator's __wrapped__ should get set to the tests function + """ + test_decorater_wrapped_gets_set.__wrapped__.__name__.should.equal('test_decorater_wrapped_gets_set')