From b80b40fdddad51f68837b54f2a5a48daf4e8d69a Mon Sep 17 00:00:00 2001
From: naseemkullah <naseem@transit.app>
Date: Fri, 21 May 2021 21:22:30 -0400
Subject: [PATCH] test: startActiveSpan in Tracer

Signed-off-by: naseemkullah <naseem@transit.app>
---
 .../opentelemetry-tracing/test/Tracer.test.ts | 30 +++++++++++++++++--
 1 file changed, 27 insertions(+), 3 deletions(-)

diff --git a/packages/opentelemetry-tracing/test/Tracer.test.ts b/packages/opentelemetry-tracing/test/Tracer.test.ts
index 7a69c4dbdbb..d0b38c69804 100644
--- a/packages/opentelemetry-tracing/test/Tracer.test.ts
+++ b/packages/opentelemetry-tracing/test/Tracer.test.ts
@@ -15,19 +15,20 @@
  */
 
 import {
-  INVALID_TRACEID,
+  context,
+  getSpan, INVALID_TRACEID,
   ROOT_CONTEXT,
   Sampler,
   SamplingDecision,
   setSpanContext,
   SpanContext,
-  TraceFlags,
+  TraceFlags
 } from '@opentelemetry/api';
 import {
   AlwaysOffSampler,
   AlwaysOnSampler,
   InstrumentationLibrary,
-  suppressTracing,
+  suppressTracing
 } from '@opentelemetry/core';
 import * as assert from 'assert';
 import { BasicTracerProvider, Span, Tracer } from '../src';
@@ -220,4 +221,27 @@ describe('Tracer', () => {
     assert.strictEqual(context.traceFlags, TraceFlags.NONE);
     span.end();
   });
+
+  it('should start an active span', () => {
+    const tracer = new Tracer(
+      { name: 'default', version: '0.0.1' },
+      { 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();
+    });
+
+    tracer.startActiveSpan('my-span', {}, ROOT_CONTEXT, span => {
+      assert.strictEqual(getSpan(context.active()), span);
+      span.end();
+    });
+
+  });
 });