client_test.go

package httprequest_test

import (
	"context"
	"fmt"
	"io"
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"reflect"
	"regexp"
	"strings"
	"testing"

	qt "github.com/frankban/quicktest"
	"github.com/julienschmidt/httprouter"
	"gopkg.in/errgo.v1"

	"gopkg.in/httprequest.v1"
)

var callTests = []struct {
	about       string
	client      httprequest.Client
	req         interface{}
	requestUUID string
	expectError string
	assertError func(c *qt.C, err error)
	expectResp  interface{}
}{{
	about: "GET success",
	req: &chM1Req{
		P: "hello",
	},
	expectResp: &chM1Resp{"hello"},
}, {
	about: "GET with nil response",
	req: &chM1Req{
		P: "hello",
	},
}, {
	about: "POST success",
	req: &chM2Req{
		P:    "hello",
		Body: struct{ I int }{999},
	},
	expectResp: &chM2Resp{"hello", 999},
}, {
	about:       "GET marshal error",
	req:         123,
	expectError: `type is not pointer to struct`,
}, {
	about: "error response",
	req: &chInvalidM2Req{
		P:    "hello",
		Body: struct{ I bool }{true},
	},
	expectError: `Post http:.*: cannot unmarshal parameters: cannot unmarshal into field Body: cannot unmarshal request body: json: cannot unmarshal .*`,
	assertError: func(c *qt.C, err error) {
		c.Assert(errgo.Cause(err), qt.Satisfies, isRemoteError)
		err1 := errgo.Cause(err).(*httprequest.RemoteError)
		c.Assert(err1.Code, qt.Equals, "bad request")
		c.Assert(err1.Message, qt.Matches, `cannot unmarshal parameters: cannot unmarshal into field Body: cannot unmarshal request body: json: cannot unmarshal .*`)
	},
}, {
	about: "error unmarshaler returns nil",
	client: httprequest.Client{
		UnmarshalError: func(*http.Response) error {
			return nil
		},
	},
	req:         &chM3Req{},
	expectError: `Get http://.*/m3: unexpected HTTP response status: 500 Internal Server Error`,
}, {
	about: "unexpected redirect",
	req: &chM2RedirectM2Req{
		Body: struct{ I int }{999},
	},
	expectResp: &chM2Resp{"foo", 999},
}, {
	about:       "bad content in successful response",
	req:         &chM4Req{},
	expectResp:  new(int),
	expectError: `Get http://.*/m4: unexpected content type text/plain; want application/json; content: bad response`,
	assertError: func(c *qt.C, err error) {
		err1, ok := errgo.Cause(err).(*httprequest.DecodeResponseError)
		c.Assert(ok, qt.Equals, true, qt.Commentf("error not of type *httprequest.DecodeResponseError (%T)", errgo.Cause(err)))
		c.Assert(err1.Response, qt.Not(qt.IsNil))
		data, err := ioutil.ReadAll(err1.Response.Body)
		c.Assert(err, qt.Equals, nil)
		c.Assert(string(data), qt.Equals, "bad response")
	},
}, {
	about:       "bad content in error response",
	req:         &chM5Req{},
	expectResp:  new(int),
	expectError: `Get http://.*/m5: cannot unmarshal error response \(status 418 I'm a teapot\): unexpected content type text/plain; want application/json; content: bad error value`,
	assertError: func(c *qt.C, err error) {
		err1, ok := errgo.Cause(err).(*httprequest.DecodeResponseError)
		c.Assert(ok, qt.Equals, true, qt.Commentf("error not of type *httprequest.DecodeResponseError (%T)", errgo.Cause(err)))
		c.Assert(err1.Response, qt.Not(qt.IsNil))
		data, err := ioutil.ReadAll(err1.Response.Body)
		c.Assert(err, qt.Equals, nil)
		c.Assert(string(data), qt.Equals, "bad error value")
		c.Assert(err1.Response.StatusCode, qt.Equals, http.StatusTeapot)
	},
}, {
	about: "doer with context",
	client: httprequest.Client{
		Doer: doerWithContextFunc(func(ctx context.Context, req *http.Request) (*http.Response, error) {
			if ctx == nil {
				panic("Do called when DoWithContext expected")
			}
			return http.DefaultClient.Do(req.WithContext(ctx))
		}),
	},
	req: &chM2Req{
		P:    "hello",
		Body: struct{ I int }{999},
	},
	expectResp: &chM2Resp{"hello", 999},
}, {
	about: "doer with context and body",
	client: httprequest.Client{
		Doer: doerWithContextFunc(func(ctx context.Context, req *http.Request) (*http.Response, error) {
			if ctx == nil {
				panic("Do called when DoWithContext expected")
			}
			return http.DefaultClient.Do(req.WithContext(ctx))
		}),
	},
	req: &chM2Req{
		P:    "hello",
		Body: struct{ I int }{999},
	},
	expectResp: &chM2Resp{"hello", 999},
}, {
	about: "doer with context and body but no body",
	client: httprequest.Client{
		Doer: doerWithContextFunc(func(ctx context.Context, req *http.Request) (*http.Response, error) {
			if ctx == nil {
				panic("Do called when DoWithContext expected")
			}
			return http.DefaultClient.Do(req.WithContext(ctx))
		}),
	},
	req: &chM1Req{
		P: "hello",
	},
	expectResp: &chM1Resp{"hello"},
}}

