Add encode_authorizer_context.

Author: purple4reina

datadog_lambda/config.py

@@ -80,6 +80,8 @@ class Config:
 
     make_inferred_span = _get_env("DD_TRACE_MANAGED_SERVICES", "true", as_bool,
                                    depends_on_tracing=True)
+    encode_authorizer_context = _get_env("DD_ENCODE_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_ENCODE_AUTHORIZER_CONTEXT = "DD_ENCODE_AUTHORIZER_CONTEXT"
 DD_DECODE_AUTHORIZER_CONTEXT = "DD_DECODE_AUTHORIZER_CONTEXT"
 DD_COLD_START_TRACING = "DD_COLD_START_TRACING"
 DD_REQUESTS_SERVICE_NAME = "DD_REQUESTS_SERVICE_NAME"
@@ -126,9 +125,6 @@ def __init__(self, func):
             depends_on_dd_tracing_enabled = (
                 lambda original_boolean: config.trace_enabled and original_boolean
             )
-            self.encode_authorizer_context = depends_on_dd_tracing_enabled(
-                os.environ.get(DD_ENCODE_AUTHORIZER_CONTEXT, "true").lower() == "true"
-            )
             self.decode_authorizer_context = depends_on_dd_tracing_enabled(
                 os.environ.get(DD_DECODE_AUTHORIZER_CONTEXT, "true").lower() == "true"
             )
@@ -337,7 +333,7 @@ def _after(self, event, context):
             if config.llmobs_enabled:
                 LLMObs.flush()
 
-            if self.encode_authorizer_context and is_authorizer_response(self.response):
+            if config.encode_authorizer_context and is_authorizer_response(self.response):
                 self._inject_authorizer_span_headers(
                     event.get("requestContext", {}).get("requestId")
                 )

tests/test_wrapper.py

@@ -562,10 +562,8 @@ def return_type_test(event, context):
 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_ENCODE_AUTHORIZER_CONTEXT] = "true"
         os.environ[wrapper.DD_DECODE_AUTHORIZER_CONTEXT] = "true"
         decorator = wrapper._LambdaDecorator(func=None)
-        self.assertFalse(decorator.encode_authorizer_context)
         self.assertFalse(decorator.decode_authorizer_context)