@@ -56,7 +56,6 @@ import datadog.trace.bootstrap.ActiveSubsystems
5656import datadog.trace.bootstrap.InstrumentationErrors
5757import datadog.trace.bootstrap.debugger.DebuggerContext
5858import datadog.trace.bootstrap.instrumentation.api.AgentSpan
59- import datadog.trace.bootstrap.instrumentation.api.AgentSpanContext
6059import datadog.trace.bootstrap.instrumentation.api.AgentTracer
6160import datadog.trace.common.metrics.EventListener
6261import datadog.trace.common.metrics.Sink
@@ -409,17 +408,8 @@ abstract class InstrumentationSpecification extends DDSpecification implements A
409408 TracerInstaller . forceInstallGlobalTracer(TEST_TRACER )
410409
411410 boolean enabledFinishTimingChecks = this . enabledFinishTimingChecks()
412- TEST_TRACER . startSpan(_ as String , _ as CharSequence ) >> { String instrName , CharSequence spanName ->
413- trackStartSpan(callRealMethod(), instrName, enabledFinishTimingChecks)
414- }
415- TEST_TRACER . startSpan(_ as String , _ as CharSequence , _ as Long ) >> { String instrName , CharSequence spanName , long micros ->
416- trackStartSpan(callRealMethod(), instrName, enabledFinishTimingChecks)
417- }
418- TEST_TRACER . startSpan(_ as String , _ as CharSequence , _ as AgentSpanContext ) >> { String instrName , CharSequence spanName , AgentSpanContext ctx ->
419- trackStartSpan(callRealMethod(), instrName, enabledFinishTimingChecks)
420- }
421- TEST_TRACER . startSpan(_ as String , _ as CharSequence , _ as AgentSpanContext , _ as Long ) >> { String instrName , CharSequence spanName , AgentSpanContext ctx , long micros ->
422- trackStartSpan(callRealMethod(), instrName, enabledFinishTimingChecks)
411+ TEST_TRACER . startSpan(* _) >> { args ->
412+ trackStartSpan(callRealMethod(), args[0 ] as String , enabledFinishTimingChecks)
423413 }
424414
425415 ClassInjector . enableClassInjection(INSTRUMENTATION )
0 commit comments