feat: split sync and async; test for nest fetching

Author: heitorlessa

Diff for: tests/e2e/tracer/handlers/async_capture.py

@@ -0,0 +1,16 @@
+import asyncio
+from uuid import uuid4
+
+from aws_lambda_powertools import Tracer
+from aws_lambda_powertools.utilities.typing import LambdaContext
+
+tracer = Tracer()
+
+
+@tracer.capture_method
+async def async_get_users():
+    return [{"id": f"{uuid4()}"} for _ in range(5)]
+
+
+def lambda_handler(event: dict, context: LambdaContext):
+    return asyncio.run(async_get_users())

Diff for: tests/e2e/tracer/handlers/basic_handler.py

@@ -1,9 +1,16 @@
+from uuid import uuid4
+
 from aws_lambda_powertools import Tracer
 from aws_lambda_powertools.utilities.typing import LambdaContext
 
 tracer = Tracer()
 
 
+@tracer.capture_method
+def get_todos():
+    return [{"id": f"{uuid4()}", "completed": False} for _ in range(5)]
+
+
 @tracer.capture_lambda_handler
 def lambda_handler(event: dict, context: LambdaContext):
-    return "success"
+    return get_todos()

Diff for: tests/e2e/tracer/handlers/sync_async_capture.py

@@ -1,6 +1,6 @@
 import pytest
 
-from tests.e2e.tracer.handlers import basic_handler, sync_async_capture
+from tests.e2e.tracer.handlers import async_capture, basic_handler
 from tests.e2e.tracer.infrastructure import TracerStack
 from tests.e2e.utils import data_builder, data_fetcher
 
@@ -16,20 +16,25 @@ def basic_handler_fn(infrastructure: dict) -> str:
 
 
 @pytest.fixture
-def sync_async_fn_arn(infrastructure: dict) -> str:
-    return infrastructure.get("SyncAsyncCaptureArn", "")
+def async_fn_arn(infrastructure: dict) -> str:
+    return infrastructure.get("AsyncCaptureArn", "")
 
 
 @pytest.fixture
-def sync_async_fn(infrastructure: dict) -> str:
-    return infrastructure.get("SyncAsyncCapture", "")
+def async_fn(infrastructure: dict) -> str:
+    return infrastructure.get("AsyncCapture", "")
 
 
 def test_lambda_handler_trace_is_visible(basic_handler_fn_arn: str, basic_handler_fn: str):
     # GIVEN
     handler_name = basic_handler.lambda_handler.__name__
     handler_subsegment = f"## {handler_name}"
     handler_metadata_key = f"{handler_name} response"
+
+    method_name = basic_handler.get_todos.__name__
+    method_subsegment = f"## {method_name}"
+    handler_metadata_key = f"{method_name} response"
+
     trace_query = data_builder.build_trace_default_query(function_name=basic_handler_fn)
 
     # WHEN
@@ -41,29 +46,24 @@ def test_lambda_handler_trace_is_visible(basic_handler_fn_arn: str, basic_handle
 
     assert len(trace.get_annotation(key="ColdStart", value=True)) == 1
     assert len(trace.get_metadata(key=handler_metadata_key, namespace=TracerStack.SERVICE_NAME)) == 2
+    assert len(trace.get_metadata(key=handler_metadata_key, namespace=TracerStack.SERVICE_NAME)) == 2
     assert len(trace.get_subsegment(name=handler_subsegment)) == 2
+    assert len(trace.get_subsegment(name=method_subsegment)) == 2
 
 
-def test_sync_async_capture_are_visible(sync_async_fn_arn: str, sync_async_fn: str):
+def test_async_trace_is_visible(async_fn_arn: str, async_fn: str):
     # GIVEN
-    sync_fn_name = sync_async_capture.get_todos.__name__
-    sync_fn_name_subsegment = f"## {sync_fn_name}"
-    sync_fn_name_metadata_key = f"{sync_fn_name} response"
-
-    async_fn_name = sync_async_capture.async_get_users.__name__
+    async_fn_name = async_capture.async_get_users.__name__
     async_fn_name_subsegment = f"## {async_fn_name}"
     async_fn_name_metadata_key = f"{async_fn_name} response"
 
-    trace_query = data_builder.build_trace_default_query(function_name=sync_async_fn)
+    trace_query = data_builder.build_trace_default_query(function_name=async_fn)
 
     # WHEN
-    _, execution_time = data_fetcher.get_lambda_response(lambda_arn=sync_async_fn_arn)
+    _, execution_time = data_fetcher.get_lambda_response(lambda_arn=async_fn_arn)
 
     # THEN
     trace = data_fetcher.get_traces(start_date=execution_time, filter_expression=trace_query)
 
-    assert len(trace.get_subsegment(name=sync_fn_name_subsegment)) == 1
-    assert len(trace.get_metadata(key=sync_fn_name_metadata_key, namespace=TracerStack.SERVICE_NAME)) == 1
-
     assert len(trace.get_subsegment(name=async_fn_name_subsegment)) == 1
     assert len(trace.get_metadata(key=async_fn_name_metadata_key, namespace=TracerStack.SERVICE_NAME)) == 1