@@ -589,10 +589,10 @@ def handler(event, context):
589
589
handler ({}, mocker .MagicMock ())
590
590
591
591
# THEN
592
- assert in_subsegment_mock .put_annotation .call_args == mocker .call (key = "ColdStart" , value = True )
592
+ assert in_subsegment_mock .put_annotation .call_args_list [ 0 ] == mocker .call (key = "ColdStart" , value = True )
593
593
594
594
handler ({}, mocker .MagicMock ())
595
- assert in_subsegment_mock .put_annotation .call_args == mocker .call (key = "ColdStart" , value = False )
595
+ assert in_subsegment_mock .put_annotation .call_args_list [ 2 ] == mocker .call (key = "ColdStart" , value = False )
596
596
597
597
598
598
def test_tracer_lambda_handler_add_service_annotation (mocker , dummy_response , provider_stub , in_subsegment_mock ):
@@ -609,3 +609,22 @@ def handler(event, context):
609
609
610
610
# THEN
611
611
assert in_subsegment_mock .put_annotation .call_args == mocker .call (key = "Service" , value = "booking" )
612
+
613
+
614
+ def test_tracer_lambda_handler_do_not_add_service_annotation_when_missing (
615
+ mocker , dummy_response , provider_stub , in_subsegment_mock
616
+ ):
617
+ # GIVEN
618
+ provider = provider_stub (in_subsegment = in_subsegment_mock .in_subsegment )
619
+ tracer = Tracer (provider = provider )
620
+
621
+ # WHEN
622
+ @tracer .capture_lambda_handler
623
+ def handler (event , context ):
624
+ return dummy_response
625
+
626
+ handler ({}, mocker .MagicMock ())
627
+
628
+ # THEN
629
+ assert in_subsegment_mock .put_annotation .call_count == 1
630
+ assert in_subsegment_mock .put_annotation .call_args == mocker .call (key = "ColdStart" , value = True )
0 commit comments