fix ut

shalk · shalk · commit 9e5b02dc6611 · 2024-10-15T20:20:27.000+08:00
diff --git a/instrumentation/kafka/kafka-streams-0.11/javaagent/src/test/java/io/opentelemetry/javaagent/instrumentation/kafkastreams/KafkaStreamsDefaultTest.java b/instrumentation/kafka/kafka-streams-0.11/javaagent/src/test/java/io/opentelemetry/javaagent/instrumentation/kafkastreams/KafkaStreamsDefaultTest.java
@@ -336,9 +336,9 @@ public Iterable<String> keys(String carrier) {
                   }
                 });
     SpanContext spanContext = Span.fromContext(context).getSpanContext();
-    List<SpanData> streamTrace = testing.spans();
+    List<List<SpanData>> streamTrace = testing.waitForTraces(3);
     assertThat(streamTrace).hasSize(3);
-    SpanData streamSendSpan = streamTrace.get(2);
+    SpanData streamSendSpan = streamTrace.get(2).get(2);
     assertThat(spanContext.getTraceId()).isEqualTo(streamSendSpan.getTraceId());
     assertThat(spanContext.getSpanId()).isEqualTo(streamSendSpan.getSpanId());
   }

Original file line number	Diff line number	Diff line change
`@@ -336,9 +336,9 @@ public Iterable<String> keys(String carrier) {`
`336`	`336`	`}`
`337`	`337`	`});`
`338`	`338`	`SpanContext spanContext = Span.fromContext(context).getSpanContext();`
`339`		`- List<SpanData> streamTrace = testing.spans();`
	`339`	`+ List<List<SpanData>> streamTrace = testing.waitForTraces(3);`
`340`	`340`	`assertThat(streamTrace).hasSize(3);`
`341`		`- SpanData streamSendSpan = streamTrace.get(2);`
	`341`	`+ SpanData streamSendSpan = streamTrace.get(2).get(2);`
`342`	`342`	`assertThat(spanContext.getTraceId()).isEqualTo(streamSendSpan.getTraceId());`
`343`	`343`	`assertThat(spanContext.getSpanId()).isEqualTo(streamSendSpan.getSpanId());`
`344`	`344`	`}`