Examples with AbstractTracingSpan org.apache.skywalking.apm.agent.core.context.trace.AbstractTracingSpan used on opensource projects

Example 56 with AbstractTracingSpan

use of org.apache.skywalking.apm.agent.core.context.trace.AbstractTracingSpan in project incubator-skywalking by apache.

the class ResinV3InterceptorTest method testWithSerializedContextData.

@Test
public void testWithSerializedContextData() throws Throwable {
    when(request.getHeader(SW3CarrierItem.HEADER_NAME)).thenReturn("1.333.2345|3|1|1|#192.168.1.8:18002|#/portal/|#/testEntrySpan|#AQA*#AQA*Et0We0tQNQA*");
    interceptor.beforeMethod(enhancedInstance, null, arguments, argumentType, methodInterceptResult);
    interceptor.afterMethod(enhancedInstance, null, arguments, argumentType, null);
    assertThat(segmentStorage.getTraceSegments().size(), is(1));
    TraceSegment traceSegment = segmentStorage.getTraceSegments().get(0);
    List<AbstractTracingSpan> spans = SegmentHelper.getSpans(traceSegment);
    assertHttpSpan(spans.get(0));
    assertTraceSegmentRef(traceSegment.getRefs().get(0));
}

Example 57 with AbstractTracingSpan

use of org.apache.skywalking.apm.agent.core.context.trace.AbstractTracingSpan in project incubator-skywalking by apache.

the class ResinV3InterceptorTest method testWithOccurException.

@Test
public void testWithOccurException() throws Throwable {
    interceptor.beforeMethod(enhancedInstance, null, arguments, argumentType, methodInterceptResult);
    interceptor.handleMethodException(enhancedInstance, null, arguments, argumentType, new RuntimeException());
    interceptor.afterMethod(enhancedInstance, null, arguments, argumentType, null);
    assertThat(segmentStorage.getTraceSegments().size(), is(1));
    TraceSegment traceSegment = segmentStorage.getTraceSegments().get(0);
    List<AbstractTracingSpan> spans = SegmentHelper.getSpans(traceSegment);
    assertHttpSpan(spans.get(0));
    List<LogDataEntity> logDataEntities = SpanHelper.getLogs(spans.get(0));
    assertThat(logDataEntities.size(), is(1));
    SpanAssert.assertException(logDataEntities.get(0), RuntimeException.class);
}

Example 58 with AbstractTracingSpan

use of org.apache.skywalking.apm.agent.core.context.trace.AbstractTracingSpan in project incubator-skywalking by apache.

the class OnSuccessInterceptorTest method testOnSuccessWithErrorStatus.

@Test
public void testOnSuccessWithErrorStatus() throws Throwable {
    when(sendResult.getSendStatus()).thenReturn(SendStatus.FLUSH_SLAVE_TIMEOUT);
    successInterceptor.beforeMethod(enhancedInstance, null, new Object[] { sendResult }, null, null);
    successInterceptor.afterMethod(enhancedInstance, null, new Object[] { sendResult }, null, null);
    assertThat(segmentStorage.getTraceSegments().size(), is(1));
    TraceSegment traceSegment = segmentStorage.getTraceSegments().get(0);
    List<AbstractTracingSpan> spans = SegmentHelper.getSpans(traceSegment);
    assertThat(spans.size(), is(1));
    AbstractTracingSpan successSpan = spans.get(0);
    SpanAssert.assertComponent(successSpan, ComponentsDefine.ROCKET_MQ);
    SpanAssert.assertOccurException(successSpan, true);
}

Example 59 with AbstractTracingSpan

use of org.apache.skywalking.apm.agent.core.context.trace.AbstractTracingSpan in project incubator-skywalking by apache.

the class InterceptorTest method assertAsyncExecute.

@Test
public void assertAsyncExecute() throws Throwable {
    executeInterceptor.beforeMethod(null, null, allArguments, null, null);
    asyncExecuteInterceptor.beforeMethod(null, null, null, null, null);
    final Map<String, Object> dataMap = ExecutorDataMap.getDataMap();
    ES.submit(new Runnable() {

        @Override
        public void run() {
            ExecutorDataMap.setDataMap(dataMap);
            sendEvent("ds_1", "select * from t_order_1");
        }
    }).get();
    asyncExecuteInterceptor.afterMethod(null, null, null, null, null);
    sendEvent("ds_0", "select * from t_order_0");
    executeInterceptor.afterMethod(null, null, allArguments, null, null);
    assertThat(segmentStorage.getTraceSegments().size(), is(2));
    TraceSegment segment0 = segmentStorage.getTraceSegments().get(0);
    TraceSegment segment1 = segmentStorage.getTraceSegments().get(1);
    assertThat(segment0.getRefs().size(), is(1));
    assertNull(segment1.getRefs());
    List<AbstractTracingSpan> spans0 = SegmentHelper.getSpans(segment0);
    assertNotNull(spans0);
    assertThat(spans0.size(), is(1));
    assertSpan(spans0.get(0), 1);
    List<AbstractTracingSpan> spans1 = SegmentHelper.getSpans(segment1);
    assertNotNull(spans1);
    assertThat(spans1.size(), is(2));
    assertSpan(spans1.get(0), 0);
    assertThat(spans1.get(1).getOperationName(), is("/SJDBC/TRUNK/DQL"));
}

Example 60 with AbstractTracingSpan

use of org.apache.skywalking.apm.agent.core.context.trace.AbstractTracingSpan in project incubator-skywalking by apache.

the class InterceptorTest method assertExecuteError.

@Test
public void assertExecuteError() throws Throwable {
    executeInterceptor.beforeMethod(null, null, allArguments, null, null);
    asyncExecuteInterceptor.beforeMethod(null, null, null, null, null);
    final Map<String, Object> dataMap = ExecutorDataMap.getDataMap();
    ES.submit(new Runnable() {

        @Override
        public void run() {
            ExecutorDataMap.setDataMap(dataMap);
            sendError();
        }
    }).get();
    asyncExecuteInterceptor.handleMethodException(null, null, null, null, new SQLException("test"));
    asyncExecuteInterceptor.afterMethod(null, null, null, null, null);
    sendEvent("ds_0", "select * from t_order_0");
    executeInterceptor.handleMethodException(null, null, allArguments, null, new SQLException("Test"));
    executeInterceptor.afterMethod(null, null, allArguments, null, null);
    assertThat(segmentStorage.getTraceSegments().size(), is(2));
    TraceSegment segment0 = segmentStorage.getTraceSegments().get(0);
    TraceSegment segment1 = segmentStorage.getTraceSegments().get(1);
    List<AbstractTracingSpan> spans0 = SegmentHelper.getSpans(segment0);
    assertNotNull(spans0);
    assertThat(spans0.size(), is(1));
    assertErrorSpan(spans0.get(0));
    List<AbstractTracingSpan> spans1 = SegmentHelper.getSpans(segment1);
    assertNotNull(spans1);
    assertThat(spans1.size(), is(2));
    assertSpan(spans1.get(0), 0);
    assertErrorSpan(spans1.get(1));
}