func TestCall(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	srv := newServer()
	c.Defer(srv.Close)

	for _, test := range callTests {
		c.Run(test.about, func(c *qt.C) {
			var resp interface{}
			if test.expectResp != nil {
				resp = reflect.New(reflect.TypeOf(test.expectResp).Elem()).Interface()
			}
			client := test.client
			client.BaseURL = srv.URL
			ctx := context.Background()
			err := client.Call(ctx, test.req, resp)
			if test.expectError != "" {
				c.Logf("err %v", errgo.Details(err))
				c.Check(err, qt.ErrorMatches, test.expectError)
				if test.assertError != nil {
					test.assertError(c, err)
				}
				return
			}
			c.Assert(err, qt.Equals, nil)
			c.Assert(resp, qt.DeepEquals, test.expectResp)
		})
	}
}

func TestCallURLNoRequestPath(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	srv := newServer()
	c.Defer(srv.Close)

	var client httprequest.Client
	req := struct {
		httprequest.Route `httprequest:"GET"`
		chM1Req
	}{
		chM1Req: chM1Req{
			P: "hello",
		},
	}
	var resp chM1Resp
	err := client.CallURL(context.Background(), srv.URL+"/m1/:P", &req, &resp)
	c.Assert(err, qt.Equals, nil)
	c.Assert(resp, qt.DeepEquals, chM1Resp{"hello"})
}

func mustNewRequest(url string, method string, body io.Reader) *http.Request {
	return mustNewRequestWithHeader(url, method, body, http.Header{
		"Content-Type": []string{"application/json"},
	})
}

func mustNewRequestWithHeader(url string, method string, body io.Reader, hdr http.Header) *http.Request {
	req, err := http.NewRequest(method, url, body)
	if err != nil {
		panic(err)
	}
	for k, v := range hdr {
		req.Header[k] = append(req.Header[k], v...)
	}
	return req
}

