72
72
import io .opencensus .trace .MessageEvent ;
73
73
import io .opencensus .trace .MessageEvent .Type ;
74
74
import io .opencensus .trace .Span ;
75
- import io .opencensus .trace .Span .Kind ;
76
75
import io .opencensus .trace .SpanBuilder ;
77
76
import io .opencensus .trace .SpanContext ;
78
77
import io .opencensus .trace .Tracer ;
@@ -295,14 +294,12 @@ public <ReqT, RespT> ClientCall<ReqT, RespT> interceptCall(
295
294
296
295
if (nonDefaultContext ) {
297
296
verify (tracer ).spanBuilderWithExplicitParent (
298
- eq ("package1.service2/ method3" ), same (fakeClientParentSpan ));
297
+ eq ("Sent. package1.service2. method3" ), same (fakeClientParentSpan ));
299
298
verify (spyClientSpanBuilder ).setRecordEvents (eq (true ));
300
- verify (spyClientSpanBuilder ).setSpanKind (eq (Kind .CLIENT ));
301
299
} else {
302
300
verify (tracer ).spanBuilderWithExplicitParent (
303
- eq ("package1.service2/ method3" ), isNull (Span .class ));
301
+ eq ("Sent. package1.service2. method3" ), isNull (Span .class ));
304
302
verify (spyClientSpanBuilder ).setRecordEvents (eq (true ));
305
- verify (spyClientSpanBuilder ).setSpanKind (eq (Kind .CLIENT ));
306
303
}
307
304
verify (spyClientSpan , never ()).end (any (EndSpanOptions .class ));
308
305
@@ -500,7 +497,7 @@ public void clientBasicTracingDefaultSpan() {
500
497
ClientStreamTracer clientStreamTracer =
501
498
callTracer .newClientStreamTracer (CallOptions .DEFAULT , headers );
502
499
verify (tracer ).spanBuilderWithExplicitParent (
503
- eq ("package1.service2/ method3" ), isNull (Span .class ));
500
+ eq ("Sent. package1.service2. method3" ), isNull (Span .class ));
504
501
verify (spyClientSpan , never ()).end (any (EndSpanOptions .class ));
505
502
506
503
clientStreamTracer .outboundMessage (0 );
@@ -607,9 +604,8 @@ public void clientStreamNeverCreatedStillRecordTracing() {
607
604
CensusTracingModule .ClientCallTracer callTracer =
608
605
censusTracing .newClientCallTracer (fakeClientParentSpan , method );
609
606
verify (tracer ).spanBuilderWithExplicitParent (
610
- eq ("package1.service2/ method3" ), same (fakeClientParentSpan ));
607
+ eq ("Sent. package1.service2. method3" ), same (fakeClientParentSpan ));
611
608
verify (spyClientSpanBuilder ).setRecordEvents (eq (true ));
612
- verify (spyClientSpanBuilder ).setSpanKind (eq (Kind .CLIENT ));
613
609
614
610
callTracer .callEnded (Status .DEADLINE_EXCEEDED .withDescription ("3 seconds" ));
615
611
verify (spyClientSpan ).end (
@@ -784,9 +780,8 @@ public void traceHeadersPropagateSpanContext() throws Exception {
784
780
verify (mockTracingPropagationHandler ).toByteArray (same (fakeClientSpanContext ));
785
781
verifyNoMoreInteractions (mockTracingPropagationHandler );
786
782
verify (tracer ).spanBuilderWithExplicitParent (
787
- eq ("package1.service2/ method3" ), same (fakeClientParentSpan ));
783
+ eq ("Sent. package1.service2. method3" ), same (fakeClientParentSpan ));
788
784
verify (spyClientSpanBuilder ).setRecordEvents (eq (true ));
789
- verify (spyClientSpanBuilder ).setSpanKind (eq (Kind .CLIENT ));
790
785
verifyNoMoreInteractions (tracer );
791
786
assertTrue (headers .containsKey (censusTracing .tracingHeader ));
792
787
@@ -795,9 +790,8 @@ public void traceHeadersPropagateSpanContext() throws Exception {
795
790
method .getFullMethodName (), headers );
796
791
verify (mockTracingPropagationHandler ).fromByteArray (same (binarySpanContext ));
797
792
verify (tracer ).spanBuilderWithRemoteParent (
798
- eq ("package1.service2/ method3" ), same (spyClientSpan .getContext ()));
793
+ eq ("Recv. package1.service2. method3" ), same (spyClientSpan .getContext ()));
799
794
verify (spyServerSpanBuilder ).setRecordEvents (eq (true ));
800
- verify (spyServerSpanBuilder ).setSpanKind (eq (Kind .SERVER ));
801
795
802
796
Context filteredContext = serverTracer .filterContext (Context .ROOT );
803
797
assertSame (spyServerSpan , ContextUtils .CONTEXT_SPAN_KEY .get (filteredContext ));
@@ -867,10 +861,8 @@ public void traceHeaderMalformed() throws Exception {
867
861
censusTracing .getServerTracerFactory ().newServerStreamTracer (
868
862
method .getFullMethodName (), headers );
869
863
verify (tracer ).spanBuilderWithRemoteParent (
870
- eq ("package1.service2/ method3" ), isNull (SpanContext .class ));
864
+ eq ("Recv. package1.service2. method3" ), isNull (SpanContext .class ));
871
865
verify (spyServerSpanBuilder ).setRecordEvents (eq (true ));
872
- verify (spyServerSpanBuilder ).setSpanKind (eq (Kind .SERVER ));
873
-
874
866
}
875
867
876
868
@ Test
@@ -1021,9 +1013,8 @@ public void serverBasicTracingNoHeaders() {
1021
1013
tracerFactory .newServerStreamTracer (method .getFullMethodName (), new Metadata ());
1022
1014
verifyZeroInteractions (mockTracingPropagationHandler );
1023
1015
verify (tracer ).spanBuilderWithRemoteParent (
1024
- eq ("package1.service2/ method3" ), isNull (SpanContext .class ));
1016
+ eq ("Recv. package1.service2. method3" ), isNull (SpanContext .class ));
1025
1017
verify (spyServerSpanBuilder ).setRecordEvents (eq (true ));
1026
- verify (spyServerSpanBuilder ).setSpanKind (eq (Kind .SERVER ));
1027
1018
1028
1019
Context filteredContext = serverStreamTracer .filterContext (Context .ROOT );
1029
1020
assertSame (spyServerSpan , ContextUtils .CONTEXT_SPAN_KEY .get (filteredContext ));
@@ -1119,6 +1110,15 @@ public void convertToTracingStatus() {
1119
1110
}
1120
1111
}
1121
1112
1113
+
1114
+ @ Test
1115
+ public void generateTraceSpanName () {
1116
+ assertEquals (
1117
+ "Sent.io.grpc.Foo" , CensusTracingModule .generateTraceSpanName (false , "io.grpc/Foo" ));
1118
+ assertEquals (
1119
+ "Recv.io.grpc.Bar" , CensusTracingModule .generateTraceSpanName (true , "io.grpc/Bar" ));
1120
+ }
1121
+
1122
1122
private static void assertNoServerContent (StatsTestUtils .MetricsRecord record ) {
1123
1123
assertNull (record .getMetric (DeprecatedCensusConstants .RPC_SERVER_ERROR_COUNT ));
1124
1124
assertNull (record .getMetric (DeprecatedCensusConstants .RPC_SERVER_REQUEST_COUNT ));
0 commit comments