mw/health: add reload test (#980)

Start CoreDNS instance in call Restart() on it.
This commit is contained in:
Miek Gieben
2017-08-26 17:39:43 +01:00
committed by GitHub
parent afad8abef3
commit 1bb836b793

View File

@@ -0,0 +1,49 @@
package test
import (
"io/ioutil"
"log"
"net/http"
"testing"
)
func TestHealthReload(t *testing.T) {
log.SetOutput(ioutil.Discard)
// Corefile with for example without proxy section.
corefile := `example.org:0 {
health localhost:35080
}
`
i, err := CoreDNSServer(corefile)
if err != nil {
t.Fatalf("Could not get CoreDNS serving instance: %s", err)
}
resp, err := http.Get("http://localhost:35080/health")
if err != nil {
t.Fatalf("Could not get health: %s", err)
}
body, err := ioutil.ReadAll(resp.Body)
if x := string(body); x != "OK" {
t.Fatalf("Expect OK, got %s", x)
}
resp.Body.Close()
i, err = i.Restart(NewInput(corefile))
if err != nil {
t.Fatalf("Could not restart CoreDNS serving instance: %s", err)
}
defer i.Stop()
resp, err = http.Get("http://localhost:35080/health")
if err != nil {
t.Fatalf("Could not get health: %s", err)
}
body, err = ioutil.ReadAll(resp.Body)
if x := string(body); x != "OK" {
t.Fatalf("Expect OK, got %s", x)
}
resp.Body.Close()
}