fix: revert to previous behaviod polling forever

The ctx passed into WaitForTaskToSucceed is only used to cancel HTTP reqests and not to cancel the wait.

pkg/client/status.go

@@ -33,9 +33,10 @@ const (
 )
 
 // WaitForTaskToSucceed will poll indefinitely every 2 seconds for the task with uuid to have status of "SUCCEEDED".
-// Returns an error from GetTaskStatus or a timeout error if the context is cancelled.
+// The polling will not stop if the ctx is cancelled, it's only used for HTTP requests in the client.
+// WaitForTaskToSucceed will exit immediately on an error getting the task.
 func WaitForTaskToSucceed(ctx context.Context, conn *nutanixClientV3.Client, uuid string) error {
-	return wait.PollImmediateInfiniteWithContext(ctx, pollingInterval, func(ctx context.Context) (done bool, err error) {
+	return wait.PollImmediateInfinite(pollingInterval, func() (done bool, err error) {
 		status, getErr := GetTaskStatus(ctx, conn, uuid)
 		return status == statusSucceeded, getErr
 	})

pkg/client/status_test.go

@@ -8,7 +8,6 @@ import (
 	"time"
 
 	"github.com/stretchr/testify/assert"
-	"k8s.io/apimachinery/pkg/util/wait"
 
 	nutanixtestclient "github.com/nutanix-cloud-native/cluster-api-provider-nutanix/test/helpers/prism-go-client/v3"
 )
@@ -127,7 +126,7 @@ func Test_WaitForTaskCompletion(t *testing.T) {
 				fmt.Fprint(w, `{"status": "PENDING"}`)
 			},
 			ctx:         ctx,
-			expectedErr: wait.ErrWaitTimeout,
+			expectedErr: context.DeadlineExceeded,
 		},
 	}
 	for _, tt := range tests {