def test_function_with_version(self):
function_version = "1"
ctx = get_mock_context(
invoked_function_arn=function_arn + ":" + function_version
)
span = create_function_execution_span(ctx, "", False, {"source": ""}, False)
self.assertEqual(span.get_tag("function_arn"), function_arn)
self.assertEqual(span.get_tag("function_version"), function_version)
def _before(self, event, context):
try:
set_cold_start()
submit_invocations_metric(context)
# Extract Datadog trace context from incoming requests
dd_context = extract_dd_trace_context(event)
self.span = None
if dd_tracing_enabled:
set_dd_trace_py_root(dd_context, self.merge_xray_traces)
self.span = create_function_execution_span(
context, self.function_name, is_cold_start(), dd_context)
else:
set_correlation_ids()
logger.debug("datadog_lambda_wrapper _before() done")
except Exception:
traceback.print_exc()
def test_function(self):
ctx = get_mock_context()
span = create_function_execution_span(ctx, "", False, {"source": ""}, False)
self.assertEqual(span.get_tag("function_arn"), function_arn)
self.assertEqual(span.get_tag("function_version"), "$LATEST")