test edge cases

DataDog · nhulston · Feb 19, 2025 · Sep 5, 2024 · Sep 9, 2024 · Sep 13, 2024
commit 0aac23904342f686880a3ef75faf68a3b2b3db5d
@@ -22,7 +22,7 @@ public static String buildTraceContext(AgentSpan span) {
   }
 
   public static String getModifiedInput(String request, String ddTraceContextJSON) {
-    StringBuilder modifiedInput = new StringBuilder(request);
+    StringBuilder modifiedInput = new StringBuilder(request.trim());
     int startPos = modifiedInput.indexOf("{");
     int endPos = modifiedInput.lastIndexOf("}");
 

@@ -115,6 +115,32 @@ abstract class SfnClientTest extends VersionedNamingTestBase {
     input["_datadog"]["x-datadog-parent-id"] != null
     input["_datadog"]["x-datadog-tags"] != null
   }
+
+  def "Doesn't cause error for Step Functions input edge cases"() {
+    def inputs = [
+      '''{}''',
+      '''{  }''',
+      ''' { } ''',
+      '''{"foo": "bar"}''',
+      '''  {  "foo"  :  "bar"  }  ''',
+      '''{"key1": "val1", "key2": "val2"}''',
+      '''  {  "key1"  :  "val1"   ,   "key2" : "val2"  } '''
+    ]
+
+    when:
+    inputs.forEach { input ->
+      TraceUtils.runUnderTrace('parent', {
+        sfnClient.startExecution { builder ->
+          builder.stateMachineArn(testStateMachineARN)
+            .input(input)
+            .build()
+        }
+      })
+    }
+
+    then:
+    noExceptionThrown()
+  }
 }
 
 class SfnClientV0Test extends SfnClientTest {