Examples with Segment com.amazonaws.xray.entities.Segment used on opensource projects

Example 16 with Segment

use of com.amazonaws.xray.entities.Segment in project aws-xray-sdk-java by aws.

the class AWSXRayRecorderTest method testNotSendingUnsampledSegment.

@Test
public void testNotSendingUnsampledSegment() {
    Emitter mockEmitter = Mockito.mock(UDPEmitter.class);
    AWSXRayRecorder recorder = AWSXRayRecorderBuilder.standard().withEmitter(mockEmitter).build();
    Segment segment = recorder.beginSegment("test");
    segment.setSampled(false);
    recorder.endSegment();
    Mockito.verify(mockEmitter, Mockito.times(0)).sendSegment(Mockito.any());
}

Example 17 with Segment

use of com.amazonaws.xray.entities.Segment in project aws-xray-sdk-java by aws.

the class AWSXRayRecorderTest method testInjectThreadLocalInjectsCurrentSegment.

@Test
public void testInjectThreadLocalInjectsCurrentSegment() {
    Segment segment = AWSXRay.beginSegment("test");
    Thread thread = new Thread() {

        public void run() {
            AWSXRay.injectThreadLocal(segment);
            Assert.assertEquals(segment, AWSXRay.getThreadLocal());
        }
    };
    thread.start();
    try {
        thread.join();
    } catch (InterruptedException ie) {
    }
    AWSXRay.endSegment();
}

Example 18 with Segment

use of com.amazonaws.xray.entities.Segment in project aws-xray-sdk-java by aws.

the class AWSXRayRecorderTest method testGetTraceEntityReturnsCurrentSegment.

@Test
public void testGetTraceEntityReturnsCurrentSegment() {
    Segment segment = AWSXRay.beginSegment("test");
    Assert.assertEquals(segment, AWSXRay.getTraceEntity());
    AWSXRay.endSegment();
}

Example 19 with Segment

use of com.amazonaws.xray.entities.Segment in project aws-xray-sdk-java by aws.

the class AWSXRayRecorderTest method testGetThreadLocalReturnsCurrentSegment.

@Test
public void testGetThreadLocalReturnsCurrentSegment() {
    Segment segment = AWSXRay.beginSegment("test");
    Assert.assertEquals(segment, AWSXRay.getThreadLocal());
    AWSXRay.endSegment();
}

Example 20 with Segment

use of com.amazonaws.xray.entities.Segment in project aws-xray-sdk-java by aws.

the class EntityTest method testInProgressSegment.

@Test
public void testInProgressSegment() throws JSONException {
    String segmentId = Entity.generateId();
    TraceID traceId = new TraceID();
    Segment segment = new SegmentImpl(AWSXRay.getGlobalRecorder(), "test", traceId);
    segment.setId(segmentId);
    segment.setStartTime(1.0);
    String expected = expectedInProgressSegment(traceId, segmentId, segment.getStartTime()).toString();
    JSONAssert.assertEquals(expected, segment.serialize(), JSONCompareMode.NON_EXTENSIBLE);
}