A rd 5221 alb triggered by support (#198)

* add trigger by load_balancer

Author: sapirlumigo

src/lumigo_tracer/event/event_trigger.py

@@ -31,6 +31,8 @@ def parse_triggered_by(event: dict):
             return None
         if _is_supported_http_method(event):
             return _parse_http_method(event)
+        if _is_load_balancer_method(event):
+            return _parse_load_balancer_method(event)
         elif _is_supported_sns(event):
             return _parse_sns(event)
         elif _is_supported_streams(event):
@@ -77,7 +79,20 @@ def _is_supported_http_method(event: dict):
         and event.get("requestContext", {}).get("stage") is not None  # noqa
     ) or (  # noqa
         event.get("version", "") == "2.0" and "headers" in event  # noqa
-    )  # noqa  # noqa
+    )  # noqa
+
+
+def _is_load_balancer_method(event: dict):
+    return (
+        "httpMethod" in event  # noqa
+        and "headers" in event  # noqa
+        and event["headers"].get("host")  # noqa
+        and "requestContext" in event  # noqa
+        and (  # noqa
+            event.get("requestContext", {}).get("elb") is not None  # noqa
+            or event.get("requestContext", {}).get("alb") is not None  # noqa
+        )  # noqa
+    )
 
 
 def _parse_http_method(event: dict):
@@ -87,6 +102,16 @@ def _parse_http_method(event: dict):
     return _parse_http_method_v1(event)
 
 
+def _parse_load_balancer_method(event: dict):
+    result = {
+        "triggeredBy": "load_balancer",
+        "httpMethod": event.get("httpMethod", ""),
+    }
+    if isinstance(event.get("headers"), dict):
+        result["api"] = event["headers"].get("host")
+    return result
+
+
 def _parse_http_method_v1(event: dict):
     result = {
         "triggeredBy": "apigw",

src/test/unit/event/test_event_trigger.py

@@ -282,7 +282,7 @@
                 "detailType": "Scheduled Event",
             },
         ),
-        (  # elb example trigger
+        (  # old elb example trigger
             {
                 "requestContext": {
                     "elb": {
@@ -302,7 +302,73 @@
                     "host": "lambd-loadb-bp68mp6nujg0-50156485.us-east-1.elb.amazonaws.com",
                 },
             },
-            {"triggeredBy": "unknown"},
+            {
+                "api": "lambd-loadb-bp68mp6nujg0-50156485.us-east-1.elb.amazonaws.com",
+                "httpMethod": "POST",
+                "triggeredBy": "load_balancer",
+            },
+        ),
+        (  # new elb example trigger
+            {
+                "requestContext": {
+                    "elb": {
+                        "targetGroupArn": "arn:aws:elasticloadbalancing:region:123456789012:targetgroup/my-target-group/111"
+                    }
+                },
+                "httpMethod": "GET",
+                "path": "/",
+                "queryStringParameters": {},
+                "headers": {
+                    "accept": "text/html,application/xhtml+xml",
+                    "accept-language": "en-US,en;q=0.8",
+                    "content-type": "text/plain",
+                    "cookie": "cookies",
+                    "host": "lambda-111-us-east-2.elb.amazonaws.com",
+                    "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_11_6)",
+                    "x-amzn-trace-id": "Root=1-111",
+                    "x-forwarded-for": "111",
+                    "x-forwarded-port": "111",
+                    "x-forwarded-proto": "https",
+                },
+                "isBase64Encoded": False,
+                "body": "request_body",
+            },
+            {
+                "api": "lambda-111-us-east-2.elb.amazonaws.com",
+                "httpMethod": "GET",
+                "triggeredBy": "load_balancer",
+            },
+        ),
+        (  # alb example trigger
+            {
+                "requestContext": {
+                    "alb": {
+                        "targetGroupArn": "arn:aws:elasticloadbalancing:region:123456789012:targetgroup/my-target-group/111"
+                    }
+                },
+                "httpMethod": "GET",
+                "path": "/",
+                "queryStringParameters": {},
+                "headers": {
+                    "accept": "text/html,application/xhtml+xml",
+                    "accept-language": "en-US,en;q=0.8",
+                    "content-type": "text/plain",
+                    "cookie": "cookies",
+                    "host": "lambda-111-us-east-2.elb.amazonaws.com",
+                    "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_11_6)",
+                    "x-amzn-trace-id": "Root=1-111",
+                    "x-forwarded-for": "111",
+                    "x-forwarded-port": "111",
+                    "x-forwarded-proto": "https",
+                },
+                "isBase64Encoded": False,
+                "body": "request_body",
+            },
+            {
+                "api": "lambda-111-us-east-2.elb.amazonaws.com",
+                "httpMethod": "GET",
+                "triggeredBy": "load_balancer",
+            },
         ),
         (  # API GW V2
             {