lint

DataDog · Dec 20, 2024 · 918a260 · 918a260
1 parent 034b271
commit 918a260
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/pkg/fleet/installer/setup/djm/emr.go b/pkg/fleet/installer/setup/djm/emr.go
@@ -162,7 +162,7 @@ func setupEmrResourceManager(s *common.Setup, clusterName string) {
 }
 
 var executeCommandWithTimeout = func(s *common.Setup, command string, args ...string) (output []byte, err error) {
-	span, ctx := telemetry.StartSpanFromContext(s.Ctx, "setup.command")
+	span, _ := telemetry.StartSpanFromContext(s.Ctx, "setup.command")
 	span.SetResourceName(command)
 	defer func() { span.Finish(err) }()
 

diff --git a/pkg/fleet/installer/setup/djm/emr_test.go b/pkg/fleet/installer/setup/djm/emr_test.go
@@ -36,7 +36,10 @@ func TestSetupEmr(t *testing.T) {
 	originalExecuteCommand := executeCommandWithTimeout
 	defer func() { executeCommandWithTimeout = originalExecuteCommand }() // Restore original after test
 
-	executeCommandWithTimeout = func(s *common.Setup, command string, args ...string) ([]byte, error) {
+	executeCommandWithTimeout = func(s *common.Setup, command string, args ...string) (output []byte, err error) {
+		span, _ := telemetry.StartSpanFromContext(s.Ctx, "setup.command")
+		span.SetResourceName(command)
+		defer func() { span.Finish(err) }()
 		if command == "aws" && args[0] == "emr" && args[1] == "describe-cluster" {
 			return []byte(emrDescribeClusterResponse), nil
 		}