From 4cd4539c71919a322180713c225fe23edf0eb12e Mon Sep 17 00:00:00 2001 From: Kiran Kamineni Date: Thu, 14 Mar 2019 15:38:13 -0700 Subject: Use httptest instead of http in unit tests Use httptest instead of http in unit tests similar to: https://golang.org/pkg/net/http/httptest/#example_ResponseRecorder Update empty body checking to account for change Issue-ID: MULTICLOUD-545 Change-Id: Ib9775078c2c9ae2878b714363b569d8d79bd7698 Signed-off-by: Kiran Kamineni --- src/k8splugin/api/profilehandler.go | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) (limited to 'src/k8splugin/api/profilehandler.go') diff --git a/src/k8splugin/api/profilehandler.go b/src/k8splugin/api/profilehandler.go index 86e0d47f..267dae0e 100644 --- a/src/k8splugin/api/profilehandler.go +++ b/src/k8splugin/api/profilehandler.go @@ -18,6 +18,7 @@ package api import ( "encoding/json" + "io" "io/ioutil" "k8splugin/internal/rb" "net/http" @@ -37,13 +38,12 @@ type rbProfileHandler struct { func (h rbProfileHandler) createHandler(w http.ResponseWriter, r *http.Request) { var v rb.Profile - if r.Body == nil { + err := json.NewDecoder(r.Body).Decode(&v) + switch { + case err == io.EOF: http.Error(w, "Empty body", http.StatusBadRequest) return - } - - err := json.NewDecoder(r.Body).Decode(&v) - if err != nil { + case err != nil: http.Error(w, err.Error(), http.StatusUnprocessableEntity) return } @@ -80,17 +80,17 @@ func (h rbProfileHandler) uploadHandler(w http.ResponseWriter, r *http.Request) vars := mux.Vars(r) uuid := vars["rbpID"] - if r.Body == nil { - http.Error(w, "Empty Body", http.StatusBadRequest) - return - } - inpBytes, err := ioutil.ReadAll(r.Body) if err != nil { http.Error(w, "Unable to read body", http.StatusBadRequest) return } + if len(inpBytes) == 0 { + http.Error(w, "Empty body", http.StatusBadRequest) + return + } + err = h.client.Upload(uuid, inpBytes) if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) -- cgit 1.2.3-korg