GraalVM Native JMXFetch smoke test

Author: ygree

dd-smoke-tests/spring-boot-3.0-native/application/build.gradle

@@ -39,6 +39,7 @@ if (hasProperty('agentPath')) {
         if (withProfiler && property('profiler') == 'true') {
           buildArgs.add("-J-Ddd.profiling.enabled=true")
         }
+        buildArgs.add("--enable-monitoring=jmxserver")
       }
     }
   }

dd-smoke-tests/spring-boot-3.0-native/src/test/groovy/SpringBootNativeInstrumentationTest.groovy

@@ -1,4 +1,5 @@
 import datadog.smoketest.AbstractServerSmokeTest
+import datadog.trace.agent.test.utils.PortUtils
 import okhttp3.Request
 import org.openjdk.jmc.common.item.IItemCollection
 import org.openjdk.jmc.common.item.ItemFilters
@@ -13,6 +14,10 @@ import java.nio.file.Files
 import java.nio.file.Path
 import java.nio.file.SimpleFileVisitor
 import java.nio.file.attribute.BasicFileAttributes
+import java.util.concurrent.CompletableFuture
+import java.util.concurrent.Executors
+import java.util.concurrent.TimeUnit
+import java.util.concurrent.TimeoutException
 import java.util.concurrent.atomic.AtomicInteger
 import java.util.concurrent.locks.LockSupport
 
@@ -21,6 +26,9 @@ class SpringBootNativeInstrumentationTest extends AbstractServerSmokeTest {
   @TempDir
   def testJfrDir
 
+  @Shared
+  def statsdPort = PortUtils.randomOpenPort()
+
   @Override
   ProcessBuilder createProcessBuilder() {
     String springNativeExecutable = System.getProperty('datadog.smoketest.spring.native.executable')
@@ -39,7 +47,10 @@ class SpringBootNativeInstrumentationTest extends AbstractServerSmokeTest {
       '-Ddd.profiling.upload.period=1',
       '-Ddd.profiling.start-force-first=true',
       "-Ddd.profiling.debug.dump_path=${testJfrDir}",
-      "-Ddd.integration.spring-boot.enabled=true"
+      "-Ddd.integration.spring-boot.enabled=true",
+      "-Ddd.trace.debug=true",
+      "-Ddd.jmxfetch.statsd.port=${statsdPort}",
+      "-Ddd.jmxfetch.start-delay=0",
     ])
     ProcessBuilder processBuilder = new ProcessBuilder(command)
     processBuilder.directory(new File(buildDirectory))
@@ -66,8 +77,18 @@ class SpringBootNativeInstrumentationTest extends AbstractServerSmokeTest {
     super.isErrorLog(log) || log.contains("ClassNotFoundException")
   }
 
+  def setupSpec() {
+    try {
+      processTestLogLines { it.contains("JMXFetch config: ") }
+    } catch (TimeoutException toe) {
+      throw new AssertionError("'JMXFetch config: ' not found in logs. Make sure it's enabled.", toe)
+    }
+  }
+
   def "check native instrumentation"() {
     setup:
+    CompletableFuture<String> udpMessage = receiveUdpMessage(statsdPort, 1000)
+
     String url = "http://localhost:${httpPort}/hello"
 
     when:
@@ -87,6 +108,8 @@ class SpringBootNativeInstrumentationTest extends AbstractServerSmokeTest {
       LockSupport.parkNanos(1_000_000)
     }
     countJfrs() > 0
+
+    udpMessage.get(1, TimeUnit.SECONDS) contains "service:smoke-test-java-app,version:99,env:smoketest"
   }
 
   int countJfrs() {
@@ -115,4 +138,20 @@ class SpringBootNativeInstrumentationTest extends AbstractServerSmokeTest {
       })
     return jfrCount.get()
   }
+
+  CompletableFuture<String> receiveUdpMessage(int port, int bufferSize) {
+    def future = new CompletableFuture<String>()
+    Executors.newSingleThreadExecutor().submit {
+      try (DatagramSocket socket = new DatagramSocket(port)) {
+        byte[] buffer = new byte[bufferSize]
+        DatagramPacket packet = new DatagramPacket(buffer, buffer.length)
+        socket.receive(packet)
+        def received = new String(packet.data, 0, packet.length)
+        future.complete(received)
+      } catch (Exception e) {
+        future.completeExceptionally(e)
+      }
+    }
+    return future
+  }
 }