improv: support kwargs

Author: heitorlessa

aws_lambda_powertools/tracing/tracer.py

@@ -269,7 +269,7 @@ def handler(event, context):
         lambda_handler_name = lambda_handler.__name__
 
         @functools.wraps(lambda_handler)
-        def decorate(event, context):
+        def decorate(event, context, **kwargs):
             with self.provider.in_subsegment(name=f"## {lambda_handler_name}") as subsegment:
                 global is_cold_start
                 if is_cold_start:
@@ -279,7 +279,7 @@ def decorate(event, context):
 
                 try:
                     logger.debug("Calling lambda handler")
-                    response = lambda_handler(event, context)
+                    response = lambda_handler(event, context, **kwargs)
                     logger.debug("Received lambda handler response successfully")
                     self._add_response_as_metadata(
                         method_name=lambda_handler_name,

tests/functional/test_tracing.py

@@ -34,6 +34,19 @@ def handler(event, context):
     handler({}, {})
 
 
+def test_capture_lambda_handler_with_additional_kwargs(dummy_response):
+    # GIVEN tracer lambda handler decorator is used
+    tracer = Tracer(disabled=True)
+
+    # WHEN a lambda handler signature has additional keyword arguments
+    @tracer.capture_lambda_handler
+    def handler(event, context, my_extra_option=None, **kwargs):
+        return dummy_response
+
+    # THEN tracer should not raise an Exception
+    handler({}, {}, blah="blah")
+
+
 def test_capture_method(dummy_response):
     # GIVEN tracer method decorator is used
     tracer = Tracer(disabled=True)