Update AWSXRayRecorderTest.java

aws · Jul 24, 2023 · 63e3122 · 63e3122
1 parent 1a88af7
commit 63e3122
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/aws-xray-recorder-sdk-core/src/test/java/com/amazonaws/xray/AWSXRayRecorderTest.java b/aws-xray-recorder-sdk-core/src/test/java/com/amazonaws/xray/AWSXRayRecorderTest.java
@@ -967,10 +967,6 @@ public void testDefaultContextMissingBehaviorEnd() {
 
     @Test
     public void testMalformedTraceId() {
-        AWSXRayRecorder recorder = AWSXRayRecorderBuilder.standard()
-            .withSamplingStrategy(new NoSamplingStrategy())
-            .build();
-
         TraceHeader malformedHeader = TraceHeader.fromString("malformedTraceID");
 
         PowerMockito.stub(PowerMockito.method(
@@ -981,6 +977,10 @@ public void testMalformedTraceId() {
                 LambdaSegmentContextResolver.class, "getLambdaTaskRoot"))
                 .toReturn("/var/task");
 
+        AWSXRayRecorder recorder = AWSXRayRecorderBuilder.standard()
+            .withSamplingStrategy(new NoSamplingStrategy())
+            .build();
+
         recorder.beginSubsegment("Test");
 
         // Sanity checks that this is a NoOpSubsegment. (We cannot compare the instance of the private class directly)