Add decode_authorizer_context.

Author: purple4reina

datadog_lambda/config.py

@@ -82,6 +82,8 @@ class Config:
                                    depends_on_tracing=True)
     encode_authorizer_context = _get_env("DD_ENCODE_AUTHORIZER_CONTEXT", "true", as_bool,
                                          depends_on_tracing=True)
+    decode_authorizer_context = _get_env("DD_DECODE_AUTHORIZER_CONTEXT", "true", as_bool,
+                                         depends_on_tracing=True)
 
     local_test = _get_env("DD_LOCAL_TEST", "false", as_bool)
 

datadog_lambda/wrapper.py

@@ -57,7 +57,6 @@
 logger = logging.getLogger(__name__)
 
 DD_TRACE_MANAGED_SERVICES = "DD_TRACE_MANAGED_SERVICES"
-DD_DECODE_AUTHORIZER_CONTEXT = "DD_DECODE_AUTHORIZER_CONTEXT"
 DD_COLD_START_TRACING = "DD_COLD_START_TRACING"
 DD_REQUESTS_SERVICE_NAME = "DD_REQUESTS_SERVICE_NAME"
 DD_SERVICE = "DD_SERVICE"
@@ -125,9 +124,6 @@ def __init__(self, func):
             depends_on_dd_tracing_enabled = (
                 lambda original_boolean: config.trace_enabled and original_boolean
             )
-            self.decode_authorizer_context = depends_on_dd_tracing_enabled(
-                os.environ.get(DD_DECODE_AUTHORIZER_CONTEXT, "true").lower() == "true"
-            )
             self.cold_start_tracing = depends_on_dd_tracing_enabled(
                 os.environ.get(DD_COLD_START_TRACING, "true").lower() == "true"
             )
@@ -229,7 +225,7 @@ def _before(self, event, context):
                 event,
                 context,
                 extractor=self.trace_extractor,
-                decode_authorizer_context=self.decode_authorizer_context,
+                decode_authorizer_context=config.decode_authorizer_context,
             )
             self.event_source = event_source
             # Create a Datadog X-Ray subsegment with the trace context
@@ -247,7 +243,7 @@ def _before(self, event, context):
                 set_dd_trace_py_root(trace_context_source, config.merge_xray_traces)
                 if config.make_inferred_span:
                     self.inferred_span = create_inferred_span(
-                        event, context, event_source, self.decode_authorizer_context
+                        event, context, event_source, config.decode_authorizer_context
                     )
                 self.span = create_function_execution_span(
                     context=context,

tests/test_wrapper.py

@@ -559,14 +559,6 @@ def return_type_test(event, context):
             self.assertFalse(MockPrintExc.called)
 
 
-class TestLambdaDecoratorSettings(unittest.TestCase):
-    @patch("datadog_lambda.config.Config.trace_enabled", False)
-    def test_some_envs_should_depend_on_dd_tracing_enabled(self):
-        os.environ[wrapper.DD_DECODE_AUTHORIZER_CONTEXT] = "true"
-        decorator = wrapper._LambdaDecorator(func=None)
-        self.assertFalse(decorator.decode_authorizer_context)
-
-
 class TestLambdaWrapperWithTraceContext(unittest.TestCase):
     xray_root = "1-5e272390-8c398be037738dc042009320"
     xray_parent = "94ae789b969f1cc5"