fix: startactiveSpan test

Signed-off-by: naseemkullah <naseem@transit.app>
open-telemetry · May 22, 2021 · 230ed4f · 230ed4f
1 parent b80b40f
commit 230ed4f
Showing 1 changed file with 22 additions and 12 deletions.
diff --git a/packages/opentelemetry-tracing/test/Tracer.test.ts b/packages/opentelemetry-tracing/test/Tracer.test.ts
@@ -228,20 +228,30 @@ describe('Tracer', () => {
       { sampler: new TestSampler() },
       tracerProvider
     );
-    tracer.startActiveSpan('my-span', span => {
-      assert.strictEqual(getSpan(context.active()), span);
-      span.end();
-    });
 
-    tracer.startActiveSpan('my-span', {}, span => {
-      assert.strictEqual(getSpan(context.active()), span);
-      span.end();
-    });
+    assert.strictEqual( tracer.startActiveSpan('my-span', span => {
+      try {
+        return 1
+      } finally {
+        span.end();
+      }
+    }), 1);
 
-    tracer.startActiveSpan('my-span', {}, ROOT_CONTEXT, span => {
-      assert.strictEqual(getSpan(context.active()), span);
-      span.end();
-    });
+    assert.strictEqual( tracer.startActiveSpan('my-span', {}, span => {
+      try {
+        return 1
+      } finally {
+        span.end();
+      }
+    }), 1);
+
+    assert.strictEqual( tracer.startActiveSpan('my-span', {}, context.active(), span => {
+      try {
+        return 1
+      } finally {
+        span.end();
+      }
+    }), 1);
 
   });
 });