From 1a88af7e7c2928e98b9f95ae385c8579bc106e86 Mon Sep 17 00:00:00 2001
From: atshaw43 <108552302+atshaw43@users.noreply.github.com>
Date: Mon, 24 Jul 2023 15:08:19 -0700
Subject: [PATCH] Update TracingHandlerLambdaTest.java

---
 .../amazonaws/xray/handlers/TracingHandlerLambdaTest.java | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/aws-xray-recorder-sdk-aws-sdk/src/test/java/com/amazonaws/xray/handlers/TracingHandlerLambdaTest.java b/aws-xray-recorder-sdk-aws-sdk/src/test/java/com/amazonaws/xray/handlers/TracingHandlerLambdaTest.java
index 371e3222..1ca05b71 100644
--- a/aws-xray-recorder-sdk-aws-sdk/src/test/java/com/amazonaws/xray/handlers/TracingHandlerLambdaTest.java
+++ b/aws-xray-recorder-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);