diff --git a/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncTableScan.java b/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncTableScan.java index 7cdb58c12604..30e3a95a7c1f 100644 --- a/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncTableScan.java +++ b/hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestAsyncTableScan.java @@ -124,17 +124,17 @@ protected void assertTraceContinuity() { .max((a, b) -> Long.compare(a.getEndEpochNanos(), b.getEndEpochNanos())) .map(SpanData::getSpanId).get(); - final Matcher onNextMatcher = hasName("TracedScanResultConsumer#onNext"); + final Matcher onNextMatcher = + allOf(hasName("TracedScanResultConsumer#onNext"), hasParentSpanId(scanOperationSpanId)); waitForSpan(onNextMatcher); - spanStream().filter(onNextMatcher::matches) - .forEach(span -> assertThat(span, allOf(onNextMatcher, hasParentSpanId(scanOperationSpanId), - hasStatusWithCode(StatusCode.OK), hasEnded()))); + spanStream().filter(onNextMatcher::matches).forEach( + span -> assertThat(span, allOf(onNextMatcher, hasStatusWithCode(StatusCode.OK), hasEnded()))); - final Matcher onCompleteMatcher = hasName("TracedScanResultConsumer#onComplete"); + final Matcher onCompleteMatcher = + allOf(hasName("TracedScanResultConsumer#onComplete"), hasParentSpanId(scanOperationSpanId)); waitForSpan(onCompleteMatcher); - spanStream().filter(onCompleteMatcher::matches) - .forEach(span -> assertThat(span, allOf(onCompleteMatcher, - hasParentSpanId(scanOperationSpanId), hasStatusWithCode(StatusCode.OK), hasEnded()))); + spanStream().filter(onCompleteMatcher::matches).forEach(span -> assertThat(span, + allOf(onCompleteMatcher, hasStatusWithCode(StatusCode.OK), hasEnded()))); } @Override @@ -163,10 +163,10 @@ protected void assertTraceContinuity() { .max((a, b) -> Long.compare(a.getEndEpochNanos(), b.getEndEpochNanos())) .map(SpanData::getSpanId).get(); - final Matcher onErrorMatcher = hasName("TracedScanResultConsumer#onError"); + final Matcher onErrorMatcher = + allOf(hasName("TracedScanResultConsumer#onError"), hasParentSpanId(scanOperationSpanId)); waitForSpan(onErrorMatcher); - spanStream().filter(onErrorMatcher::matches) - .forEach(span -> assertThat(span, allOf(onErrorMatcher, hasParentSpanId(scanOperationSpanId), - hasStatusWithCode(StatusCode.OK), hasEnded()))); + spanStream().filter(onErrorMatcher::matches).forEach(span -> assertThat(span, + allOf(onErrorMatcher, hasStatusWithCode(StatusCode.OK), hasEnded()))); } }