Disable LumigoChalice when outside AWS or by switch-off env var (#256)

Author: nirLumigo

.secrets.baseline

@@ -200,14 +200,14 @@
         "filename": "src/test/unit/test_tracer.py",
         "hashed_secret": "f13de0da12f8375756b1bc56b248cc8e62e0c302",
         "is_verified": false,
-        "line_number": 329
+        "line_number": 344
       },
       {
         "type": "Secret Keyword",
         "filename": "src/test/unit/test_tracer.py",
         "hashed_secret": "dabe695c9892b5cb937379817679c0fb0be6a3df",
         "is_verified": false,
-        "line_number": 407
+        "line_number": 422
       }
     ],
     "src/test/unit/wrappers/http/test_http_parser.py": [
@@ -220,5 +220,5 @@
       }
     ]
   },
-  "generated_at": "2022-08-08T04:42:17Z"
+  "generated_at": "2022-08-17T15:30:21Z"
 }

src/lumigo_tracer/tracer.py

@@ -109,6 +109,12 @@ class LumigoChalice:
         "on_dynamodb_record",
     ]
 
+    def __new__(cls, app, *args, **kwargs):
+        if is_aws_environment() and not is_kill_switch_on():
+            return super().__new__(cls)
+        get_logger().debug("Disabling LumigoChalice")
+        return app
+
     def __init__(self, app, *args, **kwargs):
         self.lumigo_conf_args = args
         self.lumigo_conf_kwargs = kwargs

src/test/unit/test_tracer.py

@@ -228,7 +228,10 @@ def lambda_test_function(event, context):
         assert http_spans[0]["info"]["httpInfo"]["request"]["headers"]
 
 
-def test_lumigo_chalice(context):
+def test_lumigo_chalice(context, monkeypatch):
+    # mimic aws env
+    monkeypatch.setenv("AWS_LAMBDA_FUNCTION_VERSION", "true")
+
     class App:
         @property
         def a(self):
@@ -255,7 +258,7 @@ def __call__(self, *args, **kwargs):
 
 def test_lumigo_chalice_create_extra_lambdas(monkeypatch, context):
     # mimic aws env
-    monkeypatch.setitem(os.environ, "AWS_LAMBDA_FUNCTION_VERSION", "true")
+    monkeypatch.setenv("AWS_LAMBDA_FUNCTION_VERSION", "true")
 
     class Chalice:
         """
@@ -293,6 +296,18 @@ def handler(event, context):
     assert SpansContainer._span
 
 
+def test_lumigo_chalice_disabled_when_not_in_aws(monkeypatch):
+    monkeypatch.delenv("AWS_LAMBDA_FUNCTION_VERSION", raising=False)
+    monkeypatch.delenv("LUMIGO_SWITCH_OFF", raising=False)
+    assert LumigoChalice("myApp") == "myApp"
+
+
+def test_lumigo_chalice_disabled_when_switch_off(monkeypatch):
+    monkeypatch.setenv("AWS_LAMBDA_FUNCTION_VERSION", "true")
+    monkeypatch.setenv("LUMIGO_SWITCH_OFF", "true")
+    assert LumigoChalice("myApp") == "myApp"
+
+
 @pytest.mark.parametrize(
     "event, expected_triggered_by, expected_message_id",
     [