var doTests = []struct {
	about       string
	client      httprequest.Client
	request     *http.Request
	requestUUID string

	expectError string
	expectCause interface{}
	expectResp  interface{}
}{{
	about:      "GET success",
	request:    mustNewRequest("/m1/hello", "GET", nil),
	expectResp: &chM1Resp{"hello"},
}, {
	about:   "appendURL error",
	request: mustNewRequest("/m1/hello", "GET", nil),
	client: httprequest.Client{
		BaseURL: ":::",
	},
	expectError: `cannot parse ":::": parse "?:::"?: missing protocol scheme`,
}, {
	about: "Do returns error",
	client: httprequest.Client{
		Doer: doerFunc(func(req *http.Request) (*http.Response, error) {
			return nil, errgo.Newf("an error")
		}),
	},
	request:     mustNewRequest("/m2/foo", "POST", strings.NewReader(`{"I": 999}`)),
	expectError: "Post http://.*/m2/foo: an error",
}, {
	about: "doer with context",
	client: httprequest.Client{
		Doer: doerWithContextFunc(func(ctx context.Context, req *http.Request) (*http.Response, error) {
			if ctx == nil {
				panic("Do called when DoWithContext expected")
			}
			return http.DefaultClient.Do(req.WithContext(ctx))
		}),
	},
	request:    mustNewRequest("/m2/foo", "POST", strings.NewReader(`{"I": 999}`)),
	expectResp: &chM2Resp{"foo", 999},
}}

func newInt64(i int64) *int64 {
	return &i
}

func TestDo(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	srv := newServer()
	c.Defer(srv.Close)

	for _, test := range doTests {
		test := test
		c.Run(test.about, func(c *qt.C) {
			var resp interface{}
			if test.expectResp != nil {
				resp = reflect.New(reflect.TypeOf(test.expectResp).Elem()).Interface()
			}
			client := test.client
			if client.BaseURL == "" {
				client.BaseURL = srv.URL
			}
			ctx := context.Background()
			err := client.Do(ctx, test.request, resp)
			if test.expectError != "" {
				c.Assert(err, qt.ErrorMatches, test.expectError)
				if test.expectCause != nil {
					c.Assert(errgo.Cause(err), qt.DeepEquals, test.expectCause)
				}
				return
			}
			c.Assert(err, qt.Equals, nil)
			c.Assert(resp, qt.DeepEquals, test.expectResp)
		})
	}
}

func TestDoWithHTTPReponse(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	srv := newServer()
	c.Defer(srv.Close)

	client := &httprequest.Client{
		BaseURL: srv.URL,
	}
	var resp *http.Response
	err := client.Get(context.Background(), "/m1/foo", &resp)
	c.Assert(err, qt.Equals, nil)
	defer resp.Body.Close()
	data, err := ioutil.ReadAll(resp.Body)
	c.Assert(err, qt.Equals, nil)
	c.Assert(string(data), qt.Equals, `{"P":"foo"}`)
}

func TestDoWithHTTPReponseAndError(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	srv := newServer()
	c.Defer(srv.Close)

	var doer closeCountingDoer // Also check the body is closed.
	client := &httprequest.Client{
		BaseURL: srv.URL,
		Doer:    &doer,
	}
	var resp *http.Response
	err := client.Get(context.Background(), "/m3", &resp)
	c.Assert(resp, qt.IsNil)
	c.Assert(err, qt.ErrorMatches, `Get http:.*/m3: m3 error`)
	c.Assert(doer.openedBodies, qt.Equals, 1)
	c.Assert(doer.closedBodies, qt.Equals, 1)
}

func TestCallWithHTTPResponse(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	srv := newServer()
	c.Defer(srv.Close)

	client := &httprequest.Client{
		BaseURL: srv.URL,
	}
	var resp *http.Response
	err := client.Call(context.Background(), &chM1Req{
		P: "foo",
	}, &resp)
	defer resp.Body.Close()
	data, err := ioutil.ReadAll(resp.Body)
	c.Assert(err, qt.Equals, nil)
	c.Assert(string(data), qt.Equals, `{"P":"foo"}`)
}

