mirror of
https://github.com/coredns/coredns.git
synced 2025-10-27 08:14:18 -04:00
48059a6c3e
* plugin/health: add 'overloaded metrics' Query our on health endpoint and record (and export as a metric) the time it takes. The Get has a 5s timeout, that, when reached, will set the metric duration to 5s. The actually call "I'm I overloaded" is left to an external entity. * README * golint and govet * and the tests
53 lines
1.3 KiB
Go
53 lines
1.3 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.OnStartup(); err != nil {
|
|
t.Fatalf("Unable to startup the health server: %v", err)
|
|
}
|
|
defer h.OnShutdown()
|
|
|
|
// 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))
|
|
}
|
|
}
|