forked from TrueCloudLab/distribution
cd6482ecb8
If this slice ends up empty after parsing the HTTP response body, it means the body is not well-formed. We've probably encountered an error message produced by something that uses a different JSON schema, or an error that just happens to validate as JSON. An empty errcode.Errors slice is not a very useful thing to return, since its Error() output is just `<nil>`. Detect this case, and instend return an UnexpectedHTTPResponseError. Signed-off-by: Aaron Lehmann <aaron.lehmann@docker.com>
104 lines
3.2 KiB
Go
104 lines
3.2 KiB
Go
package client
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
type nopCloser struct {
|
|
io.Reader
|
|
}
|
|
|
|
func (nopCloser) Close() error { return nil }
|
|
|
|
func TestHandleErrorResponse401ValidBody(t *testing.T) {
|
|
json := "{\"errors\":[{\"code\":\"UNAUTHORIZED\",\"message\":\"action requires authentication\"}]}"
|
|
response := &http.Response{
|
|
Status: "401 Unauthorized",
|
|
StatusCode: 401,
|
|
Body: nopCloser{bytes.NewBufferString(json)},
|
|
}
|
|
err := HandleErrorResponse(response)
|
|
|
|
expectedMsg := "unauthorized: action requires authentication"
|
|
if !strings.Contains(err.Error(), expectedMsg) {
|
|
t.Errorf("Expected \"%s\", got: \"%s\"", expectedMsg, err.Error())
|
|
}
|
|
}
|
|
|
|
func TestHandleErrorResponse401WithInvalidBody(t *testing.T) {
|
|
json := "{invalid json}"
|
|
response := &http.Response{
|
|
Status: "401 Unauthorized",
|
|
StatusCode: 401,
|
|
Body: nopCloser{bytes.NewBufferString(json)},
|
|
}
|
|
err := HandleErrorResponse(response)
|
|
|
|
expectedMsg := "unauthorized: authentication required"
|
|
if !strings.Contains(err.Error(), expectedMsg) {
|
|
t.Errorf("Expected \"%s\", got: \"%s\"", expectedMsg, err.Error())
|
|
}
|
|
}
|
|
|
|
func TestHandleErrorResponseExpectedStatusCode400ValidBody(t *testing.T) {
|
|
json := "{\"errors\":[{\"code\":\"DIGEST_INVALID\",\"message\":\"provided digest does not match\"}]}"
|
|
response := &http.Response{
|
|
Status: "400 Bad Request",
|
|
StatusCode: 400,
|
|
Body: nopCloser{bytes.NewBufferString(json)},
|
|
}
|
|
err := HandleErrorResponse(response)
|
|
|
|
expectedMsg := "digest invalid: provided digest does not match"
|
|
if !strings.Contains(err.Error(), expectedMsg) {
|
|
t.Errorf("Expected \"%s\", got: \"%s\"", expectedMsg, err.Error())
|
|
}
|
|
}
|
|
|
|
func TestHandleErrorResponseExpectedStatusCode404EmptyErrorSlice(t *testing.T) {
|
|
json := `{"randomkey": "randomvalue"}`
|
|
response := &http.Response{
|
|
Status: "404 Not Found",
|
|
StatusCode: 404,
|
|
Body: nopCloser{bytes.NewBufferString(json)},
|
|
}
|
|
err := HandleErrorResponse(response)
|
|
|
|
expectedMsg := `error parsing HTTP response: no error details found in HTTP response body: "{\"randomkey\": \"randomvalue\"}"`
|
|
if !strings.Contains(err.Error(), expectedMsg) {
|
|
t.Errorf("Expected \"%s\", got: \"%s\"", expectedMsg, err.Error())
|
|
}
|
|
}
|
|
|
|
func TestHandleErrorResponseExpectedStatusCode404InvalidBody(t *testing.T) {
|
|
json := "{invalid json}"
|
|
response := &http.Response{
|
|
Status: "404 Not Found",
|
|
StatusCode: 404,
|
|
Body: nopCloser{bytes.NewBufferString(json)},
|
|
}
|
|
err := HandleErrorResponse(response)
|
|
|
|
expectedMsg := "error parsing HTTP response: invalid character 'i' looking for beginning of object key string: \"{invalid json}\""
|
|
if !strings.Contains(err.Error(), expectedMsg) {
|
|
t.Errorf("Expected \"%s\", got: \"%s\"", expectedMsg, err.Error())
|
|
}
|
|
}
|
|
|
|
func TestHandleErrorResponseUnexpectedStatusCode501(t *testing.T) {
|
|
response := &http.Response{
|
|
Status: "501 Not Implemented",
|
|
StatusCode: 501,
|
|
Body: nopCloser{bytes.NewBufferString("{\"Error Encountered\" : \"Function not implemented.\"}")},
|
|
}
|
|
err := HandleErrorResponse(response)
|
|
|
|
expectedMsg := "received unexpected HTTP status: 501 Not Implemented"
|
|
if !strings.Contains(err.Error(), expectedMsg) {
|
|
t.Errorf("Expected \"%s\", got: \"%s\"", expectedMsg, err.Error())
|
|
}
|
|
}
|