stuff

Author: antonpirker

tests/integrations/aws_lambda/lambda_functions_with_embedded_sdk/TracesSampler/index.py

@@ -0,0 +1,25 @@
+import json
+import os
+import sentry_sdk
+from sentry_sdk.integrations.aws_lambda import AwsLambdaIntegration
+
+def trace_sampler(sampling_context):
+    print("I AM IN TRACES SAMPLER")
+    return 1
+
+sentry_sdk.init(
+    dsn=os.environ.get("SENTRY_DSN"),
+    traces_sample_rate=1.0,
+    traces_sampler=trace_sampler,
+    integrations=[AwsLambdaIntegration()],
+)
+
+def handler(event, context):
+    # This is a simple Lambda function with Sentry SDK embedded
+    return {
+        "statusCode": 200,
+        "body": json.dumps({
+            "message": "Hello from Lambda with embedded Sentry SDK!",
+            "event": event
+        })
+    }

tests/integrations/aws_lambda/test_aws_lambda.py

@@ -450,3 +450,21 @@ def test_span_origin(lambda_client, test_environment):
     assert (
         transaction_event["contexts"]["trace"]["origin"] == "auto.function.aws_lambda"
     )
+
+
+def test_traces_sampler_has_correct_sampling_context(lambda_client, test_environment):
+    print("111")
+    bla = lambda_client.invoke(
+        FunctionName="TracesSampler",
+        Payload=json.dumps({}),
+    )
+    print(bla)
+    print("222")
+    envelopes = test_environment["server"].envelopes
+    print(envelopes)
+    print("333")
+    (transaction_event,) = envelopes
+    print("444")
+    assert (
+        transaction_event["contexts"]["trace"]["origin"] == "auto.function.aws_lambda"
+    )

tests/integrations/aws_lambda/utils.py

@@ -1,12 +1,15 @@
 import gzip
 import json
 import os
+import shutil
+import subprocess
 import requests
 import sys
 import time
 import threading
 import socket
 import platform
+import tempfile
 
 from aws_cdk import (
     CfnResource,
@@ -20,6 +23,7 @@
 
 
 LAMBDA_FUNCTION_DIR = "./tests/integrations/aws_lambda/lambda_functions/"
+LAMBDA_FUNCTION_WITH_EMBEDDED_SDK_DIR = "./tests/integrations/aws_lambda/lambda_functions_with_embedded_sdk/"
 LAMBDA_FUNCTION_TIMEOUT = 10
 SAM_PORT = 3001
 
@@ -47,6 +51,20 @@ def get_host_ip():
     return host
 
 
+def get_project_root():
+    """
+    Returns the absolute path to the project root directory.
+    """
+    # Start from the current file's directory
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    
+    # Navigate up to the project root (4 levels up from tests/integrations/aws_lambda/)
+    # This is equivalent to the multiple dirname() calls
+    project_root = os.path.abspath(os.path.join(current_dir, "../../../"))
+    
+    return project_root
+
+
 class LocalLambdaStack(Stack):
     """
     Uses the AWS CDK to create a local SAM stack containing Lambda functions.
@@ -124,6 +142,65 @@ def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
                 )
             )
 
+        print(
+            "[LocalLambdaStack] Add all Lambda functions defined in "
+            "/tests/integrations/aws_lambda/lambda_functions_with_embedded_sdk/ to the SAM stack"
+        )
+        lambda_dirs = [
+            d
+            for d in os.listdir(LAMBDA_FUNCTION_WITH_EMBEDDED_SDK_DIR)
+            if os.path.isdir(os.path.join(LAMBDA_FUNCTION_WITH_EMBEDDED_SDK_DIR, d))
+        ]
+        for lambda_dir in lambda_dirs:
+            # Copy the Sentry SDK into the function directory
+            import ipdb; ipdb.set_trace()
+            sdk_path = os.path.join(LAMBDA_FUNCTION_WITH_EMBEDDED_SDK_DIR, lambda_dir, "sentry_sdk")
+            if not os.path.exists(sdk_path):
+                # Find the Sentry SDK in the current environment
+                import sentry_sdk as sdk_module
+                sdk_source = os.path.dirname(sdk_module.__file__)
+                shutil.copytree(sdk_source, sdk_path)
+
+            # Install the requirements of Sentry SDK into the function directory           
+            requirements_file = os.path.join(get_project_root(), "requirements-aws-lambda-layer.txt")
+            
+            # Install the package using pip
+            subprocess.check_call([
+                sys.executable, 
+                "-m", 
+                "pip", 
+                "install", 
+                "--target", 
+                os.path.join(LAMBDA_FUNCTION_WITH_EMBEDDED_SDK_DIR, lambda_dir),
+                "-r", 
+                requirements_file,
+            ])            
+            
+
+            CfnResource(
+                self,
+                lambda_dir,
+                type="AWS::Serverless::Function",
+                properties={
+                    "CodeUri": os.path.join(LAMBDA_FUNCTION_WITH_EMBEDDED_SDK_DIR, lambda_dir),
+                    "Handler": "sentry_sdk.integrations.init_serverless_sdk.sentry_lambda_handler",
+                    "Runtime": PYTHON_VERSION,
+                    "Timeout": LAMBDA_FUNCTION_TIMEOUT,
+                    "Environment": {
+                        "Variables": {
+                            "SENTRY_DSN": dsn,
+                        }
+                    },
+                },
+            )
+            print(
+                "[LocalLambdaStack] - Created Lambda function: %s (%s)"
+                % (
+                    lambda_dir,
+                    os.path.join(LAMBDA_FUNCTION_DIR, lambda_dir),
+                )
+            )
+
     @classmethod
     def wait_for_stack(cls, timeout=30, port=SAM_PORT):
         """