// Copyright 2023 Google LLC // // 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 testutil import ( "context" "go.opentelemetry.io/otel" sdktrace "go.opentelemetry.io/otel/sdk/trace" "go.opentelemetry.io/otel/sdk/trace/tracetest" ) // OpenTelemetryTestExporter is a test utility exporter. It should be created // with NewOpenTelemetryTestExporter. type OpenTelemetryTestExporter struct { exporter *tracetest.InMemoryExporter tp *sdktrace.TracerProvider } // NewOpenTelemetryTestExporter creates a OpenTelemetryTestExporter with // underlying InMemoryExporter and TracerProvider from OpenTelemetry. func NewOpenTelemetryTestExporter() *OpenTelemetryTestExporter { exporter := tracetest.NewInMemoryExporter() tp := sdktrace.NewTracerProvider( sdktrace.WithSyncer(exporter), sdktrace.WithSampler(sdktrace.AlwaysSample()), ) otel.SetTracerProvider(tp) return &OpenTelemetryTestExporter{ exporter: exporter, tp: tp, } } // Spans returns the current in-memory stored spans. func (te *OpenTelemetryTestExporter) Spans() tracetest.SpanStubs { return te.exporter.GetSpans() } // Unregister shuts down the underlying OpenTelemetry TracerProvider. func (te *OpenTelemetryTestExporter) Unregister(ctx context.Context) { te.tp.Shutdown(ctx) }