func TestCallClosesResponseBodyOnSuccess(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	srv := newServer()
	c.Defer(srv.Close)

	var doer closeCountingDoer
	client := &httprequest.Client{
		BaseURL: srv.URL,
		Doer:    &doer,
	}
	var resp chM1Resp
	err := client.Call(context.Background(), &chM1Req{
		P: "foo",
	}, &resp)
	c.Assert(err, qt.Equals, nil)
	c.Assert(resp, qt.DeepEquals, chM1Resp{"foo"})
	c.Assert(doer.openedBodies, qt.Equals, 1)
	c.Assert(doer.closedBodies, qt.Equals, 1)
}

func TestCallClosesResponseBodyOnError(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	srv := newServer()
	c.Defer(srv.Close)

	var doer closeCountingDoer
	client := &httprequest.Client{
		BaseURL: srv.URL,
		Doer:    &doer,
	}
	err := client.Call(context.Background(), &chM3Req{}, nil)
	c.Assert(err, qt.ErrorMatches, ".*m3 error")
	c.Assert(doer.openedBodies, qt.Equals, 1)
	c.Assert(doer.closedBodies, qt.Equals, 1)
}

func TestDoClosesResponseBodyOnSuccess(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	srv := newServer()
	c.Defer(srv.Close)

	var doer closeCountingDoer
	client := &httprequest.Client{
		BaseURL: srv.URL,
		Doer:    &doer,
	}
	req, err := http.NewRequest("GET", "/m1/foo", nil)
	c.Assert(err, qt.Equals, nil)
	var resp chM1Resp
	err = client.Do(context.Background(), req, &resp)
	c.Assert(err, qt.Equals, nil)
	c.Assert(resp, qt.DeepEquals, chM1Resp{"foo"})
	c.Assert(doer.openedBodies, qt.Equals, 1)
	c.Assert(doer.closedBodies, qt.Equals, 1)
}

func TestDoClosesResponseBodyOnError(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	srv := newServer()
	c.Defer(srv.Close)

	var doer closeCountingDoer
	client := &httprequest.Client{
		BaseURL: srv.URL,
		Doer:    &doer,
	}
	req, err := http.NewRequest("GET", "/m3", nil)
	c.Assert(err, qt.Equals, nil)
	err = client.Do(context.Background(), req, nil)
	c.Assert(err, qt.ErrorMatches, ".*m3 error")
	c.Assert(doer.openedBodies, qt.Equals, 1)
	c.Assert(doer.closedBodies, qt.Equals, 1)
}

func TestGet(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	srv := newServer()
	c.Defer(srv.Close)

	client := httprequest.Client{
		BaseURL: srv.URL,
	}
	var resp chM1Resp
	err := client.Get(context.Background(), "/m1/foo", &resp)
	c.Assert(err, qt.Equals, nil)
	c.Assert(resp, qt.DeepEquals, chM1Resp{"foo"})
}

func TestGetNoBaseURL(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	srv := newServer()
	c.Defer(srv.Close)

	client := httprequest.Client{}
	var resp chM1Resp
	err := client.Get(context.Background(), srv.URL+"/m1/foo", &resp)
	c.Assert(err, qt.Equals, nil)
	c.Assert(resp, qt.DeepEquals, chM1Resp{"foo"})
}

func TestUnmarshalJSONResponseWithBodyReadError(t *testing.T) {
	c := qt.New(t)

	resp := &http.Response{
		Header: http.Header{
			"Content-Type": {"application/json"},
		},
		StatusCode: http.StatusOK,
		Body: ioutil.NopCloser(io.MultiReader(
			strings.NewReader(`{"one": "two"}`),
			errorReader("some bad read"),
		)),
	}
	var val map[string]string
	err := httprequest.UnmarshalJSONResponse(resp, &val)
	c.Assert(err, qt.ErrorMatches, `error reading response body: some bad read`)
	c.Assert(val, qt.IsNil)
	assertDecodeResponseError(c, err, http.StatusOK, `{"one": "two"}`)
}

