Merge pull request #73516 from mfojtik/fix-post-hook-test

staging/src/k8s.io/apiserver/pkg/server/config_test.go

@@ -56,7 +56,9 @@ func TestNewWithDelegate(t *testing.T) {
 		w.WriteHeader(http.StatusForbidden)
 	})
 
+	delegatePostStartHookChan := make(chan struct{})
 	delegateServer.AddPostStartHook("delegate-post-start-hook", func(context PostStartHookContext) error {
+		defer close(delegatePostStartHookChan)
 		return nil
 	})
 
@@ -82,7 +84,9 @@ func TestNewWithDelegate(t *testing.T) {
 		w.WriteHeader(http.StatusUnauthorized)
 	})
 
+	wrappingPostStartHookChan := make(chan struct{})
 	wrappingServer.AddPostStartHook("wrapping-post-start-hook", func(context PostStartHookContext) error {
+		defer close(wrappingPostStartHookChan)
 		return nil
 	})
 
@@ -94,6 +98,10 @@ func TestNewWithDelegate(t *testing.T) {
 	server := httptest.NewServer(wrappingServer.Handler)
 	defer server.Close()
 
+	// Wait for the hooks to finish before checking the response
+	<-delegatePostStartHookChan
+	<-wrappingPostStartHookChan
+
 	checkPath(server.URL, http.StatusOK, `{
   "paths": [
     "/apis",