improv: disable tracer when using the Chalice CLI (#172)

Author: jamesls

aws_lambda_powertools/tracing/tracer.py

@@ -622,14 +622,15 @@ def _is_tracer_disabled() -> bool:
         """
         logger.debug("Verifying whether Tracing has been disabled")
         is_lambda_sam_cli = os.getenv("AWS_SAM_LOCAL")
+        is_chalice_cli = os.getenv("AWS_CHALICE_CLI_MODE")
         env_option = str(os.getenv("POWERTOOLS_TRACE_DISABLED", "false"))
         disabled_env = strtobool(env_option)
 
         if disabled_env:
             logger.debug("Tracing has been disabled via env var POWERTOOLS_TRACE_DISABLED")
             return disabled_env
 
-        if is_lambda_sam_cli:
+        if is_lambda_sam_cli or is_chalice_cli:
             logger.debug("Running under SAM CLI env or not in Lambda env; disabling Tracing")
             return True
 

tests/functional/test_tracing.py

@@ -61,6 +61,20 @@ def handler(event, context):
     handler({}, {})
 
 
+def test_tracer_chalice_cli_mode(monkeypatch, dummy_response):
+    # GIVEN tracer runs locally
+    monkeypatch.setenv("AWS_CHALICE_CLI_MODE", "true")
+    tracer = Tracer()
+
+    # WHEN a lambda function is run through the Chalice CLI.
+    @tracer.capture_lambda_handler
+    def handler(event, context):
+        return dummy_response
+
+    # THEN tracer should run in disabled mode, and not raise an Exception
+    handler({}, {})
+
+
 def test_tracer_metadata_disabled(dummy_response):
     # GIVEN tracer is disabled, and annotations/metadata are used
     tracer = Tracer(disabled=True)