Fix /info endpoint in smoke tests (#7930)

DataDog · Nov 11, 2024 · 44ac386 · 44ac386
1 parent 31bfaf7
commit 44ac386
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/...c/test/groovy/datadog/smoketest/asmstandalonebilling/AsmStandaloneBillingSmokeTest.groovy b/...c/test/groovy/datadog/smoketest/asmstandalonebilling/AsmStandaloneBillingSmokeTest.groovy
@@ -60,7 +60,8 @@ class AsmStandaloneBillingSmokeTest extends AbstractAsmStandaloneBillingSmokeTes
 
     then: 'Check if Datadog-Client-Computed-Stats header is present and set to true, Disabling the metrics computation agent-side'
     waitForTraceCount(1)
-    def computedStatsHeader = server.lastRequest.headers.get('Datadog-Client-Computed-Stats')
+    lastTraceRequestHeaders != null
+    def computedStatsHeader = lastTraceRequestHeaders.get('Datadog-Client-Computed-Stats')
     assert computedStatsHeader != null && computedStatsHeader == 'true'
 
     then:'metrics should be disabled'

diff --git a/dd-smoke-tests/src/main/groovy/datadog/smoketest/AbstractSmokeTest.groovy b/dd-smoke-tests/src/main/groovy/datadog/smoketest/AbstractSmokeTest.groovy
@@ -37,6 +37,9 @@ abstract class AbstractSmokeTest extends ProcessManager {
   @Shared
   private Throwable traceDecodingFailure = null
 
+  @Shared
+  protected TestHttpServer.Headers lastTraceRequestHeaders = null
+
   @Shared
   @AutoCleanup
   protected TestHttpServer server = httpServer {
@@ -48,7 +51,7 @@ abstract class AbstractSmokeTest extends ProcessManager {
           "endpoints": [
             "/v0.4/traces",
             "/v0.5/traces",
-            "/telemetry/proxy/",
+            "/telemetry/proxy/"
           ],
           "client_drop_p0s": true,
           "span_meta_structs": true,
@@ -82,6 +85,7 @@ abstract class AbstractSmokeTest extends ProcessManager {
           }
         }
         traceCount.addAndGet(count)
+        lastTraceRequestHeaders = request.headers
         println("Received v0.4 traces: " + countString)
         response.status(200).send()
       }
@@ -104,9 +108,13 @@ abstract class AbstractSmokeTest extends ProcessManager {
           }
         }
         traceCount.addAndGet(count)
+        lastTraceRequestHeaders = request.headers
         println("Received v0.5 traces: " + countString)
         response.status(200).send()
       }
+      prefix("/v0.6/stats") {
+        response.status(200).send()
+      }
       prefix("/v0.7/config") {
         response.status(200).send(remoteConfigResponse)
       }