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

View File

@ -57,7 +57,7 @@ class KafkaStreamsTest extends AgentTestRunner {
void onMessage(ConsumerRecord<String, String> record) {
// ensure consistent ordering of traces
// 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)
records.add(record)
}
@ -82,7 +82,7 @@ class KafkaStreamsTest extends AgentTestRunner {
.mapValues(new ValueMapper<String, String>() {
@Override
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")
return textLine.toLowerCase()
}