Add some test cases

nikita-tkachenko-datadog · nikita-tkachenko-datadog · commit 6913074b372f · 2024-02-09T14:16:05.000+01:00
diff --git a/telemetry/src/test/groovy/datadog/telemetry/TelemetryRouterSpecification.groovy b/telemetry/src/test/groovy/datadog/telemetry/TelemetryRouterSpecification.groovy
@@ -130,7 +130,7 @@ class TelemetryRouterSpecification extends Specification {
     500        | _
   }
 
-  def 'use Intake client from the start if configured to do so'() {
+  def 'when configured to prefer Intake: use Intake client from the start'() {
     Request request
 
     setup:
@@ -147,6 +147,56 @@ class TelemetryRouterSpecification extends Specification {
     request.header(apiKeyHeader) == apiKey
   }
 
+  def 'when configured to prefer Intake: do not switch to Agent if Intake request succeeds, even if Agent supports telemetry proxy'() {
+    Request request
+
+    setup:
+    def telemetryRouter = new TelemetryRouter(ddAgentFeaturesDiscovery, agentTelemetryClient, intakeTelemetryClient, true)
+
+    when:
+    telemetryRouter.sendRequest(dummyRequest())
+
+    then:
+    1 * ddAgentFeaturesDiscovery.discoverIfOutdated()
+    1 * ddAgentFeaturesDiscovery.supportsTelemetryProxy() >> true
+    1 * okHttpClient.newCall(_) >> { args -> request = args[0]; mockResponse(200) }
+    request.url() == intakeUrl
+    request.header(apiKeyHeader) == apiKey
+
+    when:
+    telemetryRouter.sendRequest(dummyRequest())
+
+    then:
+    1 * okHttpClient.newCall(_) >> { args -> request = args[0]; mockResponse(200) }
+    request.url() == intakeUrl
+    request.header(apiKeyHeader) == apiKey
+  }
+
+  def 'when configured to prefer Intake: switch to Agent if Intake request fails'() {
+    Request request
+
+    setup:
+    def telemetryRouter = new TelemetryRouter(ddAgentFeaturesDiscovery, agentTelemetryClient, intakeTelemetryClient, true)
+
+    when:
+    telemetryRouter.sendRequest(dummyRequest())
+
+    then:
+    1 * ddAgentFeaturesDiscovery.discoverIfOutdated()
+    1 * ddAgentFeaturesDiscovery.supportsTelemetryProxy() >> true
+    1 * okHttpClient.newCall(_) >> { args -> request = args[0]; mockResponse(403) }
+    request.url() == intakeUrl
+    request.header(apiKeyHeader) == apiKey
+
+    when:
+    telemetryRouter.sendRequest(dummyRequest())
+
+    then:
+    1 * okHttpClient.newCall(_) >> { args -> request = args[0]; mockResponse(200) }
+    request.url() == agentTelemetryUrl
+    request.header(apiKeyHeader) == null
+  }
+
   def 'do not switch to Intake when Agent stops supporting telemetry proxy but accepts telemetry requests'() {
     Request request
 
