fix mockito conversion issues

Author: jpbempel

dd-trace-core/src/test/java/datadog/trace/core/propagation/TracingPropagatorTest.java

@@ -61,7 +61,7 @@ void testTracingPropagatorContextInjection() {
 
     propagator.inject(span, carrier, setter);
 
-    verify(injector).inject(any(DDSpanContext.class), same(carrier), any());
+    verify(injector).inject(same((DDSpanContext) span.context()), same(carrier), any());
 
     span.finish();
     tracer.close();
@@ -175,7 +175,7 @@ void testPropagationWhenTracingIsDisabled(
 
     propagator.inject(span, carrier, setter);
 
-    verify(injector, times(injected)).inject(any(DDSpanContext.class), same(carrier), any());
+    verify(injector, times(injected)).inject(same((DDSpanContext)span.context()), same(carrier), any());
 
     span.finish();
     tracer.close();

dd-trace-core/src/test/java/datadog/trace/core/propagation/XRayHttpInjectorTest.java

@@ -5,6 +5,7 @@
 import static java.util.Collections.singletonMap;
 import static org.mockito.Mockito.clearInvocations;
 import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.verifyNoMoreInteractions;
 import static org.mockito.Mockito.when;
@@ -160,11 +161,14 @@ void injectHttpHeadersWithEndToEnd() {
     mockedContext.beginEndToEnd();
     injector.inject(mockedContext, carrier, MapSetter.INSTANCE);
 
+    // 2 calls: CoreTracer constructor + beginEndToEnd()
+    verify(timeSource, times(2)).getCurrentTimeNanos();
+    verify(timeSource, times(2)).getNanoTicks();
     verify(carrier)
         .put(
             "X-Amzn-Trace-Id",
             "Root=1-633c7675-000000000000000000000001;Parent=0000000000000002;_dd.origin=fakeOrigin;t0=1664906869195;k=v");
-    verifyNoMoreInteractions(carrier);
+    verifyNoMoreInteractions(timeSource, carrier);
 
     tracer.close();
   }