@@ -580,9 +580,8 @@ def test_mixed_parent_context_when_merging(self):
580
580
581
581
582
582
class TestAuthorizerInferredSpans (unittest .TestCase ):
583
-
584
583
def setUp (self ):
585
- patcher = patch (' ddtrace.Span.finish' , autospec = True )
584
+ patcher = patch (" ddtrace.Span.finish" , autospec = True )
586
585
self .mock_span_stop = patcher .start ()
587
586
self .addCleanup (patcher .stop )
588
587
@@ -615,17 +614,19 @@ def test_create_inferred_span_from_authorizer_request_api_gateway_v1_event(self)
615
614
self .assertEqual (span .get_tag ("apiid" ), "amddr1rix9" )
616
615
self .assertEqual (span .get_tag ("apiname" ), "amddr1rix9" )
617
616
self .assertEqual (span .get_tag ("stage" ), "dev" )
618
- self .assertEqual (span .start , 1663295021.832 ) # request_time_epoch + integrationLatency
617
+ self .assertEqual (
618
+ span .start , 1663295021.832
619
+ ) # request_time_epoch + integrationLatency
619
620
self .assertEqual (span .span_type , "http" )
620
621
self .assertEqual (span .get_tag (InferredSpanInfo .TAG_SOURCE ), "self" )
621
622
self .assertEqual (span .get_tag (InferredSpanInfo .SYNCHRONICITY ), "sync" )
622
623
623
624
# checking the upstream_authorizer_span
624
625
self .mock_span_stop .assert_called_once ()
625
626
args , kwargs = self .mock_span_stop .call_args_list [0 ]
626
- self .assertEqual (kwargs [' finish_time' ], 1663295021.832 )
627
+ self .assertEqual (kwargs [" finish_time" ], 1663295021.832 )
627
628
authorizer_span = args [0 ]
628
- self .assertEqual (authorizer_span .name , ' aws.apigateway.authorizer' )
629
+ self .assertEqual (authorizer_span .name , " aws.apigateway.authorizer" )
629
630
self .assertEqual (span .parent_id , authorizer_span .span_id )
630
631
631
632
0 commit comments