/* * Copyright 2018, OpenCensus Authors * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package io.opencensus.implcore.trace.export; import static com.google.common.truth.Truth.assertThat; 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.RecordEventsSpanImpl; import io.opencensus.implcore.trace.RecordEventsSpanImpl.StartEndHandler; import io.opencensus.testing.common.TestClock; 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.SampledSpanStore.ErrorFilter; import io.opencensus.trace.export.SampledSpanStore.LatencyFilter; import java.util.Collection; import java.util.Collections; import java.util.Random; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.junit.runners.JUnit4; import org.mockito.Mock; import org.mockito.MockitoAnnotations; /** Unit tests for {@link SampledSpanStoreImpl.NoopSampledSpanStoreImpl}. */ @RunWith(JUnit4.class) public final class NoopSampledSpanStoreImplTest { private static final String SPAN_NAME = "MySpanName"; private static final Collection<String> NAMES_FOR_COLLECTION = Collections.<String>singletonList(SPAN_NAME); private final Timestamp timestamp = Timestamp.create(1234, 5678); private final Random random = new Random(1234); private final SpanContext spanContext = SpanContext.create( TraceId.generateRandomId(random), SpanId.generateRandomId(random), TraceOptions.DEFAULT); private final TestClock testClock = TestClock.create(timestamp); private final TimestampConverter timestampConverter = TimestampConverter.now(testClock); @Mock private StartEndHandler startEndHandler; private RecordEventsSpanImpl recordEventsSpanImpl; // maxSpansToReturn=0 means all private final ErrorFilter errorFilter = ErrorFilter.create(SPAN_NAME, null /* canonicalCode */, 0 /* maxSpansToReturn */); private final LatencyFilter latencyFilter = LatencyFilter.create( SPAN_NAME, 0 /* latencyLowerNs */, Long.MAX_VALUE /* latencyUpperNs */, 0 /* maxSpansToReturn */); private final EventQueue eventQueue = new SimpleEventQueue(); private final SampledSpanStoreImpl sampledSpanStoreImpl = ExportComponentImpl.createWithoutInProcessStores(eventQueue).getSampledSpanStore(); @Before public void setUp() { MockitoAnnotations.initMocks(this); } private void getMethodsShouldReturnEmpty() { // get methods always return empty collections. assertThat(sampledSpanStoreImpl.getSummary().getPerSpanNameSummary()).isEmpty(); assertThat(sampledSpanStoreImpl.getRegisteredSpanNamesForCollection()).isEmpty(); assertThat(sampledSpanStoreImpl.getErrorSampledSpans(errorFilter)).isEmpty(); assertThat(sampledSpanStoreImpl.getLatencySampledSpans(latencyFilter)).isEmpty(); } @Test public void noopImplementation() { // None of the get methods should yield non-empty result. getMethodsShouldReturnEmpty(); // registerSpanNamesForCollection() should do nothing and do not affect the result. sampledSpanStoreImpl.registerSpanNamesForCollection(NAMES_FOR_COLLECTION); getMethodsShouldReturnEmpty(); // considerForSampling() should do nothing and do not affect the result. // It should be called after registerSpanNamesForCollection. recordEventsSpanImpl = RecordEventsSpanImpl.startSpan( spanContext, SPAN_NAME, null, null, false, TraceParams.DEFAULT, startEndHandler, timestampConverter, testClock); recordEventsSpanImpl.end(); sampledSpanStoreImpl.considerForSampling(recordEventsSpanImpl); getMethodsShouldReturnEmpty(); // unregisterSpanNamesForCollection() should do nothing and do not affect the result. sampledSpanStoreImpl.unregisterSpanNamesForCollection(NAMES_FOR_COLLECTION); getMethodsShouldReturnEmpty(); } }