fix more tests

Signed-off-by: Eliott Bouhana <[email protected]>

ddtrace/opentracer/tracer_test.go

@@ -115,12 +115,11 @@ func TestExtractError(t *testing.T) {
 }
 
 func TestSpanTelemetry(t *testing.T) {
-	telemetryClient := new(telemetrytest.MockClient)
+	telemetryClient := new(telemetrytest.RecordClient)
 	original := telemetry.GlobalClient()
 	telemetry.SwapClient(telemetryClient)
 	defer telemetry.SwapClient(original)
 	opentracing.SetGlobalTracer(New())
 	_ = opentracing.StartSpan("opentracing.span")
-	telemetryClient.AssertCalled(t, "Count", telemetry.NamespaceTracers, "spans_created", 1.0, telemetryTags, true)
-	telemetryClient.AssertNumberOfCalls(t, "Count", 1)
+	assert.NotZero(t, telemetryClient.Count(telemetry.NamespaceTracers, "spans_created", telemetryTags).Get())
 }

ddtrace/tracer/dynamic_config.go

@@ -26,11 +26,12 @@ type dynamicConfig[T any] struct {
 
 func newDynamicConfig[T any](name string, val T, apply func(T) bool, equal func(x, y T) bool) dynamicConfig[T] {
 	return dynamicConfig[T]{
-		cfgName: name,
-		current: val,
-		startup: val,
-		apply:   apply,
-		equal:   equal,
+		cfgName:   name,
+		current:   val,
+		startup:   val,
+		cfgOrigin: telemetry.OriginDefault,
+		apply:     apply,
+		equal:     equal,
 	}
 }
 

profiler/telemetry_test.go

@@ -26,7 +26,7 @@ func mockGlobalClient(client telemetry.Client) func() {
 // Test that the profiler sends the correct telemetry information
 func TestTelemetryEnabled(t *testing.T) {
 	t.Run("tracer start, profiler start", func(t *testing.T) {
-		telemetryClient := new(telemetrytest.MockClient)
+		telemetryClient := new(telemetrytest.RecordClient)
 		defer mockGlobalClient(telemetryClient)()
 
 		tracer.Start()
@@ -41,10 +41,9 @@ func TestTelemetryEnabled(t *testing.T) {
 
 		assert.True(t, telemetryClient.Products[telemetry.NamespaceProfilers])
 		assert.Contains(t, telemetryClient.Configuration, telemetry.Configuration{Name: "heap_profile_enabled", Value: true})
-		telemetryClient.AssertCalled(t, "ProductStarted", telemetry.NamespaceProfilers)
 	})
 	t.Run("only profiler start", func(t *testing.T) {
-		telemetryClient := new(telemetrytest.MockClient)
+		telemetryClient := new(telemetrytest.RecordClient)
 		defer mockGlobalClient(telemetryClient)()
 		Start(
 			WithProfileTypes(
@@ -55,6 +54,5 @@ func TestTelemetryEnabled(t *testing.T) {
 
 		assert.True(t, telemetryClient.Products[telemetry.NamespaceProfilers])
 		assert.Contains(t, telemetryClient.Configuration, telemetry.Configuration{Name: "heap_profile_enabled", Value: true})
-		telemetryClient.AssertCalled(t, "ProductStarted", telemetry.NamespaceProfilers)
 	})
 }