Fix failing tests

DataDog · Sep 24, 2024 · 8c98f51 · 8c98f51
1 parent 83d2dd3
commit 8c98f51
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.../src/test/groovy/datadog/trace/instrumentation/java/lang/StringBuilderCallSiteTest.groovy b/.../src/test/groovy/datadog/trace/instrumentation/java/lang/StringBuilderCallSiteTest.groovy
@@ -57,7 +57,7 @@ class StringBuilderCallSiteTest extends AgentTestRunner {
     if (param.class == String) {
       1 * iastModule.onStringBuilderAppend(target, (String) param)
     } else {
-      1 * iastModule.onStringBuilderAppend(target, param.toString())
+      1 * iastModule.onStringBuilderAppend(target, { it -> it.toString() == param.toString() } )
     }
     _ * TEST_PROFILING_CONTEXT_INTEGRATION._
     0 * _

diff --git a/...pi-2/src/test/groovy/datadog/trace/instrumentation/owasp/esapi/EncoderCallSiteTest.groovy b/...pi-2/src/test/groovy/datadog/trace/instrumentation/owasp/esapi/EncoderCallSiteTest.groovy
@@ -25,7 +25,7 @@ class EncoderCallSiteTest extends AgentTestRunner {
     testSuite.&"$method".call(args)
 
     then:
-    1 * module.taintObjectIfTainted(_, _, false, mark)
+    1 * module.taintStringIfTainted(_, _, false, mark)
     0 * module._
 
     where: