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

Wait for traces in kafka streams tests

This commit is contained in:
Nikolay Martynov 2019-07-11 10:51:55 -04:00
parent 765b76b0ae
commit 3ecc7c123d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dd-java-agent/instrumentation/kafka-streams-0.11/src/test/groovy/KafkaStreamsTest.groovy
View File

@ -57,7 +57,7 @@ class KafkaStreamsTest extends AgentTestRunner {
void onMessage(ConsumerRecord<String, String> record) { void onMessage(ConsumerRecord<String, String> record) {
// ensure consistent ordering of traces // ensure consistent ordering of traces
// this is the last processing step so we should see 2 traces here // this is the last processing step so we should see 2 traces here
TEST_WRITER.waitForTraces(2) TEST_WRITER.waitForTraces(3)
getTestTracer().activeSpan().setTag("testing", 123) getTestTracer().activeSpan().setTag("testing", 123)
records.add(record) records.add(record)
} }
@ -82,7 +82,7 @@ class KafkaStreamsTest extends AgentTestRunner {
.mapValues(new ValueMapper<String, String>() { .mapValues(new ValueMapper<String, String>() {
@Override @Override
String apply(String textLine) { String apply(String textLine) {
TEST_WRITER.waitForTraces(1) // ensure consistent ordering of traces TEST_WRITER.waitForTraces(2) // ensure consistent ordering of traces
getTestTracer().activeSpan().setTag("asdf", "testing") getTestTracer().activeSpan().setTag("asdf", "testing")
return textLine.toLowerCase() return textLine.toLowerCase()
} }