@Test
public void testSubmitBinaryAnnotationInt() {
when(mockServerSpan.getSpan()).thenReturn(mockSpan);
when(mockServerSpanState.getCurrentServerSpan()).thenReturn(mockServerSpan);
when(mockServerSpanState.getEndPoint()).thenReturn(mockEndPoint);
serverTracer.submitBinaryAnnotation(KEY, INT_VALUE);
verify(mockServerSpanState).getCurrentServerSpan();
verify(mockServerSpanState).getEndPoint();
verify(mockAnnotationSubmitter).submitBinaryAnnotation(mockSpan, mockEndPoint, KEY, INT_VALUE);
verifyNoMoreInteractions(mockServerSpanState, mockSpanCollector, mockAnnotationSubmitter);
}
@Test
public void testSubmitAnnotationString() {
when(mockServerSpan.getSpan()).thenReturn(mockSpan);
when(mockServerSpanState.getCurrentServerSpan()).thenReturn(mockServerSpan);
when(mockServerSpanState.getEndPoint()).thenReturn(mockEndPoint);
serverTracer.submitAnnotation(ANNOTATION_NAME);
verify(mockServerSpanState).getCurrentServerSpan();
verify(mockServerSpanState).getEndPoint();
verify(mockAnnotationSubmitter).submitAnnotation(mockSpan, mockEndPoint, ANNOTATION_NAME);
verifyNoMoreInteractions(mockServerSpanState, mockSpanCollector, mockAnnotationSubmitter);
}
@Test
public void testSetServerReceived() {
when(mockServerSpan.getSpan()).thenReturn(mockSpan);
when(mockServerSpanState.getCurrentServerSpan()).thenReturn(mockServerSpan);
when(mockServerSpanState.getEndPoint()).thenReturn(mockEndPoint);
serverTracer.setServerReceived();
verify(mockServerSpanState).getCurrentServerSpan();
verify(mockServerSpanState).getEndPoint();
verify(mockAnnotationSubmitter)
.submitAnnotation(mockSpan, mockEndPoint, zipkinCoreConstants.SERVER_RECV);
verifyNoMoreInteractions(mockServerSpanState, mockSpanCollector, mockAnnotationSubmitter);
}
@Test
public void testSetServerSendInCaseOfThreadDuration() {
when(mockServerSpan.getSpan()).thenReturn(mockSpan);
when(mockServerSpanState.getCurrentServerSpan()).thenReturn(mockServerSpan);
when(mockServerSpanState.getServerSpanThreadDuration()).thenReturn(DURATION_MS);
when(mockServerSpanState.getEndPoint()).thenReturn(mockEndPoint);
serverTracer.setServerSend();
verify(mockServerSpanState).getCurrentServerSpan();
verify(mockServerSpanState, times(2)).getEndPoint();
verify(mockAnnotationSubmitter)
.submitAnnotation(mockSpan, mockEndPoint, zipkinCoreConstants.SERVER_SEND);
verify(mockAnnotationSubmitter)
.submitBinaryAnnotation(
mockSpan, mockEndPoint, BraveAnnotations.THREAD_DURATION, String.valueOf(DURATION_MS));
verify(mockServerSpanState).getServerSpanThreadDuration();
verify(mockSpanCollector).collect(mockSpan);
verify(mockServerSpanState).setCurrentServerSpan(null);
verifyNoMoreInteractions(mockServerSpanState, mockSpanCollector, mockAnnotationSubmitter);
}
