Refactor base method to make it become reusable whenever it's needed

Author: denismakogon

client.go

@@ -218,13 +218,12 @@ func NewVersionedClient(endpoint string, apiVersionString string) (*Client, erro
 		eventMonitor:        new(eventMonitoringState),
 		requestedAPIVersion: requestedAPIVersion,
 	}
-	c.initializeNativeClient()
+	initializeNativeClient(c, defaultTransport)
 	return c, nil
 }
 
-func (c *Client) WithTransport(tr *http.Transport) {
-	initializeNativeClientTransport(c, tr)
-	c.HTTPClient.Transport = tr
+func (c *Client) WithTransport(trFunc func () *http.Transport) {
+	initializeNativeClient(c, trFunc)
 }
 
 // NewVersionnedTLSClient has been DEPRECATED, please use NewVersionedTLSClient.
@@ -344,7 +343,7 @@ func NewVersionedTLSClientFromBytes(endpoint string, certPEMBlock, keyPEMBlock,
 		eventMonitor:        new(eventMonitoringState),
 		requestedAPIVersion: requestedAPIVersion,
 	}
-	c.initializeNativeClient()
+	initializeNativeClient(c, defaultTransport)
 	return c, nil
 }
 

client_unix.go

@@ -14,16 +14,14 @@ import (
 
 // initializeNativeClient initializes the native Unix domain socket client on
 // Unix-style operating systems
-func (c *Client) initializeNativeClient() {
+func initializeNativeClient(c *Client, trFunc func () *http.Transport) {
 	if c.endpointURL.Scheme != unixProtocol {
 		return
 	}
-	tr := defaultTransport()
-	initializeNativeClientTransport(c, tr)
-}
-
-func initializeNativeClientTransport(c *Client, tr *http.Transport) {
 	sockPath := c.endpointURL.Path
+
+	tr := trFunc()
+
 	tr.Dial = func(network, addr string) (net.Conn, error) {
 		return c.Dialer.Dial(unixProtocol, sockPath)
 	}