Switch to new Tags class in testing module

This commit is contained in:
Trask Stalnaker 2019-10-04 15:06:18 -07:00
parent a45dc34c5c
commit da80be693c
4 changed files with 25 additions and 25 deletions

View File

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

View File

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

View File

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

View File

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