fix: respect cancellation after task action returns

workspace/task/runner.go

@@ -51,6 +51,9 @@ func (r *Runner) Run(ctx context.Context) {
 			task.mu.Unlock()
 
 			err := task.Action(ctx)
+			if err == nil && ctx.Err() != nil {
+				err = ctx.Err()
+			}
 
 			task.mu.Lock()
 			task.FinishedAt = time.Now()
@@ -66,4 +69,4 @@ func (r *Runner) Run(ctx context.Context) {
 		}(t)
 	}
 	wg.Wait()
-}
+}

workspace/task/runner_test.go

@@ -98,4 +98,24 @@ func TestRunner_Run_Cancel(t *testing.T) {
 	if err := t1.GetErr(); err != context.Canceled {
 		t.Errorf("expected t1 error to be context.Canceled, got %v", err)
 	}
-}
+}
+
+func TestRunner_Run_CancelDuringAction(t *testing.T) {
+	runner := NewRunner()
+	ctx, cancel := context.WithCancel(context.Background())
+
+	task := NewTask("cancel-during-action", func(ctx context.Context) error {
+		cancel()
+		return nil
+	})
+
+	runner.AddTask(task)
+	runner.Run(ctx)
+
+	if state := task.GetState(); state != StateFailed {
+		t.Errorf("expected task to fail after context cancellation, got %s", state)
+	}
+	if err := task.GetErr(); err != context.Canceled {
+		t.Errorf("expected task error to be context.Canceled, got %v", err)
+	}
+}