mirror of
https://github.com/coredns/coredns.git
synced 2025-10-27 08:14:18 -04:00
c39e5cd014
* plugin/health: add lameduck mode Add a way to configure lameduck more, i.e. set health to false, stop polling plugins. Then wait for a duration before shutting down. As the health middleware is configured early on in the plugin list, it will hold up all other shutdown, meaning we still answer queries. * Add New * More tests * golint * remove confusing text
78 lines
1.7 KiB
Go
78 lines
1.7 KiB
Go
package health
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/coredns/coredns/plugin/erratic"
|
|
)
|
|
|
|
func TestHealth(t *testing.T) {
|
|
h := newHealth(":0")
|
|
h.h = append(h.h, &erratic.Erratic{})
|
|
|
|
if err := h.OnStartup(); err != nil {
|
|
t.Fatalf("Unable to startup the health server: %v", err)
|
|
}
|
|
defer h.OnShutdown()
|
|
|
|
go func() {
|
|
<-h.pollstop
|
|
return
|
|
}()
|
|
|
|
// Reconstruct the http address based on the port allocated by operating system.
|
|
address := fmt.Sprintf("http://%s%s", h.ln.Addr().String(), path)
|
|
|
|
// Nothing set should return unhealthy
|
|
response, err := http.Get(address)
|
|
if err != nil {
|
|
t.Fatalf("Unable to query %s: %v", address, err)
|
|
}
|
|
if response.StatusCode != 503 {
|
|
t.Errorf("Invalid status code: expecting '503', got '%d'", response.StatusCode)
|
|
}
|
|
response.Body.Close()
|
|
|
|
h.poll()
|
|
|
|
response, err = http.Get(address)
|
|
if err != nil {
|
|
t.Fatalf("Unable to query %s: %v", address, err)
|
|
}
|
|
if response.StatusCode != 200 {
|
|
t.Errorf("Invalid status code: expecting '200', got '%d'", response.StatusCode)
|
|
}
|
|
content, err := ioutil.ReadAll(response.Body)
|
|
if err != nil {
|
|
t.Fatalf("Unable to get response body from %s: %v", address, err)
|
|
}
|
|
response.Body.Close()
|
|
|
|
if string(content) != ok {
|
|
t.Errorf("Invalid response body: expecting 'OK', got '%s'", string(content))
|
|
}
|
|
}
|
|
|
|
func TestHealthLameduck(t *testing.T) {
|
|
h := newHealth(":0")
|
|
h.lameduck = 250 * time.Millisecond
|
|
h.h = append(h.h, &erratic.Erratic{})
|
|
|
|
if err := h.OnStartup(); err != nil {
|
|
t.Fatalf("Unable to startup the health server: %v", err)
|
|
}
|
|
|
|
// Both these things are behind a sync.Once, fake reading from the channels.
|
|
go func() {
|
|
<-h.pollstop
|
|
<-h.stop
|
|
return
|
|
}()
|
|
|
|
h.OnShutdown()
|
|
}
|