diff options
Diffstat (limited to 'pkg/web')
| -rw-r--r-- | pkg/web/server.go | 2 | ||||
| -rw-r--r-- | pkg/web/server_test.go | 12 |
2 files changed, 14 insertions, 0 deletions
diff --git a/pkg/web/server.go b/pkg/web/server.go index 2949309..e553920 100644 --- a/pkg/web/server.go +++ b/pkg/web/server.go @@ -23,6 +23,8 @@ func NewServer(storage db.Repository) *Server { func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) { switch r.URL.String() { + case "/health": + w.WriteHeader(http.StatusOK) case "/sparkles.json": switch r.Method { case "GET": diff --git a/pkg/web/server_test.go b/pkg/web/server_test.go index 0281867..a8a0e19 100644 --- a/pkg/web/server_test.go +++ b/pkg/web/server_test.go @@ -22,6 +22,7 @@ func TestServer(t *testing.T) { response := httptest.NewRecorder() request, err := http.NewRequest("GET", "/sparkles.json", nil) require.NoError(t, err) + NewServer(store).ServeHTTP(response, request) assert.Equal(t, http.StatusOK, response.Code) @@ -34,4 +35,15 @@ func TestServer(t *testing.T) { assert.Equal(t, "for helping me", items[0].Reason) }) }) + + t.Run("GET /health", func(t *testing.T) { + t.Run("returns OK", func(t *testing.T) { + response := httptest.NewRecorder() + request, err := http.NewRequest("GET", "/health", nil) + require.NoError(t, err) + + NewServer(db.NewRepository()).ServeHTTP(response, request) + assert.Equal(t, http.StatusOK, response.Code) + }) + }) } |