@@ -11,7 +11,7 @@ import org.apache.pekko.actor.ActorSystem
11
11
import org .apache .pekko .pattern .after
12
12
import org .assertj .core .api .Assertions .assertThat
13
13
import org .junit .jupiter .api .Test
14
- import scala .concurrent .{ Await , Future }
14
+ import scala .concurrent .{Await , Future }
15
15
import scala .concurrent .duration .DurationInt
16
16
17
17
class PekkoSchedulerTest {
@@ -26,15 +26,21 @@ class PekkoSchedulerTest {
26
26
val scope = initialSpan.makeCurrent()
27
27
val futureResult = for {
28
28
result1 <- Future {
29
- assertThat(Span .current().getSpanContext().getTraceId()).isEqualTo(initialSpan.getSpanContext().getTraceId())
30
- assertThat(Span .current().getSpanContext().getSpanId()).isEqualTo(initialSpan.getSpanContext().getSpanId())
29
+ assertThat(Span .current().getSpanContext().getTraceId())
30
+ .isEqualTo(initialSpan.getSpanContext().getTraceId())
31
+ assertThat(Span .current().getSpanContext().getSpanId())
32
+ .isEqualTo(initialSpan.getSpanContext().getSpanId())
31
33
1
32
34
}
33
- _ = assertThat(Span .current().getSpanContext().getTraceId()).isEqualTo(initialSpan.getSpanContext().getTraceId())
34
- _ = assertThat(Span .current().getSpanContext().getSpanId()).isEqualTo(initialSpan.getSpanContext().getSpanId())
35
+ _ = assertThat(Span .current().getSpanContext().getTraceId())
36
+ .isEqualTo(initialSpan.getSpanContext().getTraceId())
37
+ _ = assertThat(Span .current().getSpanContext().getSpanId())
38
+ .isEqualTo(initialSpan.getSpanContext().getSpanId())
35
39
result2 <- after(200 .millis, system.scheduler)(Future .successful(2 ))
36
- _ = assertThat(Span .current().getSpanContext().getTraceId()).isEqualTo(initialSpan.getSpanContext().getTraceId())
37
- _ = assertThat(Span .current().getSpanContext().getSpanId()).isEqualTo(initialSpan.getSpanContext().getSpanId())
40
+ _ = assertThat(Span .current().getSpanContext().getTraceId())
41
+ .isEqualTo(initialSpan.getSpanContext().getTraceId())
42
+ _ = assertThat(Span .current().getSpanContext().getSpanId())
43
+ .isEqualTo(initialSpan.getSpanContext().getSpanId())
38
44
} yield result1 + result2
39
45
assertThat(Await .result(futureResult, 5 .seconds)).isEqualTo(3 )
40
46
} finally {
0 commit comments