mirror of
https://github.com/tiennm99/openai-status-bot.git
synced 2026-09-04 02:18:39 +00:00
tiennm99
b3ab6317fe
The health endpoint returned 200 unconditionally from before dependencies were ready and never reflected MongoDB state. Handler now takes a readiness check; the endpoint returns 503 while starting or when MongoDB is unreachable and 200 only when ready.
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package health
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
func TestHandlerReturnsOKWhenCheckPasses(t *testing.T) {
|
|
req := httptest.NewRequest(http.MethodGet, Path, nil)
|
|
res := httptest.NewRecorder()
|
|
|
|
Handler(func(context.Context) error { return nil }).ServeHTTP(res, req)
|
|
|
|
if res.Code != http.StatusOK {
|
|
t.Fatalf("status = %d, want %d", res.Code, http.StatusOK)
|
|
}
|
|
if res.Body.String() != "ok\n" {
|
|
t.Fatalf("body = %q, want ok", res.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestHandlerNilCheckIsHealthy(t *testing.T) {
|
|
req := httptest.NewRequest(http.MethodGet, Path, nil)
|
|
res := httptest.NewRecorder()
|
|
|
|
Handler(nil).ServeHTTP(res, req)
|
|
|
|
if res.Code != http.StatusOK {
|
|
t.Fatalf("status = %d, want %d", res.Code, http.StatusOK)
|
|
}
|
|
}
|
|
|
|
func TestHandlerReturnsUnavailableWhenCheckFails(t *testing.T) {
|
|
req := httptest.NewRequest(http.MethodGet, Path, nil)
|
|
res := httptest.NewRecorder()
|
|
|
|
Handler(func(context.Context) error { return errors.New("not ready") }).ServeHTTP(res, req)
|
|
|
|
if res.Code != http.StatusServiceUnavailable {
|
|
t.Fatalf("status = %d, want %d", res.Code, http.StatusServiceUnavailable)
|
|
}
|
|
}
|