diff options
author | Bogdan Drutu <bdrutu@google.com> | 2018-09-22 09:37:29 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-09-22 09:37:29 -0700 |
commit | f0e78ba5a10165cce2b63809bfdfe3e34b1eb456 (patch) | |
tree | 68b4ed0d345bf218d15a7b9234a2b70e2f0ec660 /impl_core/src/test/java/io/opencensus/implcore/trace/export | |
parent | 0c988a28144199450fa21ef55ddcac8c2c9323bc (diff) | |
download | opencensus-java-f0e78ba5a10165cce2b63809bfdfe3e34b1eb456.tar.gz |
Split SpanImpl into record/notrecord span impl to avoid checks of recording all time. (#1468)
* Split SpanImpl into record/notrecord span impl to avoid checks of recording all time.
* Remove one extra SuppressWarnings.
Diffstat (limited to 'impl_core/src/test/java/io/opencensus/implcore/trace/export')
5 files changed, 76 insertions, 96 deletions
diff --git a/impl_core/src/test/java/io/opencensus/implcore/trace/export/InProcessRunningSpanStoreImplTest.java b/impl_core/src/test/java/io/opencensus/implcore/trace/export/InProcessRunningSpanStoreImplTest.java index 85db0868..68ce1c18 100644 --- a/impl_core/src/test/java/io/opencensus/implcore/trace/export/InProcessRunningSpanStoreImplTest.java +++ b/impl_core/src/test/java/io/opencensus/implcore/trace/export/InProcessRunningSpanStoreImplTest.java @@ -21,17 +21,15 @@ import static com.google.common.truth.Truth.assertThat; import io.opencensus.common.Duration; import io.opencensus.implcore.common.MillisClock; import io.opencensus.implcore.internal.SimpleEventQueue; -import io.opencensus.implcore.trace.SpanImpl; -import io.opencensus.implcore.trace.SpanImpl.StartEndHandler; +import io.opencensus.implcore.trace.RecordEventsSpanImpl; +import io.opencensus.implcore.trace.RecordEventsSpanImpl.StartEndHandler; import io.opencensus.implcore.trace.StartEndHandlerImpl; -import io.opencensus.trace.Span.Options; import io.opencensus.trace.SpanContext; import io.opencensus.trace.SpanId; import io.opencensus.trace.TraceId; import io.opencensus.trace.TraceOptions; import io.opencensus.trace.config.TraceParams; import io.opencensus.trace.export.RunningSpanStore.Filter; -import java.util.EnumSet; import java.util.Random; import org.junit.Test; import org.junit.runner.RunWith; @@ -51,17 +49,15 @@ public class InProcessRunningSpanStoreImplTest { private final StartEndHandler startEndHandler = new StartEndHandlerImpl( sampledSpansServiceExporter, activeSpansExporter, null, new SimpleEventQueue()); - private final EnumSet<Options> recordSpanOptions = EnumSet.of(Options.RECORD_EVENTS); - private SpanImpl createSpan(String spanName) { + private RecordEventsSpanImpl createSpan(String spanName) { final SpanContext spanContext = SpanContext.create( TraceId.generateRandomId(random), SpanId.generateRandomId(random), TraceOptions.DEFAULT); - return SpanImpl.startSpan( + return RecordEventsSpanImpl.startSpan( spanContext, - recordSpanOptions, spanName, null, SpanId.generateRandomId(random), @@ -74,8 +70,8 @@ public class InProcessRunningSpanStoreImplTest { @Test public void getSummary_SpansWithDifferentNames() { - final SpanImpl span1 = createSpan(SPAN_NAME_1); - final SpanImpl span2 = createSpan(SPAN_NAME_2); + final RecordEventsSpanImpl span1 = createSpan(SPAN_NAME_1); + final RecordEventsSpanImpl span2 = createSpan(SPAN_NAME_2); assertThat(activeSpansExporter.getSummary().getPerSpanNameSummary().size()).isEqualTo(2); assertThat( activeSpansExporter @@ -107,9 +103,9 @@ public class InProcessRunningSpanStoreImplTest { @Test public void getSummary_SpansWithSameName() { - final SpanImpl span1 = createSpan(SPAN_NAME_1); - final SpanImpl span2 = createSpan(SPAN_NAME_1); - final SpanImpl span3 = createSpan(SPAN_NAME_1); + final RecordEventsSpanImpl span1 = createSpan(SPAN_NAME_1); + final RecordEventsSpanImpl span2 = createSpan(SPAN_NAME_1); + final RecordEventsSpanImpl span3 = createSpan(SPAN_NAME_1); assertThat(activeSpansExporter.getSummary().getPerSpanNameSummary().size()).isEqualTo(1); assertThat( activeSpansExporter @@ -142,8 +138,8 @@ public class InProcessRunningSpanStoreImplTest { @Test public void getActiveSpans_SpansWithDifferentNames() { - SpanImpl span1 = createSpan(SPAN_NAME_1); - SpanImpl span2 = createSpan(SPAN_NAME_2); + RecordEventsSpanImpl span1 = createSpan(SPAN_NAME_1); + RecordEventsSpanImpl span2 = createSpan(SPAN_NAME_2); assertThat(activeSpansExporter.getRunningSpans(Filter.create(SPAN_NAME_1, 0))) .containsExactly(span1.toSpanData()); assertThat(activeSpansExporter.getRunningSpans(Filter.create(SPAN_NAME_1, 2))) @@ -156,9 +152,9 @@ public class InProcessRunningSpanStoreImplTest { @Test public void getActiveSpans_SpansWithSameName() { - SpanImpl span1 = createSpan(SPAN_NAME_1); - SpanImpl span2 = createSpan(SPAN_NAME_1); - SpanImpl span3 = createSpan(SPAN_NAME_1); + RecordEventsSpanImpl span1 = createSpan(SPAN_NAME_1); + RecordEventsSpanImpl span2 = createSpan(SPAN_NAME_1); + RecordEventsSpanImpl span3 = createSpan(SPAN_NAME_1); assertThat(activeSpansExporter.getRunningSpans(Filter.create(SPAN_NAME_1, 0))) .containsExactly(span1.toSpanData(), span2.toSpanData(), span3.toSpanData()); assertThat(activeSpansExporter.getRunningSpans(Filter.create(SPAN_NAME_1, 2)).size()) diff --git a/impl_core/src/test/java/io/opencensus/implcore/trace/export/InProcessSampledSpanStoreImplTest.java b/impl_core/src/test/java/io/opencensus/implcore/trace/export/InProcessSampledSpanStoreImplTest.java index e1d2c4bb..7d8b434e 100644 --- a/impl_core/src/test/java/io/opencensus/implcore/trace/export/InProcessSampledSpanStoreImplTest.java +++ b/impl_core/src/test/java/io/opencensus/implcore/trace/export/InProcessSampledSpanStoreImplTest.java @@ -21,12 +21,11 @@ import static com.google.common.truth.Truth.assertThat; import io.opencensus.common.Duration; import io.opencensus.common.Timestamp; import io.opencensus.implcore.internal.SimpleEventQueue; -import io.opencensus.implcore.trace.SpanImpl; -import io.opencensus.implcore.trace.SpanImpl.StartEndHandler; +import io.opencensus.implcore.trace.RecordEventsSpanImpl; +import io.opencensus.implcore.trace.RecordEventsSpanImpl.StartEndHandler; import io.opencensus.testing.common.TestClock; import io.opencensus.trace.EndSpanOptions; import io.opencensus.trace.Span; -import io.opencensus.trace.Span.Options; import io.opencensus.trace.SpanContext; import io.opencensus.trace.SpanId; import io.opencensus.trace.Status; @@ -41,7 +40,6 @@ import io.opencensus.trace.export.SampledSpanStore.PerSpanNameSummary; import io.opencensus.trace.export.SpanData; import java.util.Collection; import java.util.Collections; -import java.util.EnumSet; import java.util.Map; import java.util.Random; import java.util.concurrent.TimeUnit; @@ -66,19 +64,18 @@ public class InProcessSampledSpanStoreImplTest { SpanContext.create( TraceId.generateRandomId(random), SpanId.generateRandomId(random), TraceOptions.DEFAULT); private final SpanId parentSpanId = SpanId.generateRandomId(random); - private final EnumSet<Options> recordSpanOptions = EnumSet.of(Options.RECORD_EVENTS); private final TestClock testClock = TestClock.create(Timestamp.create(12345, 54321)); private final InProcessSampledSpanStoreImpl sampleStore = new InProcessSampledSpanStoreImpl(new SimpleEventQueue()); private final StartEndHandler startEndHandler = new StartEndHandler() { @Override - public void onStart(SpanImpl span) { + public void onStart(RecordEventsSpanImpl span) { // Do nothing. } @Override - public void onEnd(SpanImpl span) { + public void onEnd(RecordEventsSpanImpl span) { sampleStore.considerForSampling(span); } }; @@ -88,10 +85,9 @@ public class InProcessSampledSpanStoreImplTest { sampleStore.registerSpanNamesForCollection(Collections.singletonList(REGISTERED_SPAN_NAME)); } - private SpanImpl createSampledSpan(String spanName) { - return SpanImpl.startSpan( + private RecordEventsSpanImpl createSampledSpan(String spanName) { + return RecordEventsSpanImpl.startSpan( sampledSpanContext, - recordSpanOptions, spanName, null, parentSpanId, @@ -102,10 +98,9 @@ public class InProcessSampledSpanStoreImplTest { testClock); } - private SpanImpl createNotSampledSpan(String spanName) { - return SpanImpl.startSpan( + private RecordEventsSpanImpl createNotSampledSpan(String spanName) { + return RecordEventsSpanImpl.startSpan( notSampledSpanContext, - recordSpanOptions, spanName, null, parentSpanId, @@ -216,7 +211,7 @@ public class InProcessSampledSpanStoreImplTest { @Test public void getErrorSampledSpans() { - SpanImpl span = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, 1000)); span.end(EndSpanOptions.builder().setStatus(Status.CANCELLED).build()); Collection<SpanData> samples = @@ -228,12 +223,12 @@ public class InProcessSampledSpanStoreImplTest { @Test public void getErrorSampledSpans_MaxSpansToReturn() { - SpanImpl span1 = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span1 = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, 1000)); span1.end(EndSpanOptions.builder().setStatus(Status.CANCELLED).build()); // Advance time to allow other spans to be sampled. testClock.advanceTime(Duration.create(5, 0)); - SpanImpl span2 = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span2 = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, 1000)); span2.end(EndSpanOptions.builder().setStatus(Status.CANCELLED).build()); Collection<SpanData> samples = @@ -246,10 +241,10 @@ public class InProcessSampledSpanStoreImplTest { @Test public void getErrorSampledSpans_NullCode() { - SpanImpl span1 = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span1 = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, 1000)); span1.end(EndSpanOptions.builder().setStatus(Status.CANCELLED).build()); - SpanImpl span2 = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span2 = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, 1000)); span2.end(EndSpanOptions.builder().setStatus(Status.UNKNOWN).build()); Collection<SpanData> samples = @@ -260,10 +255,10 @@ public class InProcessSampledSpanStoreImplTest { @Test public void getErrorSampledSpans_NullCode_MaxSpansToReturn() { - SpanImpl span1 = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span1 = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, 1000)); span1.end(EndSpanOptions.builder().setStatus(Status.CANCELLED).build()); - SpanImpl span2 = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span2 = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, 1000)); span2.end(EndSpanOptions.builder().setStatus(Status.UNKNOWN).build()); Collection<SpanData> samples = @@ -274,7 +269,7 @@ public class InProcessSampledSpanStoreImplTest { @Test public void getLatencySampledSpans() { - SpanImpl span = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, (int) TimeUnit.MICROSECONDS.toNanos(20))); span.end(); Collection<SpanData> samples = @@ -290,7 +285,7 @@ public class InProcessSampledSpanStoreImplTest { @Test public void getLatencySampledSpans_ExclusiveUpperBound() { - SpanImpl span = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, (int) TimeUnit.MICROSECONDS.toNanos(20))); span.end(); Collection<SpanData> samples = @@ -305,7 +300,7 @@ public class InProcessSampledSpanStoreImplTest { @Test public void getLatencySampledSpans_InclusiveLowerBound() { - SpanImpl span = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, (int) TimeUnit.MICROSECONDS.toNanos(20))); span.end(); Collection<SpanData> samples = @@ -321,12 +316,12 @@ public class InProcessSampledSpanStoreImplTest { @Test public void getLatencySampledSpans_QueryBetweenMultipleBuckets() { - SpanImpl span1 = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span1 = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, (int) TimeUnit.MICROSECONDS.toNanos(20))); span1.end(); // Advance time to allow other spans to be sampled. testClock.advanceTime(Duration.create(5, 0)); - SpanImpl span2 = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span2 = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, (int) TimeUnit.MICROSECONDS.toNanos(200))); span2.end(); Collection<SpanData> samples = @@ -341,12 +336,12 @@ public class InProcessSampledSpanStoreImplTest { @Test public void getLatencySampledSpans_MaxSpansToReturn() { - SpanImpl span1 = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span1 = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, (int) TimeUnit.MICROSECONDS.toNanos(20))); span1.end(); // Advance time to allow other spans to be sampled. testClock.advanceTime(Duration.create(5, 0)); - SpanImpl span2 = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span2 = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, (int) TimeUnit.MICROSECONDS.toNanos(200))); span2.end(); Collection<SpanData> samples = @@ -362,7 +357,7 @@ public class InProcessSampledSpanStoreImplTest { @Test public void ignoreNegativeSpanLatency() { - SpanImpl span = createSampledSpan(REGISTERED_SPAN_NAME); + RecordEventsSpanImpl span = createSampledSpan(REGISTERED_SPAN_NAME); testClock.advanceTime(Duration.create(0, (int) TimeUnit.MICROSECONDS.toNanos(-20))); span.end(); Collection<SpanData> samples = diff --git a/impl_core/src/test/java/io/opencensus/implcore/trace/export/NoopRunningSpanStoreImplTest.java b/impl_core/src/test/java/io/opencensus/implcore/trace/export/NoopRunningSpanStoreImplTest.java index 2877bb33..96669df7 100644 --- a/impl_core/src/test/java/io/opencensus/implcore/trace/export/NoopRunningSpanStoreImplTest.java +++ b/impl_core/src/test/java/io/opencensus/implcore/trace/export/NoopRunningSpanStoreImplTest.java @@ -22,17 +22,15 @@ import io.opencensus.common.Timestamp; import io.opencensus.implcore.internal.EventQueue; import io.opencensus.implcore.internal.SimpleEventQueue; import io.opencensus.implcore.internal.TimestampConverter; -import io.opencensus.implcore.trace.SpanImpl; -import io.opencensus.implcore.trace.SpanImpl.StartEndHandler; +import io.opencensus.implcore.trace.RecordEventsSpanImpl; +import io.opencensus.implcore.trace.RecordEventsSpanImpl.StartEndHandler; import io.opencensus.testing.common.TestClock; -import io.opencensus.trace.Span.Options; import io.opencensus.trace.SpanContext; import io.opencensus.trace.SpanId; import io.opencensus.trace.TraceId; import io.opencensus.trace.TraceOptions; import io.opencensus.trace.config.TraceParams; import io.opencensus.trace.export.RunningSpanStore.Filter; -import java.util.EnumSet; import java.util.Random; import org.junit.Before; import org.junit.Test; @@ -54,9 +52,8 @@ public class NoopRunningSpanStoreImplTest { TraceId.generateRandomId(random), SpanId.generateRandomId(random), TraceOptions.DEFAULT); private final TestClock testClock = TestClock.create(timestamp); private final TimestampConverter timestampConverter = TimestampConverter.now(testClock); - private final EnumSet<Options> recordSpanOptions = EnumSet.of(Options.RECORD_EVENTS); @Mock private StartEndHandler startEndHandler; - private SpanImpl spanImpl; + private RecordEventsSpanImpl recordEventsSpanImpl; // maxSpansToReturn=0 means all private final Filter filter = Filter.create(SPAN_NAME, 0 /* maxSpansToReturn */); private final EventQueue eventQueue = new SimpleEventQueue(); @@ -66,10 +63,9 @@ public class NoopRunningSpanStoreImplTest { @Before public void setUp() { MockitoAnnotations.initMocks(this); - spanImpl = - SpanImpl.startSpan( + recordEventsSpanImpl = + RecordEventsSpanImpl.startSpan( spanContext, - recordSpanOptions, SPAN_NAME, null, null, @@ -90,10 +86,10 @@ public class NoopRunningSpanStoreImplTest { public void noopImplementation() { getMethodsShouldReturnEmpty(); // onStart() does not affect the result. - runningSpanStoreImpl.onStart(spanImpl); + runningSpanStoreImpl.onStart(recordEventsSpanImpl); getMethodsShouldReturnEmpty(); // onEnd() does not affect the result. - runningSpanStoreImpl.onEnd(spanImpl); + runningSpanStoreImpl.onEnd(recordEventsSpanImpl); getMethodsShouldReturnEmpty(); } } diff --git a/impl_core/src/test/java/io/opencensus/implcore/trace/export/NoopSampledSpanStoreImplTest.java b/impl_core/src/test/java/io/opencensus/implcore/trace/export/NoopSampledSpanStoreImplTest.java index 3ba33fd1..b9fbd432 100644 --- a/impl_core/src/test/java/io/opencensus/implcore/trace/export/NoopSampledSpanStoreImplTest.java +++ b/impl_core/src/test/java/io/opencensus/implcore/trace/export/NoopSampledSpanStoreImplTest.java @@ -22,10 +22,9 @@ import io.opencensus.common.Timestamp; import io.opencensus.implcore.internal.EventQueue; import io.opencensus.implcore.internal.SimpleEventQueue; import io.opencensus.implcore.internal.TimestampConverter; -import io.opencensus.implcore.trace.SpanImpl; -import io.opencensus.implcore.trace.SpanImpl.StartEndHandler; +import io.opencensus.implcore.trace.RecordEventsSpanImpl; +import io.opencensus.implcore.trace.RecordEventsSpanImpl.StartEndHandler; import io.opencensus.testing.common.TestClock; -import io.opencensus.trace.Span.Options; import io.opencensus.trace.SpanContext; import io.opencensus.trace.SpanId; import io.opencensus.trace.TraceId; @@ -35,7 +34,6 @@ import io.opencensus.trace.export.SampledSpanStore.ErrorFilter; import io.opencensus.trace.export.SampledSpanStore.LatencyFilter; import java.util.Collection; import java.util.Collections; -import java.util.EnumSet; import java.util.Random; import org.junit.Before; import org.junit.Test; @@ -59,9 +57,8 @@ public final class NoopSampledSpanStoreImplTest { TraceId.generateRandomId(random), SpanId.generateRandomId(random), TraceOptions.DEFAULT); private final TestClock testClock = TestClock.create(timestamp); private final TimestampConverter timestampConverter = TimestampConverter.now(testClock); - private final EnumSet<Options> recordSpanOptions = EnumSet.of(Options.RECORD_EVENTS); @Mock private StartEndHandler startEndHandler; - private SpanImpl spanImpl; + private RecordEventsSpanImpl recordEventsSpanImpl; // maxSpansToReturn=0 means all private final ErrorFilter errorFilter = ErrorFilter.create(SPAN_NAME, null /* canonicalCode */, 0 /* maxSpansToReturn */); @@ -99,10 +96,9 @@ public final class NoopSampledSpanStoreImplTest { // considerForSampling() should do nothing and do not affect the result. // It should be called after registerSpanNamesForCollection. - spanImpl = - SpanImpl.startSpan( + recordEventsSpanImpl = + RecordEventsSpanImpl.startSpan( spanContext, - recordSpanOptions, SPAN_NAME, null, null, @@ -111,8 +107,8 @@ public final class NoopSampledSpanStoreImplTest { startEndHandler, timestampConverter, testClock); - spanImpl.end(); - sampledSpanStoreImpl.considerForSampling(spanImpl); + recordEventsSpanImpl.end(); + sampledSpanStoreImpl.considerForSampling(recordEventsSpanImpl); getMethodsShouldReturnEmpty(); // unregisterSpanNamesForCollection() should do nothing and do not affect the result. diff --git a/impl_core/src/test/java/io/opencensus/implcore/trace/export/SpanExporterImplTest.java b/impl_core/src/test/java/io/opencensus/implcore/trace/export/SpanExporterImplTest.java index d99776d1..f8f1d917 100644 --- a/impl_core/src/test/java/io/opencensus/implcore/trace/export/SpanExporterImplTest.java +++ b/impl_core/src/test/java/io/opencensus/implcore/trace/export/SpanExporterImplTest.java @@ -23,11 +23,10 @@ import static org.mockito.Mockito.doThrow; import io.opencensus.common.Duration; import io.opencensus.implcore.common.MillisClock; import io.opencensus.implcore.internal.SimpleEventQueue; -import io.opencensus.implcore.trace.SpanImpl; -import io.opencensus.implcore.trace.SpanImpl.StartEndHandler; +import io.opencensus.implcore.trace.RecordEventsSpanImpl; +import io.opencensus.implcore.trace.RecordEventsSpanImpl.StartEndHandler; import io.opencensus.implcore.trace.StartEndHandlerImpl; import io.opencensus.testing.export.TestHandler; -import io.opencensus.trace.Span.Options; import io.opencensus.trace.SpanContext; import io.opencensus.trace.SpanId; import io.opencensus.trace.TraceId; @@ -35,7 +34,6 @@ import io.opencensus.trace.TraceOptions; import io.opencensus.trace.config.TraceParams; import io.opencensus.trace.export.SpanData; import io.opencensus.trace.export.SpanExporter.Handler; -import java.util.EnumSet; import java.util.List; import java.util.Random; import org.junit.Before; @@ -60,7 +58,6 @@ public class SpanExporterImplTest { SpanContext.create( TraceId.generateRandomId(random), SpanId.generateRandomId(random), TraceOptions.DEFAULT); private final RunningSpanStoreImpl runningSpanStore = new InProcessRunningSpanStoreImpl(); - private final EnumSet<Options> recordSpanOptions = EnumSet.of(Options.RECORD_EVENTS); private final TestHandler serviceHandler = new TestHandler(); @Mock private Handler mockServiceHandler; @@ -69,11 +66,11 @@ public class SpanExporterImplTest { MockitoAnnotations.initMocks(this); } - private SpanImpl createSampledEndedSpan(StartEndHandler startEndHandler, String spanName) { - SpanImpl span = - SpanImpl.startSpan( + private RecordEventsSpanImpl createSampledEndedSpan( + StartEndHandler startEndHandler, String spanName) { + RecordEventsSpanImpl span = + RecordEventsSpanImpl.startSpan( sampledSpanContext, - recordSpanOptions, spanName, null, null, @@ -86,11 +83,11 @@ public class SpanExporterImplTest { return span; } - private SpanImpl createNotSampledEndedSpan(StartEndHandler startEndHandler, String spanName) { - SpanImpl span = - SpanImpl.startSpan( + private RecordEventsSpanImpl createNotSampledEndedSpan( + StartEndHandler startEndHandler, String spanName) { + RecordEventsSpanImpl span = + RecordEventsSpanImpl.startSpan( notSampledSpanContext, - recordSpanOptions, spanName, null, null, @@ -111,8 +108,8 @@ public class SpanExporterImplTest { spanExporter.registerHandler("test.service", serviceHandler); - SpanImpl span1 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); - SpanImpl span2 = createSampledEndedSpan(startEndHandler, SPAN_NAME_2); + RecordEventsSpanImpl span1 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); + RecordEventsSpanImpl span2 = createSampledEndedSpan(startEndHandler, SPAN_NAME_2); List<SpanData> exported = serviceHandler.waitForExport(2); assertThat(exported).containsExactly(span1.toSpanData(), span2.toSpanData()); } @@ -125,12 +122,12 @@ public class SpanExporterImplTest { spanExporter.registerHandler("test.service", serviceHandler); - SpanImpl span1 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); - SpanImpl span2 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); - SpanImpl span3 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); - SpanImpl span4 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); - SpanImpl span5 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); - SpanImpl span6 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); + RecordEventsSpanImpl span1 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); + RecordEventsSpanImpl span2 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); + RecordEventsSpanImpl span3 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); + RecordEventsSpanImpl span4 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); + RecordEventsSpanImpl span5 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); + RecordEventsSpanImpl span6 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); List<SpanData> exported = serviceHandler.waitForExport(6); assertThat(exported) .containsExactly( @@ -167,11 +164,11 @@ public class SpanExporterImplTest { spanExporter.registerHandler("test.service", serviceHandler); spanExporter.registerHandler("mock.service", mockServiceHandler); - SpanImpl span1 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); + RecordEventsSpanImpl span1 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); List<SpanData> exported = serviceHandler.waitForExport(1); assertThat(exported).containsExactly(span1.toSpanData()); // Continue to export after the exception was received. - SpanImpl span2 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); + RecordEventsSpanImpl span2 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); exported = serviceHandler.waitForExport(1); assertThat(exported).containsExactly(span2.toSpanData()); } @@ -186,8 +183,8 @@ public class SpanExporterImplTest { TestHandler serviceHandler2 = new TestHandler(); spanExporter.registerHandler("test.service2", serviceHandler2); - SpanImpl span1 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); - SpanImpl span2 = createSampledEndedSpan(startEndHandler, SPAN_NAME_2); + RecordEventsSpanImpl span1 = createSampledEndedSpan(startEndHandler, SPAN_NAME_1); + RecordEventsSpanImpl span2 = createSampledEndedSpan(startEndHandler, SPAN_NAME_2); List<SpanData> exported1 = serviceHandler.waitForExport(2); List<SpanData> exported2 = serviceHandler2.waitForExport(2); assertThat(exported1).containsExactly(span1.toSpanData(), span2.toSpanData()); @@ -202,8 +199,8 @@ public class SpanExporterImplTest { spanExporter.registerHandler("test.service", serviceHandler); - SpanImpl span1 = createNotSampledEndedSpan(startEndHandler, SPAN_NAME_1); - SpanImpl span2 = createSampledEndedSpan(startEndHandler, SPAN_NAME_2); + RecordEventsSpanImpl span1 = createNotSampledEndedSpan(startEndHandler, SPAN_NAME_1); + RecordEventsSpanImpl span2 = createSampledEndedSpan(startEndHandler, SPAN_NAME_2); // Spans are recorded and exported in the same order as they are ended, we test that a non // sampled span is not exported by creating and ending a sampled span after a non sampled span // and checking that the first exported span is the sampled span (the non sampled did not get @@ -224,7 +221,7 @@ public class SpanExporterImplTest { spanExporter.registerHandler("test.service", serviceHandler); - SpanImpl span2 = createSampledEndedSpan(startEndHandler, SPAN_NAME_2); + RecordEventsSpanImpl span2 = createSampledEndedSpan(startEndHandler, SPAN_NAME_2); // Force a flush, without this, the #waitForExport() call below would block indefinitely. spanExporter.flush(); |