test: add failing startActiveSpan test...

Signed-off-by: naseemkullah <[email protected]>

Author: naseemkullah

test/proxy-implementations/proxy-tracer.test.ts

@@ -17,18 +17,20 @@
 import * as assert from 'assert';
 import * as sinon from 'sinon';
 import {
-  ProxyTracerProvider,
-  SpanKind,
-  TracerProvider,
-  ProxyTracer,
-  Tracer,
-  Span,
+  Context,
   NoopTracer,
+  ProxyTracer,
+  ProxyTracerProvider,
   ROOT_CONTEXT,
+  Span,
+  SpanKind,
   SpanOptions,
+  Tracer,
+  TracerProvider,
 } from '../../src';
+import { ContextAPI } from '../../src/api/context';
+import { getSpan, setSpan } from '../../src/trace/context-utils';
 import { NonRecordingSpan } from '../../src/trace/NonRecordingSpan';
-
 describe('ProxyTracer', () => {
   let provider: ProxyTracerProvider;
   const sandbox = sinon.createSandbox();
@@ -97,8 +99,48 @@ describe('ProxyTracer', () => {
           return delegateSpan;
         },
 
-        startActiveSpan() {
-          return Reflect.apply(tracer.startActiveSpan, tracer, arguments);
+        startActiveSpan<F extends (span: Span) => ReturnType<F>>(
+          name: string,
+          arg2: F | SpanOptions,
+          arg3?: F | Context,
+          arg4?: F
+        ): ReturnType<F> | undefined {
+          let fn: F | undefined,
+            options: SpanOptions | undefined,
+            activeContext: Context | undefined;
+          if (arguments.length === 2 && typeof arg2 === 'function') {
+            fn = arg2;
+          } else if (
+            arguments.length === 3 &&
+            typeof arg2 === 'object' &&
+            typeof arg3 === 'function'
+          ) {
+            options = arg2;
+            fn = arg3;
+          } else if (
+            arguments.length === 4 &&
+            typeof arg2 === 'object' &&
+            typeof arg3 === 'object' &&
+            typeof arg4 === 'function'
+          ) {
+            options = arg2;
+            activeContext = arg3;
+            fn = arg4;
+          }
+
+          const contextApiSingleton = ContextAPI.getInstance();
+          const activeCtx = activeContext ?? contextApiSingleton.active();
+          const span = this.startSpan(name, options, activeCtx);
+
+          if (fn) {
+            return contextApiSingleton.with(
+              setSpan(activeCtx, span),
+              fn,
+              undefined,
+              span
+            );
+          }
+          return;
         },
       };
 
@@ -118,6 +160,12 @@ describe('ProxyTracer', () => {
       assert.strictEqual(span, delegateSpan);
     });
 
+    it('should create active spans using the delegate tracer', () => {
+      tracer.startActiveSpan('test', span => {
+        assert.strictEqual(getSpan(ContextAPI.getInstance().active()), span);
+      });
+    });
+
     it('should pass original arguments to DelegateTracer#startSpan', () => {
       const startSpanStub = sandbox.stub(delegateTracer, 'startSpan');