Add test for spring webflux webclient

This commit is contained in:
Johan Vandeweerd 2019-05-15 14:13:32 +02:00 committed by Tyler Benson
parent 872a099722
commit 7059cca3cc
1 changed files with 31 additions and 0 deletions

View File

@ -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
}
}