11package io .opentracing .contrib .agent ;
22
3- import static org .junit . Assert .* ;
3+ import static org .assertj . core . api . Assertions . assertThat ;
44
5+ import org .junit .Before ;
56import org .junit .Test ;
67
8+ import io .opentracing .mock .MockTracer ;
9+ import io .opentracing .util .GlobalTracer ;
10+
711public class TraceAnnotationsManagerTest {
812
13+ protected MockTracer tracer = new MockTracer ();
14+
15+ @ Before
16+ public void beforeTest () throws Exception {
17+ GlobalTracer .register (tracer );
18+ }
19+
920 @ Test
1021 public void test () {
22+ //Test single span in new trace
23+ SayTracedHello .sayHello ();
24+
25+ assertThat (tracer .finishedSpans ().size ()).isEqualTo (1 );
26+ assertThat (tracer .finishedSpans ().get (0 ).operationName ()).isEqualTo ("SAY_HELLO" );
27+ assertThat (tracer .finishedSpans ().get (0 ).tags ().get ("service-name" )).isEqualTo ("test" );
28+
29+ tracer .reset ();
30+
31+ //Test new trace with 2 children spans
32+ SayTracedHello .sayHELLOsayHA ();
33+ assertThat (tracer .finishedSpans ().size ()).isEqualTo (3 );
34+ assertThat (tracer .finishedSpans ().get (0 ).operationName ()).isEqualTo ("SAY_HELLO" );
35+ assertThat (tracer .finishedSpans ().get (0 ).tags ().get ("service-name" )).isEqualTo ("test" );
36+
37+ long traceId = tracer .finishedSpans ().get (0 ).context ().traceId ();
38+ long parentId = tracer .finishedSpans ().get (0 ).parentId ();
39+
40+ assertThat (tracer .finishedSpans ().get (1 ).operationName ()).isEqualTo ("SAY_HA" );
41+ assertThat (tracer .finishedSpans ().get (1 ).parentId ()).isEqualTo (parentId );
42+ assertThat (tracer .finishedSpans ().get (1 ).context ().traceId ()).isEqualTo (traceId );
43+ assertThat (tracer .finishedSpans ().get (1 ).tags ().get ("span-type" )).isEqualTo ("DB" );
44+ assertThat (tracer .finishedSpans ().get (1 ).tags ().get ("service-name" )).isEqualTo ("test" );
45+
46+ assertThat (tracer .finishedSpans ().get (2 ).operationName ()).isEqualTo ("NEW_TRACE" );
47+ assertThat (tracer .finishedSpans ().get (2 ).parentId ()).isEqualTo (0 );//ROOT / no parent
48+ assertThat (tracer .finishedSpans ().get (2 ).context ().traceId ()).isEqualTo (traceId );
49+ assertThat (tracer .finishedSpans ().get (2 ).tags ().get ("service-name" )).isEqualTo ("test2" );
50+
1151
12- System .out .println ("hello" );
52+ System .out .println (tracer .finishedSpans ());
53+ tracer .reset ();
1354 }
1455
15- }
56+ }
0 commit comments