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

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
 
 

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')