Merge pull request #6 from hellofresh/feature/reset

Added checks reset function

README.md

@@ -46,8 +46,8 @@ import (
   "net/http"
   "time"
 
+  "github.com/go-chi/chi"
   "github.com/hellofresh/health-go"
-  "github.com/pressly/chi"
 )
 
 func main() {

health.go

@@ -15,6 +15,8 @@ const (
 	statusOK                 = "OK"
 	statusPartiallyAvailable = "Partially Available"
 	statusUnavailable        = "Unavailable"
+
+	failureTimeout = "Timeout during health check"
 )
 
 // Config carries the parameters to run the check.
@@ -58,6 +60,14 @@ func Register(c Config) {
 	checks = append(checks, c)
 }
 
+// Reset unregisters all previously set check configs
+func Reset() {
+	mu.Lock()
+	defer mu.Unlock()
+
+	checks = []Config{}
+}
+
 // Handler returns an HTTP handler (http.HandlerFunc).
 func Handler() http.Handler {
 	return http.HandlerFunc(HandlerFunc)
@@ -79,7 +89,7 @@ func HandlerFunc(w http.ResponseWriter, r *http.Request) {
 		for {
 			select {
 			case <-time.After(c.Timeout):
-				failures[c.Name] = "Timeout during health check"
+				failures[c.Name] = failureTimeout
 				setStatus(&status, c.SkipOnErr)
 				break loop
 			case res := <-resChan:

health_test.go

@@ -6,10 +6,11 @@ import (
 	"net/http"
 	"net/http/httptest"
 	"testing"
+	"time"
 )
 
 const (
-	checkErr = "Failed duriung rabbitmq health check"
+	checkErr = "Failed during RabbitMQ health check"
 )
 
 func TestHealthHandler(t *testing.T) {
@@ -30,6 +31,16 @@ func TestHealthHandler(t *testing.T) {
 		Check: func() error { return nil },
 	})
 
+	Register(Config{
+		Name:      "snail-service",
+		SkipOnErr: true,
+		Timeout:   time.Second * 1,
+		Check: func() error {
+			time.Sleep(time.Second * 2)
+			return nil
+		},
+	})
+
 	h := http.Handler(Handler())
 	h.ServeHTTP(res, req)
 
@@ -57,6 +68,15 @@ func TestHealthHandler(t *testing.T) {
 	}
 
 	if f["rabbitmq"] != checkErr {
-		t.Errorf("body returned wrong status: got %s want %s", failure, checkErr)
+		t.Errorf("body returned wrong status for rabbitmq: got %s want %s", failure, checkErr)
+	}
+
+	if f["snail-service"] != failureTimeout {
+		t.Errorf("body returned wrong status for snail-service: got %s want %s", failure, failureTimeout)
+	}
+
+	Reset()
+	if len(checks) != 0 {
+		t.Errorf("checks lenght differes from zero: got %d", len(checks))
 	}
 }