@Test public void noRequestHandled() { SpanImpl result = interceptor.handleResponse(null, responseAdapter); assertThat(result, is(nullValue())); }
@Test public void tagsNull() { when(responseAdapter.getTags()).thenReturn(null); interceptor.handleResponse(span, responseAdapter); verify(span).finish(); verifyNoMoreInteractions(span); verifyZeroInteractions(tracer); }
@Test public void happyPath() { when(responseAdapter.getTags()).thenReturn(Collections.<String, String> singletonMap(Tags.HTTP_STATUS.getKey(), "200")); SpanImpl result = interceptor.handleResponse(span, responseAdapter); assertThat(result, is(span)); verify(span).setTag(Tags.HTTP_STATUS.getKey(), "200"); verify(span).finish(); verifyNoMoreInteractions(span); verifyZeroInteractions(tracer); }
/** * {@inheritDoc} */ @Override public void secondAfterBody(ICoreService coreService, long methodId, long sensorTypeId, Object object, Object[] parameters, Object result, boolean exception, RegisteredSensorConfig rsc) {// NOCHK:8-params // check if in the right(first) invocation if (REF_MARKER.isMarkerSet() && REF_MARKER.matchesFirst()) { // call ended, remove the marker. REF_MARKER.remove(); // extract span from thread local SpanImpl span = spanStack.get(); spanStack.remove(); if (null != span) { // get requestAdapter and handle ResponseAdapter responseAdapter = serverAdapterProvider.getServerResponseAdapter(object, parameters, result, exception, rsc); serverInterceptor.handleResponse(span, responseAdapter); if (LOG.isDebugEnabled()) { LOG.debug("Remote server hook after body span " + span); } AbstractSpan transformedSpan = SpanTransformer.transformSpan(span); transformedSpan.setPlatformIdent(platformManager.getPlatformId()); transformedSpan.setMethodIdent(methodId); transformedSpan.setSensorTypeIdent(sensorTypeId); // add to core service (use span id as prefix) coreService.addDefaultData(transformedSpan); } } }
when(spanImpl.isClient()).thenReturn(false); when(serverInterceptor.handleRequest(requestAdapter)).thenReturn(spanImpl); when(serverInterceptor.handleResponse(spanImpl, responseAdapter)).thenReturn(spanImpl); verify(serverInterceptor).handleResponse(spanImpl, responseAdapter); verify(adapterProvider).getServerRequestAdapter(object, parameters, rsc); verify(adapterProvider).getServerResponseAdapter(object, parameters, result, exception, rsc);
when(spanImpl.isClient()).thenReturn(false); when(serverInterceptor.handleRequest(requestAdapter)).thenReturn(spanImpl); when(serverInterceptor.handleResponse(spanImpl, responseAdapter)).thenReturn(spanImpl); verify(serverInterceptor).handleResponse(spanImpl, responseAdapter); verify(adapterProvider).getServerRequestAdapter(object, parameters, rsc); verify(adapterProvider).getServerResponseAdapter(object, parameters, result, false, rsc);