test: Add custom transport test case

client_test.go

@@ -3193,3 +3193,61 @@ func Test_AddMissingPort(t *testing.T) {
 		})
 	}
 }
+
+type TransportEx struct {
+	base    RoundTripper
+	reqLog  *string
+	respLog *string
+}
+
+func (t TransportEx) RoundTrip(hc *HostClient, req *Request, resp *Response) (bool, error) {
+	req.Header.Set("trace-id", "123")
+	*t.reqLog = req.String()
+	retry, err := t.base.RoundTrip(hc, req, resp)
+	*t.respLog = resp.String()
+	return retry, err
+}
+
+func TestClientTransportEx(t *testing.T) {
+	var reqLog string
+	var respLog string
+
+	sHTTP := startEchoServer(t, "tcp", "127.0.0.1:")
+	defer sHTTP.Stop()
+
+	sHTTPS := startEchoServerTLS(t, "tcp", "127.0.0.1:")
+	defer sHTTPS.Stop()
+
+	c := &Client{
+		TLSConfig: &tls.Config{
+			InsecureSkipVerify: true,
+		},
+		ConfigureClient: func(hc *HostClient) error {
+			hc.Transport = TransportEx{base: DefaultTransport, reqLog: &reqLog, respLog: &respLog}
+			return nil
+		},
+	}
+
+	for i := 0; i < 4; i++ {
+		addr := "http://" + sHTTP.Addr()
+		if i&1 != 0 {
+			addr = "https://" + sHTTPS.Addr()
+		}
+		// test get
+		testClientGet(t, c, addr, 20)
+		if !strings.Contains(reqLog, "Trace-Id: 123") {
+			t.Errorf("request log should contains: %v", "Trace-Id: 123")
+		}
+		if !strings.Contains(respLog, "HTTP/1.1 200 OK") {
+			t.Errorf("response log should contains: %v", "HTTP/1.1 200 OK")
+		}
+		// test post
+		testClientPost(t, c, addr, 10)
+		if !strings.Contains(reqLog, "Trace-Id: 123") {
+			t.Errorf("request log should contains: %v", "Trace-Id: 123")
+		}
+		if !strings.Contains(respLog, "HTTP/1.1 200 OK") {
+			t.Errorf("response log should contains: %v", "HTTP/1.1 200 OK")
+		}
+	}
+}