From 8c98f519ddc30cf837c7b227ad68eec92ebaa9ba Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Manuel=20=C3=81lvarez=20=C3=81lvarez?= Date: Tue, 24 Sep 2024 13:27:31 +0200 Subject: [PATCH] Fix failing tests --- .../instrumentation/java/lang/StringBuilderCallSiteTest.groovy | 2 +- .../instrumentation/owasp/esapi/EncoderCallSiteTest.groovy | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dd-java-agent/instrumentation/java-lang/src/test/groovy/datadog/trace/instrumentation/java/lang/StringBuilderCallSiteTest.groovy b/dd-java-agent/instrumentation/java-lang/src/test/groovy/datadog/trace/instrumentation/java/lang/StringBuilderCallSiteTest.groovy index f8a25369ebb..109262fab2f 100644 --- a/dd-java-agent/instrumentation/java-lang/src/test/groovy/datadog/trace/instrumentation/java/lang/StringBuilderCallSiteTest.groovy +++ b/dd-java-agent/instrumentation/java-lang/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/dd-java-agent/instrumentation/owasp-esapi-2/src/test/groovy/datadog/trace/instrumentation/owasp/esapi/EncoderCallSiteTest.groovy b/dd-java-agent/instrumentation/owasp-esapi-2/src/test/groovy/datadog/trace/instrumentation/owasp/esapi/EncoderCallSiteTest.groovy index 800b7653a77..c90ad8e96bc 100644 --- a/dd-java-agent/instrumentation/owasp-esapi-2/src/test/groovy/datadog/trace/instrumentation/owasp/esapi/EncoderCallSiteTest.groovy +++ b/dd-java-agent/instrumentation/owasp-esapi-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: