@@ -111,3 +111,82 @@ func TestNewClient_Error(t *testing.T) {
|
||||
_, err := NewClient(nil, errorOpt)
|
||||
require.EqualError(t, err, "foo")
|
||||
}
|
||||
|
||||
func TestClient_OnRequestComplemented(t *testing.T) {
|
||||
setup()
|
||||
defer teardown()
|
||||
|
||||
var i int
|
||||
cb := func(*http.Request, *http.Response) {
|
||||
i++
|
||||
}
|
||||
client.OnRequestCompleted(cb)
|
||||
|
||||
mux.HandleFunc("/api/v1/test", func(w http.ResponseWriter, r *http.Request) {
|
||||
require.Equal(t, http.MethodGet, r.Method)
|
||||
})
|
||||
|
||||
req, err := client.NewRequest(http.MethodGet, "api/v1/test", nil)
|
||||
require.NoError(t, err)
|
||||
|
||||
_, _ = client.Do(ctx, req, nil)
|
||||
require.Equal(t, 1, i)
|
||||
|
||||
_, _ = client.Do(ctx, req, nil)
|
||||
_, _ = client.Do(ctx, req, nil)
|
||||
_, _ = client.Do(ctx, req, nil)
|
||||
_, _ = client.Do(ctx, req, nil)
|
||||
require.Equal(t, 5, i)
|
||||
|
||||
_, _ = client.Do(ctx, req, nil)
|
||||
require.Equal(t, 6, i)
|
||||
}
|
||||
|
||||
func TestClient_JSONErrorResponse(t *testing.T) {
|
||||
setup()
|
||||
defer teardown()
|
||||
|
||||
mux.HandleFunc("/api/v1/test", func(w http.ResponseWriter, r *http.Request) {
|
||||
require.Equal(t, http.MethodGet, r.Method)
|
||||
fmt.Fprint(w, `{
|
||||
"json": {
|
||||
"errors": [
|
||||
[
|
||||
"TEST_ERROR",
|
||||
"this is a test error",
|
||||
"test field"
|
||||
]
|
||||
]
|
||||
}
|
||||
}`)
|
||||
})
|
||||
|
||||
req, err := client.NewRequest(http.MethodGet, "api/v1/test", nil)
|
||||
require.NoError(t, err)
|
||||
|
||||
resp, err := client.Do(ctx, req, nil)
|
||||
require.IsType(t, &JSONErrorResponse{}, err)
|
||||
require.EqualError(t, err, fmt.Sprintf(`GET %s/api/v1/test: 200 field "test field" caused TEST_ERROR: this is a test error`, server.URL))
|
||||
require.Equal(t, http.StatusOK, resp.StatusCode)
|
||||
}
|
||||
|
||||
func TestClient_ErrorResponse(t *testing.T) {
|
||||
setup()
|
||||
defer teardown()
|
||||
|
||||
mux.HandleFunc("/api/v1/test", func(w http.ResponseWriter, r *http.Request) {
|
||||
require.Equal(t, http.MethodGet, r.Method)
|
||||
w.WriteHeader(http.StatusForbidden)
|
||||
fmt.Fprint(w, `{
|
||||
"message": "error message"
|
||||
}`)
|
||||
})
|
||||
|
||||
req, err := client.NewRequest(http.MethodGet, "api/v1/test", nil)
|
||||
require.NoError(t, err)
|
||||
|
||||
resp, err := client.Do(ctx, req, nil)
|
||||
require.IsType(t, &ErrorResponse{}, err)
|
||||
require.EqualError(t, err, fmt.Sprintf(`GET %s/api/v1/test: 403 error message`, server.URL))
|
||||
require.Equal(t, http.StatusForbidden, resp.StatusCode)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user