Add client.WithTimeout() option
Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
This commit is contained in:
parent
6d18c6a062
commit
44982c775e
|
@ -6,6 +6,7 @@ import (
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/docker/go-connections/sockets"
|
"github.com/docker/go-connections/sockets"
|
||||||
"github.com/docker/go-connections/tlsconfig"
|
"github.com/docker/go-connections/tlsconfig"
|
||||||
|
@ -102,6 +103,14 @@ func WithHTTPClient(client *http.Client) Opt {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// WithTimeout configures the time limit for requests made by the HTTP client
|
||||||
|
func WithTimeout(timeout time.Duration) Opt {
|
||||||
|
return func(c *Client) error {
|
||||||
|
c.client.Timeout = timeout
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// WithHTTPHeaders overrides the client default http headers
|
// WithHTTPHeaders overrides the client default http headers
|
||||||
func WithHTTPHeaders(headers map[string]string) Opt {
|
func WithHTTPHeaders(headers map[string]string) Opt {
|
||||||
return func(c *Client) error {
|
return func(c *Client) error {
|
||||||
|
|
|
@ -0,0 +1,16 @@
|
||||||
|
package client
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"gotest.tools/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestOptionWithTimeout(t *testing.T) {
|
||||||
|
timeout := 10 * time.Second
|
||||||
|
c, err := NewClientWithOpts(WithTimeout(timeout))
|
||||||
|
assert.NilError(t, err)
|
||||||
|
assert.Check(t, c.client != nil)
|
||||||
|
assert.Equal(t, c.client.Timeout, timeout)
|
||||||
|
}
|
Loading…
Reference in New Issue