test: allow specifying HTTP protocol (#822)

Signed-off-by: spacewander <spacewanderlzx@gmail.com>

test/e2e/conformance/utils/http/http.go

@@ -88,6 +88,7 @@ const (
 // values can be provided, as a comma-separated value.
 type Request struct {
 	Host             string
+	Protocol         string
 	Method           string
 	Path             string
 	Headers          map[string]string
@@ -162,12 +163,10 @@ func MakeRequestAndExpectEventuallyConsistentResponse(t *testing.T, r roundtripp
 
 	var (
 		scheme    = "http"
-		protocol  = "HTTP"
 		tlsConfig *roundtripper.TLSConfig
 	)
 	if expected.Request.ActualRequest.TLSConfig != nil {
 		scheme = "https"
-		protocol = "HTTPS"
 		clientKeyPairs := make([]roundtripper.ClientKeyPair, 0, len(expected.Request.ActualRequest.TLSConfig.Certificates.ClientKeyPairs))
 		for _, keyPair := range expected.Request.ActualRequest.TLSConfig.Certificates.ClientKeyPairs {
 			clientKeyPairs = append(clientKeyPairs, roundtripper.ClientKeyPair{
@@ -213,6 +212,11 @@ func MakeRequestAndExpectEventuallyConsistentResponse(t *testing.T, r roundtripp
 
 	path, query, _ := strings.Cut(expected.Request.ActualRequest.Path, "?")
 
+	protocol := "HTTP/1.1"
+	if expected.Request.ActualRequest.Protocol != "" {
+		protocol = expected.Request.ActualRequest.Protocol
+	}
+
 	req := roundtripper.Request{
 		Method:           expected.Request.ActualRequest.Method,
 		Host:             expected.Request.ActualRequest.Host,

test/e2e/conformance/utils/roundtripper/roundtripper.go

@@ -26,6 +26,8 @@ import (
 	"net/url"
 	"regexp"
 
+	"golang.org/x/net/http2"
+
 	"github.com/alibaba/higress/test/e2e/conformance/utils/config"
 )
 
@@ -109,6 +111,53 @@ type DefaultRoundTripper struct {
 	TimeoutConfig config.TimeoutConfig
 }
 
+func (d *DefaultRoundTripper) initTransport(client *http.Client, protocol string, tlsConfig *TLSConfig) error {
+	var tlsClientConfig *tls.Config
+	if tlsConfig != nil {
+		pool := x509.NewCertPool()
+		for _, caCert := range tlsConfig.Certificates.CACert {
+			pool.AppendCertsFromPEM(caCert)
+		}
+		var clientCerts []tls.Certificate
+		for _, keyPair := range tlsConfig.Certificates.ClientKeyPairs {
+			newClientCert, err := tls.X509KeyPair(keyPair.ClientCert, keyPair.ClientKey)
+			if err != nil {
+				return fmt.Errorf("failed to load client key pair: %w", err)
+			}
+			clientCerts = append(clientCerts, newClientCert)
+		}
+
+		tlsClientConfig = &tls.Config{
+			MinVersion:         tlsConfig.MinVersion,
+			MaxVersion:         tlsConfig.MaxVersion,
+			ServerName:         tlsConfig.SNI,
+			CipherSuites:       tlsConfig.CipherSuites,
+			RootCAs:            pool,
+			Certificates:       clientCerts,
+			InsecureSkipVerify: true,
+		}
+	}
+
+	switch protocol {
+	case "HTTP/2.0":
+		tr := &http2.Transport{}
+		if tlsClientConfig != nil {
+			tr.TLSClientConfig = tlsClientConfig
+		}
+		client.Transport = tr
+	default: // HTTP1
+		if tlsClientConfig != nil {
+			client.Transport = &http.Transport{
+				TLSHandshakeTimeout: d.TimeoutConfig.TLSHandshakeTimeout,
+				DisableKeepAlives:   true,
+				TLSClientConfig:     tlsClientConfig,
+			}
+		}
+	}
+
+	return nil
+}
+
 // CaptureRoundTrip makes a request with the provided parameters and returns the
 // captured request and response from echoserver. An error will be returned if
 // there is an error running the function but not if an HTTP error status code
@@ -152,6 +201,8 @@ func (d *DefaultRoundTripper) CaptureRoundTrip(request Request) (*CapturedReques
 		}
 	}
 
+	d.initTransport(client, request.Protocol, request.TLSConfig)
+
 	method := "GET"
 	if request.Method != "" {
 		method = request.Method

test/e2e/conformance/utils/roundtripper/roundtripper_test.go

@@ -0,0 +1,77 @@
+// Copyright (c) 2022 Alibaba Group Holding Ltd.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package roundtripper
+
+import (
+	"crypto/tls"
+	"crypto/x509"
+	"net/http"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"golang.org/x/net/http2"
+)
+
+func TestTransport(t *testing.T) {
+	req := Request{
+		Protocol: "HTTP/2.0",
+	}
+	tests := []struct {
+		name          string
+		req           Request
+		prevTransport http.RoundTripper
+		tlsConfig     *TLSConfig
+		transport     http.RoundTripper
+	}{
+		{
+			name: "default",
+			req:  Request{},
+		},
+		{
+			name:      "http2",
+			req:       req,
+			transport: &http2.Transport{},
+		},
+		{
+			name: "http1",
+			req: Request{
+				Protocol: "HTTP/1.1",
+			},
+		},
+		{
+			name: "https",
+			req:  req,
+			tlsConfig: &TLSConfig{
+				SNI: "www.example.com",
+			},
+			transport: &http2.Transport{
+				TLSClientConfig: &tls.Config{
+					RootCAs:            x509.NewCertPool(),
+					ServerName:         "www.example.com",
+					InsecureSkipVerify: true,
+				},
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			d := DefaultRoundTripper{}
+			c := http.Client{}
+			d.initTransport(&c, tt.req.Protocol, tt.tlsConfig)
+			assert.Equal(t, tt.transport, c.Transport)
+		})
+	}
+}