smoke test

DataDog · sezen-datadog · Jan 16, 2025 · Jan 10, 2025 · Jan 10, 2025 · Jan 10, 2025
commit fe5078c4b97bd7fb8c34754d7fc40fc618720093
@@ -0,0 +1,35 @@
+plugins {
+  id 'idea'
+  id 'java-test-fixtures'
+}
+
+
+apply from: "$rootDir/gradle/java.gradle"
+
+description = 'iast-smoke-tests-utils-java-17'
+
+idea {
+  module {
+    jdkName = '17'
+  }
+}
+
+dependencies {
+  api project(':dd-smoke-tests')
+  compileOnly group: 'org.springframework.boot', name: 'spring-boot-starter-web', version: '2.2.0.RELEASE'
+
+  testFixturesImplementation testFixtures(project(":dd-smoke-tests:iast-util"))
+}
+
+project.tasks.withType(AbstractCompile).configureEach {
+  setJavaVersion(it, 17)
+  sourceCompatibility = JavaVersion.VERSION_17
+  targetCompatibility = JavaVersion.VERSION_17
+  if (it instanceof JavaCompile) {
+    it.options.release.set(17)
+  }
+}
+
+forbiddenApisMain {
+  failOnMissingClasses = false
+}
@@ -0,0 +1,17 @@
+package datadog.smoketest.springboot.controller;
+
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+@RequestMapping("/string")
+public class StringOperationController {
+
+  @PostMapping("/translateEscapes")
+  public String translateEscapes(@RequestParam(value = "parameter") final String parameter) {
+    parameter.translateEscapes();
+    return "ok";
+  }
+}
@@ -0,0 +1,56 @@
+package datadog.smoketest
+
+import okhttp3.FormBody
+import okhttp3.Request
+
+import static datadog.trace.api.config.IastConfig.IAST_DEBUG_ENABLED
+import static datadog.trace.api.config.IastConfig.IAST_DETECTION_MODE
+import static datadog.trace.api.config.IastConfig.IAST_ENABLED
+
+abstract class AbstractIast17SpringBootTest extends AbstractIastServerSmokeTest {
+
+  @Override
+  ProcessBuilder createProcessBuilder() {
+    String springBootShadowJar = System.getProperty('datadog.smoketest.springboot.shadowJar.path')
+
+    List<String> command = []
+    command.add(javaPath())
+    command.addAll(defaultJavaProperties)
+    command.addAll(iastJvmOpts())
+    command.addAll((String[]) ['-jar', springBootShadowJar, "--server.port=${httpPort}"])
+    ProcessBuilder processBuilder = new ProcessBuilder(command)
+    processBuilder.directory(new File(buildDirectory))
+    // Spring will print all environment variables to the log, which may pollute it and affect log assertions.
+    processBuilder.environment().clear()
+    return processBuilder
+  }
+
+  protected List<String> iastJvmOpts() {
+    return [
+      withSystemProperty(IAST_ENABLED, true),
+      withSystemProperty(IAST_DETECTION_MODE, 'FULL'),
+      withSystemProperty(IAST_DEBUG_ENABLED, true),
+    ]
+  }
+
+  void 'test String#translateEscapes'() {
+    setup:
+    final url = "http://localhost:${httpPort}/string/translateEscapes"
+    final body = new FormBody.Builder()
+      .add('paramater', value)
+      .build()
+    final request = new Request.Builder().url(url).post(body).build()
+
+    when:
+    client.newCall(request).execute()
+
+    then:
+    hasTainted { tainted ->
+      tainted.value == expected
+    }
+
+    where:
+    value       | expected
+    "withEscape/ttab" | "withEscape#0009tab"
+  }
+}
@@ -168,6 +168,7 @@ include ':dd-smoke-tests:datastreams:kafkaschemaregistry'
 include ':dd-smoke-tests:iast-propagation'
 include ':dd-smoke-tests:iast-util'
 include ':dd-smoke-tests:iast-util:iast-util-11'
+include ':dd-smoke-tests:iast-util:iast-util-17'
 // TODO this fails too often with a jgit failure, so disable until fixed
 //include ':dd-smoke-tests:debugger-integration-tests:latest-jdk-app'