Use OkHttpUtils in more places

This commit is contained in:
Nikolay Martynov 2018-09-26 15:07:20 -04:00
parent 655d490da2
commit 5d627733d8
4 changed files with 13 additions and 18 deletions

View File

@ -1,8 +1,8 @@
import datadog.trace.agent.test.AgentTestRunner import datadog.trace.agent.test.AgentTestRunner
import datadog.trace.agent.test.utils.OkHttpUtils
import datadog.trace.api.DDSpanTypes import datadog.trace.api.DDSpanTypes
import datadog.trace.api.DDTags import datadog.trace.api.DDTags
import io.opentracing.tag.Tags import io.opentracing.tag.Tags
import okhttp3.OkHttpClient
import okhttp3.Request import okhttp3.Request
import spock.lang.Shared import spock.lang.Shared
@ -18,6 +18,9 @@ class AkkaHttpServerInstrumentationTest extends AgentTestRunner {
@Shared @Shared
int syncPort int syncPort
@Shared
def client = OkHttpUtils.client()
def setupSpec() { def setupSpec() {
AkkaHttpTestAsyncWebServer.start() AkkaHttpTestAsyncWebServer.start()
asyncPort = AkkaHttpTestAsyncWebServer.port() asyncPort = AkkaHttpTestAsyncWebServer.port()
@ -32,7 +35,6 @@ class AkkaHttpServerInstrumentationTest extends AgentTestRunner {
def "#server 200 request trace"() { def "#server 200 request trace"() {
setup: setup:
OkHttpClient client = new OkHttpClient.Builder().build()
def request = new Request.Builder() def request = new Request.Builder()
.url("http://localhost:$port/test") .url("http://localhost:$port/test")
.header("x-datadog-trace-id", "123") .header("x-datadog-trace-id", "123")
@ -79,7 +81,6 @@ class AkkaHttpServerInstrumentationTest extends AgentTestRunner {
def "#server exceptions trace for #endpoint"() { def "#server exceptions trace for #endpoint"() {
setup: setup:
OkHttpClient client = new OkHttpClient.Builder().build()
def request = new Request.Builder() def request = new Request.Builder()
.url("http://localhost:$port/$endpoint") .url("http://localhost:$port/$endpoint")
.get() .get()
@ -120,7 +121,6 @@ class AkkaHttpServerInstrumentationTest extends AgentTestRunner {
def "#server 5xx trace"() { def "#server 5xx trace"() {
setup: setup:
OkHttpClient client = new OkHttpClient.Builder().build()
def request = new Request.Builder() def request = new Request.Builder()
.url("http://localhost:$port/server-error") .url("http://localhost:$port/server-error")
.get() .get()
@ -160,7 +160,6 @@ class AkkaHttpServerInstrumentationTest extends AgentTestRunner {
def "#server 4xx trace"() { def "#server 4xx trace"() {
setup: setup:
OkHttpClient client = new OkHttpClient.Builder().build()
def request = new Request.Builder() def request = new Request.Builder()
.url("http://localhost:$port/not-found") .url("http://localhost:$port/not-found")
.get() .get()

View File

@ -1,7 +1,7 @@
import datadog.trace.agent.test.AgentTestRunner import datadog.trace.agent.test.AgentTestRunner
import datadog.trace.agent.test.TestUtils import datadog.trace.agent.test.TestUtils
import datadog.trace.agent.test.utils.OkHttpUtils
import datadog.trace.api.DDSpanTypes import datadog.trace.api.DDSpanTypes
import okhttp3.OkHttpClient
import okhttp3.Request import okhttp3.Request
import play.api.test.TestServer import play.api.test.TestServer
import play.test.Helpers import play.test.Helpers
@ -19,6 +19,9 @@ class Play26Test extends AgentTestRunner {
@Shared @Shared
TestServer testServer TestServer testServer
@Shared
def client = OkHttpUtils.client()
def setupSpec() { def setupSpec() {
port = TestUtils.randomOpenPort() port = TestUtils.randomOpenPort()
testServer = Helpers.testServer(port, Play26TestUtils.buildTestApp()) testServer = Helpers.testServer(port, Play26TestUtils.buildTestApp())
@ -31,7 +34,6 @@ class Play26Test extends AgentTestRunner {
def "request traces"() { def "request traces"() {
setup: setup:
OkHttpClient client = new OkHttpClient.Builder().build()
def request = new Request.Builder() def request = new Request.Builder()
.url("http://localhost:$port/helloplay/spock") .url("http://localhost:$port/helloplay/spock")
.header("x-datadog-trace-id", "123") .header("x-datadog-trace-id", "123")
@ -88,7 +90,6 @@ class Play26Test extends AgentTestRunner {
def "5xx errors trace"() { def "5xx errors trace"() {
setup: setup:
OkHttpClient client = new OkHttpClient.Builder().build()
def request = new Request.Builder() def request = new Request.Builder()
.url("http://localhost:$port/make-error") .url("http://localhost:$port/make-error")
.get() .get()
@ -139,7 +140,6 @@ class Play26Test extends AgentTestRunner {
def "error thrown in request"() { def "error thrown in request"() {
setup: setup:
OkHttpClient client = new OkHttpClient.Builder().build()
def request = new Request.Builder() def request = new Request.Builder()
.url("http://localhost:$port/exception") .url("http://localhost:$port/exception")
.get() .get()
@ -194,7 +194,6 @@ class Play26Test extends AgentTestRunner {
def "4xx errors trace"() { def "4xx errors trace"() {
setup: setup:
OkHttpClient client = new OkHttpClient.Builder().build()
def request = new Request.Builder() def request = new Request.Builder()
.url("http://localhost:$port/nowhere") .url("http://localhost:$port/nowhere")
.get() .get()

View File

@ -1,8 +1,8 @@
import datadog.opentracing.DDSpan import datadog.opentracing.DDSpan
import datadog.trace.agent.test.AgentTestRunner import datadog.trace.agent.test.AgentTestRunner
import datadog.trace.agent.test.TestUtils import datadog.trace.agent.test.TestUtils
import datadog.trace.agent.test.utils.OkHttpUtils
import datadog.trace.api.DDSpanTypes import datadog.trace.api.DDSpanTypes
import okhttp3.OkHttpClient
import okhttp3.Request import okhttp3.Request
import play.api.test.TestServer import play.api.test.TestServer
import play.test.Helpers import play.test.Helpers
@ -14,6 +14,9 @@ class Play24Test extends AgentTestRunner {
@Shared @Shared
TestServer testServer TestServer testServer
@Shared
def client = OkHttpUtils.client()
def setupSpec() { def setupSpec() {
port = TestUtils.randomOpenPort() port = TestUtils.randomOpenPort()
testServer = Helpers.testServer(port, Play24TestUtils.buildTestApp()) testServer = Helpers.testServer(port, Play24TestUtils.buildTestApp())
@ -26,7 +29,6 @@ class Play24Test extends AgentTestRunner {
def "request traces"() { def "request traces"() {
setup: setup:
OkHttpClient client = new OkHttpClient.Builder().build()
def request = new Request.Builder() def request = new Request.Builder()
.url("http://localhost:$port/helloplay/spock") .url("http://localhost:$port/helloplay/spock")
.header("x-datadog-trace-id", "123") .header("x-datadog-trace-id", "123")
@ -63,7 +65,6 @@ class Play24Test extends AgentTestRunner {
def "5xx errors trace"() { def "5xx errors trace"() {
setup: setup:
OkHttpClient client = new OkHttpClient.Builder().build()
def request = new Request.Builder() def request = new Request.Builder()
.url("http://localhost:$port/make-error") .url("http://localhost:$port/make-error")
.get() .get()
@ -91,7 +92,6 @@ class Play24Test extends AgentTestRunner {
def "error thrown in request"() { def "error thrown in request"() {
setup: setup:
OkHttpClient client = new OkHttpClient.Builder().build()
def request = new Request.Builder() def request = new Request.Builder()
.url("http://localhost:$port/exception") .url("http://localhost:$port/exception")
.get() .get()
@ -122,7 +122,6 @@ class Play24Test extends AgentTestRunner {
def "4xx errors trace"() { def "4xx errors trace"() {
setup: setup:
OkHttpClient client = new OkHttpClient.Builder().build()
def request = new Request.Builder() def request = new Request.Builder()
.url("http://localhost:$port/nowhere") .url("http://localhost:$port/nowhere")
.get() .get()

View File

@ -5,7 +5,6 @@ import datadog.trace.agent.test.AgentTestRunner
import datadog.trace.agent.test.utils.OkHttpUtils import datadog.trace.agent.test.utils.OkHttpUtils
import datadog.trace.common.writer.ListWriter import datadog.trace.common.writer.ListWriter
import okhttp3.MultipartBody import okhttp3.MultipartBody
import okhttp3.OkHttpClient
import okhttp3.Request import okhttp3.Request
import spock.lang.Shared import spock.lang.Shared
@ -13,7 +12,6 @@ import static datadog.trace.agent.test.asserts.ListWriterAssert.assertTraces
import static datadog.trace.agent.test.server.http.TestHttpServer.httpServer import static datadog.trace.agent.test.server.http.TestHttpServer.httpServer
/* Don't actually need AgentTestRunner, but it messes up the classloader for AgentTestRunnerTest if this runs first. */ /* Don't actually need AgentTestRunner, but it messes up the classloader for AgentTestRunnerTest if this runs first. */
class ServerTest extends AgentTestRunner { class ServerTest extends AgentTestRunner {
@Shared @Shared
def client = OkHttpUtils.client() def client = OkHttpUtils.client()
@ -233,7 +231,7 @@ class ServerTest extends AgentTestRunner {
def "server redirect"() { def "server redirect"() {
setup: setup:
client = new OkHttpClient().newBuilder().followRedirects(followRedirects).build() client = OkHttpUtils.clientBuilder().followRedirects(followRedirects).build()
def server = httpServer { def server = httpServer {
handlers { handlers {
get("/redirect") { get("/redirect") {