Use the rove package ServerStatus type

This commit is contained in:
Marc Di Luzio 2020-05-29 23:05:37 +01:00
parent a6d86c2581
commit 522e8205b3
2 changed files with 5 additions and 7 deletions

View file

@ -5,6 +5,7 @@ import (
"net/http"
"github.com/gorilla/mux"
"github.com/mdiluz/rove/pkg/rove"
)
func NewRouter() (router *mux.Router) {
@ -17,14 +18,9 @@ func NewRouter() (router *mux.Router) {
return
}
// RouterStatus is a struct that contains information on the status of the server
type RouterStatus struct {
Ready bool `json:"ready"`
}
// HandleStatus handles HTTP requests to the /status endpoint
func HandleStatus(w http.ResponseWriter, r *http.Request) {
var status = RouterStatus{
var status = rove.ServerStatus{
Ready: true,
}

View file

@ -5,6 +5,8 @@ import (
"net/http"
"net/http/httptest"
"testing"
"github.com/mdiluz/rove/pkg/rove"
)
func TestHandleStatus(t *testing.T) {
@ -13,7 +15,7 @@ func TestHandleStatus(t *testing.T) {
HandleStatus(response, request)
var status RouterStatus
var status rove.ServerStatus
json.NewDecoder(response.Body).Decode(&status)
if status.Ready != true {