Return error & send error event on cancel/timeout

cmd/apply/cmdapply.go

@@ -91,11 +91,7 @@ type ApplyRunner struct {
 }
 
 func (r *ApplyRunner) RunE(cmd *cobra.Command, args []string) error {
-	// If specified, use command context.
 	ctx := cmd.Context()
-	if ctx == nil {
-		ctx = context.Background()
-	}
 	// If specified, cancel with timeout.
 	if r.timeout != 0 {
 		var cancel context.CancelFunc

cmd/destroy/cmddestroy.go

@@ -77,11 +77,7 @@ type DestroyRunner struct {
 }
 
 func (r *DestroyRunner) RunE(cmd *cobra.Command, args []string) error {
-	// If specified, use command context.
 	ctx := cmd.Context()
-	if ctx == nil {
-		ctx = context.Background()
-	}
 	// If specified, cancel with timeout.
 	if r.timeout != 0 {
 		var cancel context.CancelFunc

cmd/preview/cmdpreview.go

@@ -88,11 +88,7 @@ type PreviewRunner struct {
 
 // RunE is the function run from the cobra command.
 func (r *PreviewRunner) RunE(cmd *cobra.Command, args []string) error {
-	// If specified, use command context.
 	ctx := cmd.Context()
-	if ctx == nil {
-		ctx = context.Background()
-	}
 	// If specified, cancel with timeout.
 	if r.timeout != 0 {
 		var cancel context.CancelFunc

cmd/status/cmdstatus.go

@@ -127,7 +127,7 @@ func (r *StatusRunner) runE(cmd *cobra.Command, args []string) error {
 
 	// If the user has specified a timeout, we create a context with timeout,
 	// otherwise we create a context with cancel.
-	ctx := context.Background()
+	ctx := cmd.Context()
 	var cancel func()
 	if r.timeout != 0 {
 		ctx, cancel = context.WithTimeout(ctx, r.timeout)

cmd/status/cmdstatus_test.go

@@ -236,12 +236,15 @@ deployment.apps/foo is InProgress: inProgress
 				timeout:   tc.timeout,
 			}
 
-			cmd := &cobra.Command{}
+			cmd := &cobra.Command{
+				RunE: runner.runE,
+			}
 			cmd.SetIn(strings.NewReader(tc.input))
 			var buf bytes.Buffer
 			cmd.SetOut(&buf)
+			cmd.SetArgs([]string{})
 
-			err := runner.runE(cmd, []string{})
+			err := cmd.Execute()
 
 			if tc.expectedErrMsg != "" {
 				if !assert.Error(t, err) {

pkg/apply/applier_test.go

@@ -709,6 +709,13 @@ func TestApplierCancel(t *testing.T) {
 				// 		Type:      event.Finished,
 				// 	},
 				// },
+				{
+					// Error
+					EventType: event.ErrorType,
+					ErrorEvent: &testutil.ExpErrorEvent{
+						Err: context.DeadlineExceeded,
+					},
+				},
 			},
 		},
 		"completed with timeout": {

pkg/apply/destroyer_test.go

@@ -143,6 +143,13 @@ func TestDestroyerCancel(t *testing.T) {
 				},
 				// Inventory cannot be deleted, because the objects still exist,
 				// even tho they've been deleted (ex: blocked by finalizer).
+				{
+					// Error
+					EventType: event.ErrorType,
+					ErrorEvent: &testutil.ExpErrorEvent{
+						Err: context.DeadlineExceeded,
+					},
+				},
 			},
 		},
 		"completed with timeout": {

pkg/apply/taskrunner/runner.go

@@ -9,7 +9,6 @@ import (
 	"sort"
 	"time"
 
-	"k8s.io/klog/v2"
 	"sigs.k8s.io/cli-utils/pkg/apply/cache"
 	"sigs.k8s.io/cli-utils/pkg/apply/event"
 	"sigs.k8s.io/cli-utils/pkg/apply/poller"
@@ -257,7 +256,7 @@ func (b *baseRunner) run(ctx context.Context, taskQueue chan Task,
 		case <-doneCh:
 			doneCh = nil // Set doneCh to nil so we don't enter a busy loop.
 			abort = true
-			klog.V(3).Info("taskrunner cancelled by caller")
+			abortReason = ctx.Err() // always non-nil when doneCh is closed
 			completeIfWaitTask(currentTask, taskContext)
 		}
 	}

pkg/apply/taskrunner/runner_test.go

@@ -297,6 +297,7 @@ func TestBaseRunnerCancellation(t *testing.T) {
 				},
 			},
 			contextTimeout: 2 * time.Second,
+			expectedError:  context.Canceled,
 			expectedEventTypes: []event.Type{
 				event.ActionGroupType,
 				event.ApplyType,
@@ -315,6 +316,7 @@ func TestBaseRunnerCancellation(t *testing.T) {
 				},
 			},
 			contextTimeout: 2 * time.Second,
+			expectedError:  context.Canceled,
 			expectedEventTypes: []event.Type{
 				event.ActionGroupType,
 				event.ActionGroupType,