Merge pull request #17 from ajbarry/fix-decorators

Fix _wrapped_func swallows return val

Author: rstevens011

src/aws_secretsmanager_caching/decorators.py

@@ -46,7 +46,7 @@ def _wrapped_func(*args, **kwargs):
             """
             Internal function to execute wrapped function
             """
-            func(secret, *args, **kwargs)
+            return func(secret, *args, **kwargs)
 
         return _wrapped_func
 
@@ -99,6 +99,6 @@ def _wrapped_func(*args, **kwargs):
             """
             Internal function to execute wrapped function
             """
-            func(*args, **resolved_kwargs, **kwargs)
+            return func(*args, **resolved_kwargs, **kwargs)
 
         return _wrapped_func

test/unit/test_decorators.py

@@ -56,8 +56,9 @@ def function_to_be_decorated(func_username, func_password, keyworded_argument='f
             self.assertEqual(secret['username'], func_username)
             self.assertEqual(secret['password'], func_password)
             self.assertEqual(keyworded_argument, 'foo')
+            return 'OK'
 
-        function_to_be_decorated()
+        self.assertEqual(function_to_be_decorated(), 'OK')
 
     def test_valid_json_with_mixed_args(self):
         secret = {
@@ -170,8 +171,9 @@ def function_to_be_decorated(arg1, arg2, arg3):
             self.assertEqual(arg1, secret)
             self.assertEqual(arg2, 'foo')
             self.assertEqual(arg3, 'bar')
+            return 'OK'
 
-        function_to_be_decorated('foo', 'bar')
+        self.assertEqual(function_to_be_decorated('foo', 'bar'), 'OK')
 
     def test_string_with_additional_kwargs(self):
         secret = 'not json'