@@ -38,6 +38,7 @@ import {
3838 InMemorySpanExporter ,
3939 NodeTracerProvider ,
4040 ReadableSpan ,
41+ SpanProcessor ,
4142 TimedEvent ,
4243} from '@opentelemetry/sdk-trace-node' ;
4344import { setLogFunction , Firestore } from '../src' ;
@@ -242,30 +243,28 @@ describe('Tracing Tests', () => {
242243 contextManager . enable ( ) ;
243244 context . setGlobalContextManager ( contextManager ) ;
244245
246+ inMemorySpanExporter = new InMemorySpanExporter ( ) ;
247+ consoleSpanExporter = new ConsoleSpanExporter ( ) ;
248+ gcpTraceExporter = new TraceExporter ( ) ;
249+
250+ const spanProcessors : SpanProcessor [ ] = [
251+ // Always add the console exporter for local debugging.
252+ new BatchSpanProcessor ( consoleSpanExporter ) ,
253+ ] ;
254+
255+ if ( testConfig . e2e ) {
256+ spanProcessors . push ( new BatchSpanProcessor ( gcpTraceExporter ) ) ;
257+ } else {
258+ spanProcessors . push ( new BatchSpanProcessor ( inMemorySpanExporter ) ) ;
259+ }
260+
245261 // Create a new tracer and span processor for each test to make sure there
246262 // are no overlaps when reading the results.
247263 tracerProvider = new NodeTracerProvider ( {
248264 sampler : new AlwaysOnSampler ( ) ,
265+ spanProcessors,
249266 } ) ;
250267
251- inMemorySpanExporter = new InMemorySpanExporter ( ) ;
252- consoleSpanExporter = new ConsoleSpanExporter ( ) ;
253- gcpTraceExporter = new TraceExporter ( ) ;
254-
255- // TODO(node-18 upgrade) uncomment and fix this code
256- // // Always add the console exporter for local debugging.
257- // tracerProvider.addSpanProcessor(
258- // new BatchSpanProcessor(consoleSpanExporter)
259- // );
260-
261- // if (testConfig.e2e) {
262- // tracerProvider.addSpanProcessor(new BatchSpanProcessor(gcpTraceExporter));
263- // } else {
264- // tracerProvider.addSpanProcessor(
265- // new BatchSpanProcessor(inMemorySpanExporter)
266- // );
267- // }
268-
269268 if ( testConfig . useGlobalOpenTelemetry ) {
270269 trace . setGlobalTracerProvider ( tracerProvider ) ;
271270 }
0 commit comments