Update TracingHandlerLambdaTest.java

aws · Jul 24, 2023 · 1a88af7 · 1a88af7
1 parent dda2f84
commit 1a88af7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/...order-sdk-aws-sdk/src/test/java/com/amazonaws/xray/handlers/TracingHandlerLambdaTest.java b/...order-sdk-aws-sdk/src/test/java/com/amazonaws/xray/handlers/TracingHandlerLambdaTest.java
@@ -73,6 +73,8 @@ public void testSamplingOverrideFalseInLambda() throws Exception {
     public void testSamplingOverrideTrueInLambda() {
         Emitter mockedEmitted = Mockito.mock(DefaultEmitter.class);
 
+        TraceHeader header = TraceHeader.fromString(TRACE_HEADER);
+
         PowerMockito.stub(PowerMockito.method(
                 LambdaSegmentContext.class, "getTraceHeaderFromEnvironment")).toReturn(header);
         PowerMockito.stub(PowerMockito.method(
@@ -83,8 +85,6 @@ public void testSamplingOverrideTrueInLambda() {
                 .withEmitter(mockedEmitted)
                 .build();
 
-        TraceHeader header = TraceHeader.fromString(TRACE_HEADER);
-
         Mockito.doAnswer(invocation -> { return true; }).when(mockedEmitted).sendSubsegment(any());
 
         lambdaTestHelper(recorder, "testTrue", true);
@@ -95,6 +95,8 @@ public void testSamplingOverrideTrueInLambda() {
     public void testSamplingOverrideMixedInLambda() {
         Emitter mockedEmitted = Mockito.mock(DefaultEmitter.class);
 
+        TraceHeader header = TraceHeader.fromString(TRACE_HEADER);
+
         PowerMockito.stub(PowerMockito.method(
                 LambdaSegmentContext.class, "getTraceHeaderFromEnvironment")).toReturn(header);
         PowerMockito.stub(PowerMockito.method(
@@ -105,8 +107,6 @@ public void testSamplingOverrideMixedInLambda() {
                 .withEmitter(mockedEmitted)
                 .build();
 
-        TraceHeader header = TraceHeader.fromString(TRACE_HEADER);
-
         Mockito.doAnswer(invocation -> { return true; }).when(mockedEmitted).sendSubsegment(any());
 
         lambdaTestHelper(recorder, "test1", true);