Add GetReasonFromBody() that extracts reason from JSON
This commit is contained in:
@@ -17,8 +17,13 @@
|
|||||||
package http
|
package http
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"io/ioutil"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
"github.com/BBVA/kapow/internal/server/srverrors"
|
||||||
)
|
)
|
||||||
|
|
||||||
// GetReason returns the reason phrase part of an HTTP response
|
// GetReason returns the reason phrase part of an HTTP response
|
||||||
@@ -28,3 +33,22 @@ func GetReason(r *http.Response) string {
|
|||||||
}
|
}
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func GetReasonFromBody(r *http.Response) (string, error) {
|
||||||
|
body, err := ioutil.ReadAll(r.Body)
|
||||||
|
if err != nil {
|
||||||
|
return "", errors.New("error reading response's body")
|
||||||
|
}
|
||||||
|
|
||||||
|
reason := &srverrors.ServerErrMessage{}
|
||||||
|
err = json.Unmarshal(body, reason)
|
||||||
|
if err != nil {
|
||||||
|
return "", errors.New("error unmarshaling JSON")
|
||||||
|
}
|
||||||
|
|
||||||
|
if reason.Reason == "" {
|
||||||
|
return "", errors.New("no reason")
|
||||||
|
}
|
||||||
|
|
||||||
|
return reason.Reason, nil
|
||||||
|
}
|
||||||
|
|||||||
@@ -17,7 +17,9 @@
|
|||||||
package http
|
package http
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"io/ioutil"
|
||||||
nethttp "net/http"
|
nethttp "net/http"
|
||||||
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -62,3 +64,83 @@ func TestBehaveWithOddSizeStatusCode(t *testing.T) {
|
|||||||
t.Errorf("Unexpected reason found")
|
t.Errorf("Unexpected reason found")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestGetReasonFromBodyExtractsReasonFromJSON(t *testing.T) {
|
||||||
|
r := &nethttp.Response{
|
||||||
|
Status: "200 OK",
|
||||||
|
Body: ioutil.NopCloser(
|
||||||
|
strings.NewReader(
|
||||||
|
`{"reason": "Because reasons", "foo": "bar"}`,
|
||||||
|
),
|
||||||
|
),
|
||||||
|
}
|
||||||
|
|
||||||
|
reason, _ := GetReasonFromBody(r)
|
||||||
|
|
||||||
|
if reason != "Because reasons" {
|
||||||
|
t.Errorf(`reason mismatch, want "Because reasons", got %q`, reason)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetReasonFromBodyErrorsOnJSONWithNoReason(t *testing.T) {
|
||||||
|
r := &nethttp.Response{
|
||||||
|
Status: "200 OK",
|
||||||
|
Body: ioutil.NopCloser(
|
||||||
|
strings.NewReader(
|
||||||
|
`{"madness": "Because madness", "foo": "bar"}`,
|
||||||
|
),
|
||||||
|
),
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err := GetReasonFromBody(r)
|
||||||
|
|
||||||
|
if err == nil {
|
||||||
|
t.Error("error not reported")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetReasonFromBodyErrorsOnJSONWithEmptyReason(t *testing.T) {
|
||||||
|
r := &nethttp.Response{
|
||||||
|
Body: ioutil.NopCloser(
|
||||||
|
strings.NewReader(
|
||||||
|
`{"reason": "", "foo": "bar"}`,
|
||||||
|
),
|
||||||
|
),
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err := GetReasonFromBody(r)
|
||||||
|
|
||||||
|
if err == nil {
|
||||||
|
t.Error("error not reported")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetReasonFromBodyErrorsOnNoJSON(t *testing.T) {
|
||||||
|
r := &nethttp.Response{
|
||||||
|
Body: ioutil.NopCloser(
|
||||||
|
strings.NewReader(""),
|
||||||
|
),
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err := GetReasonFromBody(r)
|
||||||
|
|
||||||
|
if err == nil {
|
||||||
|
t.Error("error not reported")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetReasonFromBodyErrorsOnInvalidJSON(t *testing.T) {
|
||||||
|
r := &nethttp.Response{
|
||||||
|
Body: ioutil.NopCloser(
|
||||||
|
strings.NewReader(
|
||||||
|
`{"reason": "Because reasons", "cliffhanger...`,
|
||||||
|
),
|
||||||
|
),
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err := GetReasonFromBody(r)
|
||||||
|
|
||||||
|
if err == nil {
|
||||||
|
t.Error("error not reported")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user