Fixing tests

denismakogon · denismakogon · commit a8f5e915cb6e · 2018-01-21T21:38:15.000+02:00
diff --git a/client.go b/client.go
@@ -223,8 +223,8 @@ func NewVersionedClient(endpoint string, apiVersionString string) (*Client, erro
 }
 
 func (c *Client) WithTransport(tr *http.Transport) {
-	c.initializeNativeClientTransport(tr)
-	c.HTTPClient = &http.Client{Transport: tr}
+	initializeNativeClientTransport(c, tr)
+	c.HTTPClient.Transport = tr
 }
 
 // NewVersionnedTLSClient has been DEPRECATED, please use NewVersionedTLSClient.
diff --git a/client_unix.go b/client_unix.go
@@ -19,10 +19,10 @@ func (c *Client) initializeNativeClient() {
 		return
 	}
 	tr := defaultTransport()
-	c.initializeNativeClientTransport(tr)
+	initializeNativeClientTransport(c, tr)
 }
 
-func (c *Client) initializeNativeClientTransport(tr *http.Transport) {
+func initializeNativeClientTransport(c *Client, tr *http.Transport) {
 	sockPath := c.endpointURL.Path
 	tr.Dial = func(network, addr string) (net.Conn, error) {
 		return c.Dialer.Dial(unixProtocol, sockPath)

Original file line number	Diff line number	Diff line change
`@@ -223,8 +223,8 @@ func NewVersionedClient(endpoint string, apiVersionString string) (*Client, erro`
`223`	`223`	`}`
`224`	`224`
`225`	`225`	`func (c Client) WithTransport(tr http.Transport) {`
`226`		`- c.initializeNativeClientTransport(tr)`
`227`		`- c.HTTPClient = &http.Client{Transport: tr}`
	`226`	`+ initializeNativeClientTransport(c, tr)`
	`227`	`+ c.HTTPClient.Transport = tr`
`228`	`228`	`}`
`229`	`229`
`230`	`230`	`// NewVersionnedTLSClient has been DEPRECATED, please use NewVersionedTLSClient.`
Original file line number	Diff line number	Diff line change
`@@ -19,10 +19,10 @@ func (c *Client) initializeNativeClient() {`
`19`	`19`	`return`
`20`	`20`	`}`
`21`	`21`	`tr := defaultTransport()`
`22`		`- c.initializeNativeClientTransport(tr)`
	`22`	`+ initializeNativeClientTransport(c, tr)`
`23`	`23`	`}`
`24`	`24`
`25`		`-func (c Client) initializeNativeClientTransport(tr http.Transport) {`
	`25`	`+func initializeNativeClientTransport(c Client, tr http.Transport) {`
`26`	`26`	`sockPath := c.endpointURL.Path`
`27`	`27`	`tr.Dial = func(network, addr string) (net.Conn, error) {`
`28`	`28`	`return c.Dialer.Dial(unixProtocol, sockPath)`