Added transaction source and tests for AWS lambda

Author: antonpirker

sentry_sdk/integrations/aws_lambda.py

@@ -6,6 +6,7 @@
 from sentry_sdk.tracing import Transaction
 from sentry_sdk._compat import reraise
 from sentry_sdk.utils import (
+    TRANSACTION_SOURCE_COMPONENT,
     AnnotatedValue,
     capture_internal_exceptions,
     event_from_exception,
@@ -139,7 +140,10 @@ def sentry_handler(aws_event, aws_context, *args, **kwargs):
             if headers is None:
                 headers = {}
             transaction = Transaction.continue_from_headers(
-                headers, op="serverless.function", name=aws_context.function_name
+                headers,
+                op="serverless.function",
+                name=aws_context.function_name,
+                source=TRANSACTION_SOURCE_COMPONENT,
             )
             with hub.start_transaction(
                 transaction,

tests/integrations/aws_lambda/test_aws.py

@@ -362,6 +362,7 @@ def test_handler(event, context):
     assert envelope["type"] == "transaction"
     assert envelope["contexts"]["trace"]["op"] == "serverless.function"
     assert envelope["transaction"].startswith("test_function_")
+    assert envelope["transaction_info"] == {"source": "component"}
     assert envelope["transaction"] in envelope["request"]["url"]
 
 
@@ -390,6 +391,7 @@ def test_handler(event, context):
     assert envelope["type"] == "transaction"
     assert envelope["contexts"]["trace"]["op"] == "serverless.function"
     assert envelope["transaction"].startswith("test_function_")
+    assert envelope["transaction_info"] == {"source": "component"}
     assert envelope["transaction"] in envelope["request"]["url"]