open-telemetry
/
opentelemetry-java-instrumentation
mirror of https://github.com/open-telemetry/opentelemetry-java-instrumentation.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

More consistent custom trace sorting in tests (#2728)

This commit is contained in:
Trask Stalnaker 2021-04-07 10:29:41 -07:00 committed by GitHub
parent 9926288342
commit 410957f206
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
8 changed files with 36 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
instrumentation/elasticsearch/elasticsearch-transport-5.0/javaagent/src/test/groovy/Elasticsearch5TransportClientTest.groovy
View File

@ -176,14 +176,9 @@ class Elasticsearch5TransportClientTest extends AgentInstrumentationSpecificatio
and:
assertTraces(5) {
sortTraces {
// IndexAction and PutMappingAction run in separate threads and so their order is not always the same
if (traces[2][0].name == "IndexAction") {
def tmp = traces[2]
traces[2] = traces[3]
traces[3] = tmp
}
}
// PutMappingAction and IndexAction run in separate threads so their order can vary
traces.subList(2, 4).sort(orderByRootSpanName("PutMappingAction", "IndexAction"))
trace(0, 1) {
span(0) {
name "CreateIndexAction"

11
instrumentation/elasticsearch/elasticsearch-transport-5.3/javaagent/src/test/groovy/Elasticsearch53TransportClientTest.groovy
View File

@ -181,14 +181,9 @@ class Elasticsearch53TransportClientTest extends AgentInstrumentationSpecificati
and:
assertTraces(5) {
sortTraces {
// IndexAction and PutMappingAction run in separate threads and so their order is not always the same
if (traces[2][0].name == "IndexAction") {
def tmp = traces[2]
traces[2] = traces[3]
traces[3] = tmp
}
}
// PutMappingAction and IndexAction run in separate threads so their order can vary
traces.subList(2, 4).sort(orderByRootSpanName("PutMappingAction", "IndexAction"))
trace(0, 1) {
span(0) {
name "CreateIndexAction"

14
instrumentation/elasticsearch/elasticsearch-transport-6.0/javaagent/src/test/groovy/Elasticsearch6TransportClientTest.groovy
View File

@ -174,14 +174,12 @@ class Elasticsearch6TransportClientTest extends AgentInstrumentationSpecificatio
and:
assertTraces(5) {
sortTraces {
// IndexAction and PutMappingAction run in separate threads and so their order is not always the same
if (traces[2][0].name == "IndexAction") {
def tmp = traces[2]
traces[2] = traces[3]
traces[3] = tmp
}
}
// PutMappingAction and IndexAction run in separate threads so their order can vary
traces.subList(2, 4).sort(orderByRootSpanName(
"PutMappingAction", // elasticsearch < 7
"AutoPutMappingAction", // elasticsearch >= 7
"IndexAction"))
trace(0, 1) {
span(0) {
name "CreateIndexAction"

15
instrumentation/jms-1.1/javaagent/src/jms2Test/groovy/SpringTemplateJms2Test.groovy
View File

@ -119,16 +119,11 @@ class SpringTemplateJms2Test extends AgentInstrumentationSpecification {
expect:
receivedMessage.text == "responded!"
assertTraces(4) {
sortTraces {
def expectedOrder = ["$destinationName receive",
"$destinationName send",
"(temporary) receive",
"(temporary) send"]
// ensure that traces appear in expected order
traces.sort {a,b ->
expectedOrder.indexOf(a[0].name) - expectedOrder.indexOf(b[0].name)
}
}
traces.sort(orderByRootSpanName(
"$destinationName receive",
"$destinationName send",
"(temporary) receive",
"(temporary) send"))
trace(0, 1) {
consumerSpan(it, 0, destinationType, destinationName, msgId.get(), null, "receive")

10
instrumentation/jms-1.1/javaagent/src/test/groovy/SpringListenerJms1Test.groovy
View File

@ -5,6 +5,7 @@
import static Jms1Test.consumerSpan
import static Jms1Test.producerSpan
import static io.opentelemetry.api.trace.SpanKind.CONSUMER
import static io.opentelemetry.api.trace.SpanKind.PRODUCER
import io.opentelemetry.instrumentation.test.AgentInstrumentationSpecification
@ -25,14 +26,7 @@ class SpringListenerJms1Test extends AgentInstrumentationSpecification {
expect:
assertTraces(2) {
sortTraces {
// ensure that traces appear in expected order
if (traces[0][0].kind == PRODUCER) {
def tmp = traces[0]
traces[0] = traces[1]
traces[1] = tmp
}
}
traces.sort(orderByRootSpanKind(CONSUMER, PRODUCER))
trace(0, 1) {
consumerSpan(it, 0, "queue", "SpringListenerJms1", "", null, "receive")

15
instrumentation/jms-1.1/javaagent/src/test/groovy/SpringTemplateJms1Test.groovy
View File

@ -97,16 +97,11 @@ class SpringTemplateJms1Test extends AgentInstrumentationSpecification {
expect:
receivedMessage.text == "responded!"
assertTraces(4) {
sortTraces {
def expectedOrder = ["$destinationName receive",
"$destinationName send",
"(temporary) receive",
"(temporary) send"]
// ensure that traces appear in expected order
traces.sort {a,b ->
expectedOrder.indexOf(a[0].name) - expectedOrder.indexOf(b[0].name)
}
}
traces.sort(orderByRootSpanName(
"$destinationName receive",
"$destinationName send",
"(temporary) receive",
"(temporary) send"))
trace(0, 1) {
consumerSpan(it, 0, destinationType, destinationName, msgId.get(), null, "receive")

15
testing-common/src/main/groovy/io/opentelemetry/instrumentation/test/asserts/InMemoryExporterAssert.groovy
View File

@ -9,15 +9,15 @@ import static TraceAssert.assertTrace
import groovy.transform.stc.ClosureParams
import groovy.transform.stc.SimpleType
import io.opentelemetry.api.trace.SpanKind
import io.opentelemetry.instrumentation.testing.util.TelemetryDataUtil
import io.opentelemetry.sdk.trace.data.SpanData
import java.util.function.Supplier
import org.codehaus.groovy.runtime.powerassert.PowerAssertionError
import org.spockframework.runtime.Condition
import org.spockframework.runtime.ConditionNotSatisfiedError
import org.spockframework.runtime.model.TextPosition
import java.util.function.Supplier
class InMemoryExporterAssert {
private final List<List<SpanData>> traces
private final Supplier<List<SpanData>> spanSupplier
@ -75,9 +75,14 @@ class InMemoryExporterAssert {
assertTrace(spanSupplier, traces[index][0].traceId, expectedSize, spec)
}
// this doesn't provide any functionality, just a self-documenting marker
static void sortTraces(Closure callback) {
callback.call()
static Comparator<List<SpanData>> orderByRootSpanName(String... names) {
def list = Arrays.asList(names)
return Comparator.comparing { item -> list.indexOf(item[0].name) }
}
static Comparator<List<SpanData>> orderByRootSpanKind(SpanKind... spanKinds) {
def list = Arrays.asList(spanKinds)
return Comparator.comparing { item -> list.indexOf(item[0].kind) }
}
void assertTracesAllVerified() {

2
testing-common/src/main/groovy/io/opentelemetry/instrumentation/test/base/HttpClientTest.groovy
View File

@ -209,6 +209,8 @@ abstract class HttpClientTest extends InstrumentationSpecification {
// there should be 2 separate traces since the nested CLIENT span is suppressed
// (and the span context propagation along with it)
assertTraces(2) {
traces.sort(orderByRootSpanKind(CLIENT, SERVER))
trace(0, 1) {
basicClientSpan(it, 0, "parent-client-span")
}