var unmarshalJSONResponseWithVariedJSONContentTypesTests = []struct {
	contentType string
	expectError bool
}{{
	contentType: "application/json",
}, {
	contentType: "application/json+other",
}, {
	contentType: "application/vnd.schemaregistry.v1+json",
}, {
	contentType: "other/json",
	expectError: true,
}, {
	contentType: "other/jsonx",
	expectError: true,
}, {
	contentType: "other/xjson",
	expectError: true,
}, {
	contentType: "other/other+xjson",
	expectError: true,
}, {
	contentType: "other/other+jsonx",
	expectError: true,
}, {
	contentType: "application/other+json+foo",
}, {
	contentType: "application/other+json",
}, {
	contentType: "application/other+json+",
}, {
	contentType: "application/+json+",
}}

func TestUnmarshalJSONResponseWithVariedJSONContentTypes(t *testing.T) {
	c := qt.New(t)

	for _, test := range unmarshalJSONResponseWithVariedJSONContentTypesTests {
		c.Run(test.contentType, func(c *qt.C) {
			resp := &http.Response{
				Header: http.Header{
					"Content-Type": {test.contentType},
				},
				StatusCode: http.StatusTeapot,
				Body:       ioutil.NopCloser(strings.NewReader(`{}`)),
			}
			var val map[string]string
			err := httprequest.UnmarshalJSONResponse(resp, &val)
			if !test.expectError {
				c.Assert(err, qt.IsNil)
				return
			}
			c.Assert(err, qt.ErrorMatches, `unexpected content type `+regexp.QuoteMeta(test.contentType)+`; want application/json; content: "{}"`)
			c.Assert(val, qt.IsNil)
			assertDecodeResponseError(c, err, http.StatusTeapot, `{}`)
		})
	}
}

func TestUnmarshalJSONResponseWithErrorAndLargeBody(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	c.Patch(httprequest.MaxErrorBodySize, 11)

	resp := &http.Response{
		Header: http.Header{
			"Content-Type": {"foo/bar"},
		},
		StatusCode: http.StatusOK,
		Body:       ioutil.NopCloser(strings.NewReader(`123456789 123456789`)),
	}
	var val map[string]string
	err := httprequest.UnmarshalJSONResponse(resp, &val)
	c.Assert(err, qt.ErrorMatches, `unexpected content type foo/bar; want application/json; content: "123456789 1"`)
	c.Assert(val, qt.IsNil)
	assertDecodeResponseError(c, err, http.StatusOK, `123456789 1`)
}

func TestUnmarshalJSONResponseWithLargeBody(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	c.Patch(httprequest.MaxErrorBodySize, 11)

	resp := &http.Response{
		Header: http.Header{
			"Content-Type": {"application/json"},
		},
		StatusCode: http.StatusOK,
		Body:       ioutil.NopCloser(strings.NewReader(`"23456789 123456789"`)),
	}
	var val string
	err := httprequest.UnmarshalJSONResponse(resp, &val)
	c.Assert(err, qt.Equals, nil)
	c.Assert(val, qt.Equals, "23456789 123456789")
}

func TestUnmarshalJSONWithDecodeError(t *testing.T) {
	c := qt.New(t)

	resp := &http.Response{
		Header: http.Header{
			"Content-Type": {"application/json"},
		},
		StatusCode: http.StatusOK,
		Body:       ioutil.NopCloser(strings.NewReader(`{"one": "two"}`)),
	}
	var val chan string
	err := httprequest.UnmarshalJSONResponse(resp, &val)
	c.Assert(err, qt.ErrorMatches, `json: cannot unmarshal object into Go value of type chan string`)
	c.Assert(val, qt.IsNil)
	assertDecodeResponseError(c, err, http.StatusOK, `{"one": "two"}`)
}

