Verify content type on getRoute method
This commit is contained in:
@@ -86,7 +86,6 @@ func addRoute(res http.ResponseWriter, req *http.Request) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
if route.Method == "" {
|
if route.Method == "" {
|
||||||
spec / test / features / control / get / success.feature
|
|
||||||
res.WriteHeader(http.StatusUnprocessableEntity)
|
res.WriteHeader(http.StatusUnprocessableEntity)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -420,6 +420,26 @@ func TestGetRouteReturns404sWhenRouteDoesntExist(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestGetRouteSetsCorrctContentType(t *testing.T) {
|
||||||
|
handler := mux.NewRouter()
|
||||||
|
handler.HandleFunc("/routes/{id}", getRoute).
|
||||||
|
Methods("GET")
|
||||||
|
r := httptest.NewRequest(http.MethodGet, "/routes/FOO", nil)
|
||||||
|
w := httptest.NewRecorder()
|
||||||
|
user.Routes.Append(model.Route{ID: "FOO"})
|
||||||
|
|
||||||
|
handler.ServeHTTP(w, r)
|
||||||
|
|
||||||
|
resp := w.Result()
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
t.Errorf("HTTP status mismatch. Expected: %d, got: %d", http.StatusOK, resp.StatusCode)
|
||||||
|
}
|
||||||
|
|
||||||
|
if hVal := resp.Header.Get("Content-Type"); hVal != "application/json" {
|
||||||
|
t.Errorf(`Route mismatch. Expected: "application/json". Got: %s`, hVal)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestGetRouteReturnsTheRequestedRoute(t *testing.T) {
|
func TestGetRouteReturnsTheRequestedRoute(t *testing.T) {
|
||||||
handler := mux.NewRouter()
|
handler := mux.NewRouter()
|
||||||
handler.HandleFunc("/routes/{id}", getRoute).
|
handler.HandleFunc("/routes/{id}", getRoute).
|
||||||
@@ -427,6 +447,7 @@ func TestGetRouteReturnsTheRequestedRoute(t *testing.T) {
|
|||||||
r := httptest.NewRequest(http.MethodGet, "/routes/FOO", nil)
|
r := httptest.NewRequest(http.MethodGet, "/routes/FOO", nil)
|
||||||
w := httptest.NewRecorder()
|
w := httptest.NewRecorder()
|
||||||
user.Routes.Append(model.Route{ID: "FOO"})
|
user.Routes.Append(model.Route{ID: "FOO"})
|
||||||
|
|
||||||
handler.ServeHTTP(w, r)
|
handler.ServeHTTP(w, r)
|
||||||
|
|
||||||
resp := w.Result()
|
resp := w.Result()
|
||||||
|
|||||||
Reference in New Issue
Block a user