Use OkHttpUtils in more places
This commit is contained in:
parent
655d490da2
commit
5d627733d8
|
@ -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()
|
||||||
|
|
|
@ -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()
|
||||||
|
|
|
@ -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()
|
||||||
|
|
|
@ -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") {
|
||||||
|
|
Loading…
Reference in New Issue