func TestUnmarshalJSONWithDecodeErrorAndLargeBody(t *testing.T) {
	c := qt.New(t)
	defer c.Done()

	c.Patch(httprequest.MaxErrorBodySize, 11)

	resp := &http.Response{
		Header: http.Header{
			"Content-Type": {"application/json"},
		},
		StatusCode: http.StatusOK,
		Body:       ioutil.NopCloser(strings.NewReader(`"23456789 123456789"`)),
	}
	var val chan string
	err := httprequest.UnmarshalJSONResponse(resp, &val)
	c.Assert(err, qt.ErrorMatches, `json: cannot unmarshal string into Go value of type chan string`)
	c.Assert(val, qt.IsNil)
	assertDecodeResponseError(c, err, http.StatusOK, `"23456789 1`)
}

func assertDecodeResponseError(c *qt.C, err error, status int, body string) {
	err1, ok := errgo.Cause(err).(*httprequest.DecodeResponseError)
	c.Assert(ok, qt.Equals, true, qt.Commentf("error not of type *httprequest.DecodeResponseError (%T)", errgo.Cause(err)))
	data, err := ioutil.ReadAll(err1.Response.Body)
	c.Assert(err, qt.Equals, nil)
	c.Assert(err1.Response.StatusCode, qt.Equals, status)
	c.Assert(string(data), qt.Equals, body)
}

func newServer() *httptest.Server {
	f := func(p httprequest.Params) (clientHandlers, context.Context, error) {
		return clientHandlers{}, p.Context, nil
	}
	handlers := testServer.Handlers(f)
	router := httprouter.New()
	for _, h := range handlers {
		router.Handle(h.Method, h.Path, h.Handle)
	}

	return httptest.NewServer(router)
}

var appendURLTests = []struct {
	u           string
	p           string
	expect      string
	expectError string
}{{
	u:      "http://foo",
	p:      "bar",
	expect: "http://foo/bar",
}, {
	u:      "http://foo",
	p:      "/bar",
	expect: "http://foo/bar",
}, {
	u:      "http://foo/",
	p:      "bar",
	expect: "http://foo/bar",
}, {
	u:      "http://foo/",
	p:      "/bar",
	expect: "http://foo/bar",
}, {
	u:      "",
	p:      "bar",
	expect: "/bar",
}, {
	u:      "http://xxx",
	p:      "",
	expect: "http://xxx",
}, {
	u:           "http://xxx.com",
	p:           "http://foo.com",
	expectError: "relative URL specifies a host",
}, {
	u:      "http://xxx.com/a/b",
	p:      "foo?a=45&b=c",
	expect: "http://xxx.com/a/b/foo?a=45&b=c",
}, {
	u:      "http://xxx.com",
	p:      "?a=45&b=c",
	expect: "http://xxx.com?a=45&b=c",
}, {
	u:      "http://xxx.com/a?z=w",
	p:      "foo?a=45&b=c",
	expect: "http://xxx.com/a/foo?z=w&a=45&b=c",
}, {
	u:      "http://xxx.com?z=w",
	p:      "/a/b/c",
	expect: "http://xxx.com/a/b/c?z=w",
}}

func TestAppendURL(t *testing.T) {
	c := qt.New(t)

	for _, test := range appendURLTests {
		test := test
		c.Run(fmt.Sprintf("%s_%s", test.u, test.p), func(c *qt.C) {
			u, err := httprequest.AppendURL(test.u, test.p)
			if test.expectError != "" {
				c.Assert(u, qt.IsNil)
				c.Assert(err, qt.ErrorMatches, test.expectError)
			} else {
				c.Assert(err, qt.Equals, nil)
				c.Assert(u.String(), qt.Equals, test.expect)
			}
		})
	}
}

type clientHandlers struct{}

type chM1Req struct {
	httprequest.Route `httprequest:"GET /m1/:P"`
	P                 string `httprequest:",path"`
}

type chM1Resp struct {
	P string
}

func (clientHandlers) M1(p *chM1Req) (*chM1Resp, error) {
	return &chM1Resp{p.P}, nil
}

type chM2Req struct {
	httprequest.Route `httprequest:"POST /m2/:P"`
	P                 string `httprequest:",path"`
	Body              struct {
		I int
	} `httprequest:",body"`
}

type chInvalidM2Req struct {
	httprequest.Route `httprequest:"POST /m2/:P"`
	P                 string `httprequest:",path"`
	Body              struct {
		I bool
	} `httprequest:",body"`
}

type chM2RedirectM2Req struct {
	httprequest.Route `httprequest:"POST /m2/foo//"`
	Body              struct {
		I int
	} `httprequest:",body"`
}

type chM2Resp struct {
	P   string
	Arg int
}

func (clientHandlers) M2(p *chM2Req) (*chM2Resp, error) {
	return &chM2Resp{p.P, p.Body.I}, nil
}

type chM3Req struct {
	httprequest.Route `httprequest:"GET /m3"`
}

func (clientHandlers) M3(p *chM3Req) error {
	return errgo.New("m3 error")
}

type chM4Req struct {
	httprequest.Route `httprequest:"GET /m4"`
}

func (clientHandlers) M4(p httprequest.Params, _ *chM4Req) {
	p.Response.Write([]byte("bad response"))
}

type chM5Req struct {
	httprequest.Route `httprequest:"GET /m5"`
}

func (clientHandlers) M5(p httprequest.Params, _ *chM5Req) {
	p.Response.WriteHeader(http.StatusTeapot)
	p.Response.Write([]byte("bad error value"))
}

type chContentLengthReq struct {
	httprequest.Route `httprequest:"PUT /content-length"`
}

func (clientHandlers) ContentLength(rp httprequest.Params, p *chContentLengthReq) (int64, error) {
	return rp.Request.ContentLength, nil
}

type doerFunc func(req *http.Request) (*http.Response, error)

func (f doerFunc) Do(req *http.Request) (*http.Response, error) {
	return f(req)
}

type doerWithContextFunc func(ctx context.Context, req *http.Request) (*http.Response, error)

func (f doerWithContextFunc) Do(req *http.Request) (*http.Response, error) {
	return f(nil, req)
}

func (f doerWithContextFunc) DoWithContext(ctx context.Context, req *http.Request) (*http.Response, error) {
	if ctx == nil {
		panic("unexpected nil context")
	}
	return f(ctx, req)
}

type closeCountingDoer struct {
	// openBodies records the number of response bodies
	// that have been returned.
	openedBodies int

	// closedBodies records the number of response bodies
	// that have been closed.
	closedBodies int
}

func (doer *closeCountingDoer) Do(req *http.Request) (*http.Response, error) {
	resp, err := http.DefaultClient.Do(req)
	if err != nil {
		return nil, err
	}
	resp.Body = &closeCountingReader{
		doer:       doer,
		ReadCloser: resp.Body,
	}
	doer.openedBodies++
	return resp, nil
}

type closeCountingReader struct {
	doer *closeCountingDoer
	io.ReadCloser
}

func (r *closeCountingReader) Close() error {
	r.doer.closedBodies++
	return r.ReadCloser.Close()
}

// largeReader implements a reader that produces up to total bytes
// in 1 byte reads.
type largeReader struct {
	byte  byte
	total int
	n     int
}

func (r *largeReader) Read(buf []byte) (int, error) {
	if r.n >= r.total {
		return 0, io.EOF
	}
	r.n++
	return copy(buf, []byte{r.byte}), nil
}

func (r *largeReader) Seek(offset int64, whence int) (int64, error) {
	if offset != 0 || whence != 0 {
		panic("unexpected seek")
	}
	r.n = 0
	return 0, nil
}

func (r *largeReader) Close() error {
	// By setting n to zero, we ensure that if there's
	// a concurrent read, it will also read from n
	// and so the race detector should pick up the
	// problem.
	r.n = 0
	return nil
}

func isRemoteError(err error) bool {
	_, ok := err.(*httprequest.RemoteError)
	return ok
}