test: assert active spans are indeed active with help of TestStackCon…

…textManager Signed-off-by: naseemkullah <[email protected]>
open-telemetry · May 28, 2021 · 1944b4e · 1944b4e
1 parent 0335473
commit 1944b4e
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/packages/opentelemetry-tracing/test/Tracer.test.ts b/packages/opentelemetry-tracing/test/Tracer.test.ts
@@ -24,6 +24,7 @@ import {
  trace,
  TraceFlags
 } from '@opentelemetry/api';
+import { getSpan } from '@opentelemetry/api/build/src/trace/context-utils';
 import {
  AlwaysOffSampler,
  AlwaysOnSampler,
@@ -32,6 +33,7 @@ import {
 } from '@opentelemetry/core';
 import * as assert from 'assert';
 import { BasicTracerProvider, Span, Tracer } from '../src';
+import { TestStackContextManager } from './export/TestStackContextManager';
 
 describe('Tracer', () => {
  const tracerProvider = new BasicTracerProvider();
@@ -50,7 +52,13 @@ describe('Tracer', () => {
  }
  }
 
+ beforeEach(() => {
+ const contextManager = new TestStackContextManager().enable();
+ context.setGlobalContextManager(contextManager);
+ });
+
  afterEach(() => {
+ context.disable();
  delete envSource.OTEL_TRACES_SAMPLER;
  delete envSource.OTEL_TRACES_SAMPLER_ARG;
  });
@@ -231,7 +239,7 @@ describe('Tracer', () => {
 
  assert.strictEqual(tracer.startActiveSpan('my-span', span => {
  try {
- // todo: assert.strictEqual(getSpan(context.active()), span)
+ assert.strictEqual(getSpan(context.active()), span)
  return 1
  } finally {
  span.end();
@@ -248,7 +256,7 @@ describe('Tracer', () => {
 
  assert.strictEqual(tracer.startActiveSpan('my-span', {}, span => {
  try {
- // todo: assert.strictEqual(getSpan(context.active()), span)
+ assert.strictEqual(getSpan(context.active()), span)
  return 1
  } finally {
  span.end();
@@ -269,7 +277,7 @@ describe('Tracer', () => {
 
  assert.strictEqual(tracer.startActiveSpan('my-span', {}, ctx, span => {
  try {
- // todo: assert.strictEqual(getSpan(context.active()), span)
+ assert.strictEqual(getSpan(context.active()), span)
  assert.strictEqual(ctx.getValue(ctxKey), 'bar')
  return 1
  } finally {