diff --git a/api.go b/api.go index 7f0a11a..62290ea 100644 --- a/api.go +++ b/api.go @@ -15,19 +15,28 @@ const ( type API struct { api string // api to call key string // key in Authorization: Bearer + cli *http.Client } func NewAPI(api, key string) API { return API{api: api, key: key} } +func (api *API) SetHTTPClient(c *http.Client) { + api.cli = c +} + func (api *API) Request(model Model) (string, error) { req, err := http.NewRequest("POST", model.API(api.api, api.key), model.Body()) if err != nil { return "", err } model.Header(api.key, req.Header) - resp, err := http.DefaultClient.Do(req) + cli := http.DefaultClient + if api.cli != nil { + cli = api.cli + } + resp, err := cli.Do(req) if err != nil { return "", err }