Add GetReasonFromBody() that extracts reason from JSON

This commit is contained in:
pancho horrillo
2019-11-19 20:26:00 +01:00
parent 28f4b969f9
commit e7619238ae
2 changed files with 106 additions and 0 deletions
+82
View File
@@ -17,7 +17,9 @@
package http
import (
"io/ioutil"
nethttp "net/http"
"strings"
"testing"
)
@@ -62,3 +64,83 @@ func TestBehaveWithOddSizeStatusCode(t *testing.T) {
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")
}
}