52b49f4838
Implement health.Healther in erratic and kubernetes plugin. The kubernetes' healtcheck is only performed on startup - i.e. turn healthy after the initial loading. Erratic follow the drop count: every query%drop turns the healthcheck unhealthy. Fixes: #985
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package health
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/coredns/coredns/plugin/erratic"
|
|
)
|
|
|
|
func TestHealth(t *testing.T) {
|
|
h := health{Addr: ":0"}
|
|
h.h = append(h.h, &erratic.Erratic{})
|
|
|
|
if err := h.Startup(); err != nil {
|
|
t.Fatalf("Unable to startup the health server: %v", err)
|
|
}
|
|
defer h.Shutdown()
|
|
|
|
// 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))
|
|
}
|
|
}
|