github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-04 18:00:08 +00:00
Code Issues Projects Releases Wiki Activity

apiserver: check that request fail after shutdown

Browse Source
This commit is contained in:
Dr. Stefan Schimanski 2018-12-19 12:41:58 +01:00
parent b18660eb6c
commit c0d8ad1578
1 changed files with 19 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
staging/src/k8s.io/apiserver/pkg/server/genericapiserver_test.go
View File

@ -517,19 +517,24 @@ func TestGracefulShutdown(t *testing.T) {
return handler
}
handler := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
wg.Done()
time.Sleep(2 * time.Second)
w.WriteHeader(http.StatusOK)
graceShutdown = true
})
s, err := config.Complete(nil).New("test", NewEmptyDelegate())
if err != nil {
t.Fatalf("Error in bringing up the server: %v", err)
}
s.Handler.NonGoRestfulMux.Handle("/test", handler)
twoSecondHandler := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
wg.Done()
time.Sleep(2 * time.Second)
w.WriteHeader(http.StatusOK)
graceShutdown = true
})
okHandler := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
wg.Done()
w.WriteHeader(http.StatusOK)
})
s.Handler.NonGoRestfulMux.Handle("/test", twoSecondHandler)
s.Handler.NonGoRestfulMux.Handle("/200", okHandler)
insecureServer := &http.Server{
Addr: "0.0.0.0:0",
@ -565,6 +570,12 @@ func TestGracefulShutdown(t *testing.T) {
// close stopCh after request sent to server to guarantee request handler is running.
wg.Wait()
close(stopCh)
time.Sleep(500 * time.Millisecond)
if _, err := http.Get("http://127.0.0.1:" + strconv.Itoa(serverPort) + "/200"); err == nil {
t.Errorf("Unexpected http success after stopCh was closed")
}
// wait for wait group handler finish
s.HandlerChainWaitGroup.Wait()