diff --git a/pkg/client/client.go b/pkg/client/client.go index 45ddece..a6d3798 100644 --- a/pkg/client/client.go +++ b/pkg/client/client.go @@ -137,8 +137,13 @@ func NewClient(opts *Config) (*Client, error) { // Jar: jar, // Timeout: 0, // } + transport := http.DefaultTransport + if opts.Transport != nil { + transport = opts.Transport + } + client.httpClient = &http.Client{ - Transport: http.DefaultTransport, + Transport: transport, } serverURL := DefaultServerURL diff --git a/pkg/client/config.go b/pkg/client/config.go index fa3ed5e..dfe7f8f 100644 --- a/pkg/client/config.go +++ b/pkg/client/config.go @@ -17,6 +17,7 @@ limitations under the License. package client import ( + "net/http" "os" ) @@ -38,6 +39,7 @@ type Config struct { Password string AccessToken string UserAgent string + Transport http.RoundTripper } func (c *Config) ReadFromEnvironment() {