From 7059cca3ccb7d0d31bff99a4b08fef70b4d39c1a Mon Sep 17 00:00:00 2001
From: Johan Vandeweerd <johan.v@sparkcentral.com>
Date: Wed, 15 May 2019 14:13:32 +0200
Subject: [PATCH] Add test for spring webflux webclient

---
 .../client/SpringWebfluxHttpClientTest.groovy | 31 +++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 dd-java-agent/instrumentation/spring-webflux/src/test/groovy/dd/trace/instrumentation/springwebflux/client/SpringWebfluxHttpClientTest.groovy

diff --git a/dd-java-agent/instrumentation/spring-webflux/src/test/groovy/dd/trace/instrumentation/springwebflux/client/SpringWebfluxHttpClientTest.groovy b/dd-java-agent/instrumentation/spring-webflux/src/test/groovy/dd/trace/instrumentation/springwebflux/client/SpringWebfluxHttpClientTest.groovy
new file mode 100644
index 0000000000..7a097ed937
--- /dev/null
+++ b/dd-java-agent/instrumentation/spring-webflux/src/test/groovy/dd/trace/instrumentation/springwebflux/client/SpringWebfluxHttpClientTest.groovy
@@ -0,0 +1,31 @@
+package dd.trace.instrumentation.springwebflux.client
+
+import datadog.trace.agent.test.base.HttpClientTest
+import datadog.trace.instrumentation.springwebflux.client.SpringWebfluxHttpClientDecorator
+import org.springframework.web.reactive.function.client.ClientResponse
+import org.springframework.web.reactive.function.client.WebClient
+import spock.lang.Shared
+
+class SpringWebfluxHttpClientTest extends HttpClientTest<SpringWebfluxHttpClientDecorator> {
+
+  @Shared
+  def client = WebClient.builder().build()
+
+  @Override
+  int doRequest(String method, URI uri, Map<String, String> headers, Closure callback) {
+    assert method == "GET"
+    ClientResponse response = client.get()
+      .headers({ h -> headers.forEach({ key, value -> h.add(key, value) }) })
+      .uri(uri)
+      .exchange()
+      .block()
+
+    callback?.call()
+    response.statusCode().value()
+  }
+
+  @Override
+  SpringWebfluxHttpClientDecorator decorator() {
+    return SpringWebfluxHttpClientDecorator.DECORATE
+  }
+}