Switch to new Tags class in testing module
This commit is contained in:
parent
a45dc34c5c
commit
da80be693c
|
@ -2,9 +2,9 @@ package datadog.trace.agent.test.asserts
|
||||||
|
|
||||||
import datadog.opentracing.DDSpan
|
import datadog.opentracing.DDSpan
|
||||||
import datadog.trace.api.Config
|
import datadog.trace.api.Config
|
||||||
|
import datadog.trace.instrumentation.api.Tags
|
||||||
import groovy.transform.stc.ClosureParams
|
import groovy.transform.stc.ClosureParams
|
||||||
import groovy.transform.stc.SimpleType
|
import groovy.transform.stc.SimpleType
|
||||||
import io.opentracing.tag.Tags
|
|
||||||
|
|
||||||
import java.util.regex.Pattern
|
import java.util.regex.Pattern
|
||||||
|
|
||||||
|
@ -50,7 +50,7 @@ class TagsAssert {
|
||||||
assert tags[Config.RUNTIME_ID_TAG] == null
|
assert tags[Config.RUNTIME_ID_TAG] == null
|
||||||
}
|
}
|
||||||
|
|
||||||
boolean isServer = (tags[Tags.SPAN_KIND.key] == Tags.SPAN_KIND_SERVER)
|
boolean isServer = (tags[Tags.SPAN_KIND] == Tags.SPAN_KIND_SERVER)
|
||||||
if (isRoot || distributedRootSpan || isServer) {
|
if (isRoot || distributedRootSpan || isServer) {
|
||||||
assert tags[Config.LANGUAGE_TAG_KEY] == Config.LANGUAGE_TAG_VALUE
|
assert tags[Config.LANGUAGE_TAG_KEY] == Config.LANGUAGE_TAG_VALUE
|
||||||
} else {
|
} else {
|
||||||
|
|
|
@ -7,7 +7,7 @@ import datadog.trace.agent.test.asserts.TraceAssert
|
||||||
import datadog.trace.api.Config
|
import datadog.trace.api.Config
|
||||||
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 datadog.trace.instrumentation.api.Tags
|
||||||
import spock.lang.AutoCleanup
|
import spock.lang.AutoCleanup
|
||||||
import spock.lang.Shared
|
import spock.lang.Shared
|
||||||
import spock.lang.Unroll
|
import spock.lang.Unroll
|
||||||
|
@ -324,20 +324,20 @@ abstract class HttpClientTest<DECORATOR extends HttpClientDecorator> extends Age
|
||||||
if (exception) {
|
if (exception) {
|
||||||
errorTags(exception.class, exception.message)
|
errorTags(exception.class, exception.message)
|
||||||
}
|
}
|
||||||
"$Tags.COMPONENT.key" clientDecorator.component()
|
"$Tags.COMPONENT" clientDecorator.component()
|
||||||
if (status) {
|
if (status) {
|
||||||
"$Tags.HTTP_STATUS.key" status
|
"$Tags.HTTP_STATUS" status
|
||||||
}
|
}
|
||||||
"$Tags.HTTP_URL.key" "${uri.resolve(uri.path)}"
|
"$Tags.HTTP_URL" "${uri.resolve(uri.path)}"
|
||||||
if (tagQueryString) {
|
if (tagQueryString) {
|
||||||
"$DDTags.HTTP_QUERY" uri.query
|
"$DDTags.HTTP_QUERY" uri.query
|
||||||
"$DDTags.HTTP_FRAGMENT" { it == null || it == uri.fragment } // Optional
|
"$DDTags.HTTP_FRAGMENT" { it == null || it == uri.fragment } // Optional
|
||||||
}
|
}
|
||||||
"$Tags.PEER_HOSTNAME.key" "localhost"
|
"$Tags.PEER_HOSTNAME" "localhost"
|
||||||
"$Tags.PEER_PORT.key" uri.port
|
"$Tags.PEER_PORT" uri.port
|
||||||
"$Tags.PEER_HOST_IPV4.key" { it == null || it == "127.0.0.1" } // Optional
|
"$Tags.PEER_HOST_IPV4" { it == null || it == "127.0.0.1" } // Optional
|
||||||
"$Tags.HTTP_METHOD.key" method
|
"$Tags.HTTP_METHOD" method
|
||||||
"$Tags.SPAN_KIND.key" Tags.SPAN_KIND_CLIENT
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_CLIENT
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -8,9 +8,9 @@ import datadog.trace.agent.test.asserts.TraceAssert
|
||||||
import datadog.trace.agent.test.utils.OkHttpUtils
|
import datadog.trace.agent.test.utils.OkHttpUtils
|
||||||
import datadog.trace.agent.test.utils.PortUtils
|
import datadog.trace.agent.test.utils.PortUtils
|
||||||
import datadog.trace.api.DDSpanTypes
|
import datadog.trace.api.DDSpanTypes
|
||||||
|
import datadog.trace.instrumentation.api.Tags
|
||||||
import groovy.transform.stc.ClosureParams
|
import groovy.transform.stc.ClosureParams
|
||||||
import groovy.transform.stc.SimpleType
|
import groovy.transform.stc.SimpleType
|
||||||
import io.opentracing.tag.Tags
|
|
||||||
import okhttp3.HttpUrl
|
import okhttp3.HttpUrl
|
||||||
import okhttp3.OkHttpClient
|
import okhttp3.OkHttpClient
|
||||||
import okhttp3.Request
|
import okhttp3.Request
|
||||||
|
@ -440,21 +440,21 @@ abstract class HttpServerTest<SERVER, DECORATOR extends HttpServerDecorator> ext
|
||||||
}
|
}
|
||||||
tags {
|
tags {
|
||||||
defaultTags(true)
|
defaultTags(true)
|
||||||
"$Tags.COMPONENT.key" serverDecorator.component()
|
"$Tags.COMPONENT" serverDecorator.component()
|
||||||
if (endpoint.errored) {
|
if (endpoint.errored) {
|
||||||
"$Tags.ERROR.key" endpoint.errored
|
"$Tags.ERROR" endpoint.errored
|
||||||
}
|
}
|
||||||
"$Tags.HTTP_STATUS.key" endpoint.status
|
"$Tags.HTTP_STATUS" endpoint.status
|
||||||
"$Tags.HTTP_URL.key" "${endpoint.resolve(address)}"
|
"$Tags.HTTP_URL" "${endpoint.resolve(address)}"
|
||||||
// if (tagQueryString) {
|
// if (tagQueryString) {
|
||||||
// "$DDTags.HTTP_QUERY" uri.query
|
// "$DDTags.HTTP_QUERY" uri.query
|
||||||
// "$DDTags.HTTP_FRAGMENT" { it == null || it == uri.fragment } // Optional
|
// "$DDTags.HTTP_FRAGMENT" { it == null || it == uri.fragment } // Optional
|
||||||
// }
|
// }
|
||||||
"$Tags.PEER_HOSTNAME.key" { it == "localhost" || it == "127.0.0.1" }
|
"$Tags.PEER_HOSTNAME" { it == "localhost" || it == "127.0.0.1" }
|
||||||
"$Tags.PEER_PORT.key" Integer
|
"$Tags.PEER_PORT" Integer
|
||||||
"$Tags.PEER_HOST_IPV4.key" { it == null || it == "127.0.0.1" } // Optional
|
"$Tags.PEER_HOST_IPV4" { it == null || it == "127.0.0.1" } // Optional
|
||||||
"$Tags.HTTP_METHOD.key" method
|
"$Tags.HTTP_METHOD" method
|
||||||
"$Tags.SPAN_KIND.key" Tags.SPAN_KIND_SERVER
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_SERVER
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -3,7 +3,7 @@ package datadog.trace.agent.test.server.http
|
||||||
import datadog.opentracing.DDSpan
|
import datadog.opentracing.DDSpan
|
||||||
import datadog.trace.agent.test.asserts.ListWriterAssert
|
import datadog.trace.agent.test.asserts.ListWriterAssert
|
||||||
import datadog.trace.instrumentation.api.AgentSpan
|
import datadog.trace.instrumentation.api.AgentSpan
|
||||||
import io.opentracing.tag.Tags
|
import datadog.trace.instrumentation.api.Tags
|
||||||
import org.eclipse.jetty.http.HttpMethods
|
import org.eclipse.jetty.http.HttpMethods
|
||||||
import org.eclipse.jetty.server.Handler
|
import org.eclipse.jetty.server.Handler
|
||||||
import org.eclipse.jetty.server.Request
|
import org.eclipse.jetty.server.Request
|
||||||
|
@ -104,7 +104,7 @@ class TestHttpServer implements AutoCloseable {
|
||||||
childOf(parentSpan)
|
childOf(parentSpan)
|
||||||
}
|
}
|
||||||
tags {
|
tags {
|
||||||
"$Tags.SPAN_KIND.key" Tags.SPAN_KIND_SERVER
|
"$Tags.SPAN_KIND" Tags.SPAN_KIND_SERVER
|
||||||
defaultTags(parentSpan != null)
|
defaultTags(parentSpan != null)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -235,10 +235,10 @@ class TestHttpServer implements AutoCloseable {
|
||||||
final AgentSpan.Context extractedContext = propagate().extract(req, GETTER)
|
final AgentSpan.Context extractedContext = propagate().extract(req, GETTER)
|
||||||
if (extractedContext != null) {
|
if (extractedContext != null) {
|
||||||
startSpan("test-http-server", extractedContext)
|
startSpan("test-http-server", extractedContext)
|
||||||
.setTag(Tags.SPAN_KIND.key, Tags.SPAN_KIND_SERVER).finish()
|
.setTag(Tags.SPAN_KIND, Tags.SPAN_KIND_SERVER).finish()
|
||||||
} else {
|
} else {
|
||||||
startSpan("test-http-server")
|
startSpan("test-http-server")
|
||||||
.setTag(Tags.SPAN_KIND.key, Tags.SPAN_KIND_SERVER).finish()
|
.setTag(Tags.SPAN_KIND, Tags.SPAN_KIND_SERVER).finish()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue