diff --git a/rest/httpc/service.go b/rest/httpc/service.go new file mode 100644 index 00000000..eb917391 --- /dev/null +++ b/rest/httpc/service.go @@ -0,0 +1,39 @@ +package httpc + +import ( + "io" + "net/http" +) + +type ( + Service interface { + Do(r *http.Request, opts ...Option) (*http.Response, error) + Get(url string, opts ...Option) (*http.Response, error) + Post(url, contentType string, body io.Reader, opts ...Option) (*http.Response, error) + } + + namedService struct { + name string + opts []Option + } +) + +func NewService(name string, opts ...Option) Service { + return namedService{ + name: name, + opts: opts, + } +} + +func (s namedService) Do(r *http.Request, opts ...Option) (*http.Response, error) { + return Do(s.name, r, append(s.opts, opts...)...) +} + +func (s namedService) Get(url string, opts ...Option) (*http.Response, error) { + return Get(s.name, url, append(s.opts, opts...)...) +} + +func (s namedService) Post(url, contentType string, body io.Reader, opts ...Option) ( + *http.Response, error) { + return Post(s.name, url, contentType, body, append(s.opts, opts...)...) +} diff --git a/rest/httpc/service_test.go b/rest/httpc/service_test.go new file mode 100644 index 00000000..82e78bca --- /dev/null +++ b/rest/httpc/service_test.go @@ -0,0 +1,40 @@ +package httpc + +import ( + "net/http" + "net/http/httptest" + "testing" + + "github.com/stretchr/testify/assert" +) + +func TestNamedService_Do(t *testing.T) { + svr := httptest.NewServer(http.RedirectHandler("/foo", http.StatusMovedPermanently)) + req, err := http.NewRequest(http.MethodGet, svr.URL, nil) + assert.Nil(t, err) + service := NewService("foo") + _, err = service.Do(req) + // too many redirects + assert.NotNil(t, err) +} + +func TestNamedService_Get(t *testing.T) { + svr := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + })) + service := NewService("foo") + resp, err := service.Get(svr.URL, func(cli *http.Client) { + cli.Transport = http.DefaultTransport + }) + assert.Nil(t, err) + assert.Equal(t, http.StatusOK, resp.StatusCode) +} + +func TestNamedService_Post(t *testing.T) { + svr := httptest.NewServer(http.NotFoundHandler()) + service := NewService("foo") + _, err := service.Post("tcp://bad request", "application/json", nil) + assert.NotNil(t, err) + resp, err := service.Post(svr.URL, "application/json", nil) + assert.Nil(t, err) + assert.Equal(t, http.StatusNotFound, resp.StatusCode) +}