fix: continue cleanup after shutdown deadline

main_test.go

@@ -70,6 +70,48 @@ func TestShutdownAllContinuesAfterTimeoutWithSharedDeadline(t *testing.T) {
 	}
 }
 
+func TestShutdownAllContinuesWhenStepIgnoresDeadline(t *testing.T) {
+	ctx, cancel := context.WithTimeout(context.Background(), 20*time.Millisecond)
+	defer cancel()
+
+	blocked := make(chan struct{})
+	defer close(blocked)
+	cleanupStarted := make(chan struct{})
+	rootCtx, rootCancel := context.WithCancel(context.Background())
+	result := make(chan error, 1)
+	go func() {
+		result <- shutdownAll(ctx, rootCancel,
+			shutdownStep{name: "blocked", run: func(context.Context) error {
+				<-blocked
+				return nil
+			}},
+			shutdownStep{name: "cleanup", run: func(context.Context) error {
+				close(cleanupStarted)
+				return nil
+			}},
+		)
+	}()
+
+	select {
+	case err := <-result:
+		if !errors.Is(err, context.DeadlineExceeded) {
+			t.Fatalf("shutdownAll() error = %v, want deadline exceeded", err)
+		}
+	case <-time.After(250 * time.Millisecond):
+		t.Fatal("shutdownAll did not enforce the shared deadline")
+	}
+	select {
+	case <-cleanupStarted:
+	default:
+		t.Fatal("cleanup after blocked step was not started")
+	}
+	select {
+	case <-rootCtx.Done():
+	default:
+		t.Fatal("root context was not canceled")
+	}
+}
+
 func recordShutdown(order *[]string, contexts *[]context.Context, name string, err error) func(context.Context) error {
 	return func(ctx context.Context) error {
 		*order = append(*order, name)