fix: goroutine leak when timeout

Author: andyzhangx

go.mod

@@ -27,6 +27,7 @@ require (
 	github.com/onsi/gomega v1.36.2
 	github.com/rubiojr/go-vhd v0.0.0-20200706105327-02e210299021
 	github.com/stretchr/testify v1.10.0
+	go.uber.org/goleak v1.3.0
 	go.uber.org/mock v0.5.0
 	golang.org/x/net v0.37.0
 	golang.org/x/sys v0.31.0

pkg/util/util.go

@@ -217,7 +217,7 @@ type TimeoutFunc func() (err error)
 // WaitUntilTimeout waits for the exec function to complete or return timeout error
 func WaitUntilTimeout(timeout time.Duration, execFunc ExecFunc, timeoutFunc TimeoutFunc) error {
 	// Create a channel to receive the result of the azcopy exec function
-	done := make(chan bool)
+	done := make(chan bool, 1)
 	var err error
 
 	// Start exec function in a goroutine

pkg/util/util_test.go

@@ -22,6 +22,7 @@ import (
 	"testing"
 	"time"
 
+	"go.uber.org/goleak"
 	gomock "go.uber.org/mock/gomock"
 )
 
@@ -351,6 +352,7 @@ func TestParseAzcopyJobShow(t *testing.T) {
 }
 
 func TestWaitUntilTimeout(t *testing.T) {
+	defer goleak.VerifyNone(t)
 	tests := []struct {
 		desc        string
 		timeout     time.Duration
@@ -377,6 +379,7 @@ func TestWaitUntilTimeout(t *testing.T) {
 				return nil
 			},
 			timeoutFunc: func() error {
+				time.Sleep(2 * time.Second)
 				return fmt.Errorf("timeout error")
 			},
 			expectedErr: fmt.Errorf("timeout error"),