log level fix & custom transport fixes

bundleman · Oct 5, 2023 · a24ac88 · a24ac88
1 parent 7e947c4
commit a24ac88
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/e2e/cli.go b/e2e/cli.go
@@ -13,6 +13,7 @@ import (
 	"path/filepath"
 	rt "runtime"
 	"runtime/pprof"
+	"strconv"
 	"strings"
 	"time"
 
@@ -222,7 +223,7 @@ var (
 
 	logLevel = flag.String(
 		"log-level",
-		logging.ErrorLevel.String(),
+		strconv.Itoa(logging.ErrorLevel),
 		"log level",
 	)
 )

diff --git a/pkg/drivers/http/driver.go b/pkg/drivers/http/driver.go
@@ -81,7 +81,10 @@ func addProxy(httpClient *pester.Client, proxyStr string) error {
 	proxy := http.ProxyURL(proxyURL)
 
 	if httpClient.Transport != nil {
-		httpClient.Transport.(*http.Transport).Proxy = proxy
+		transport, isHTTPTransport := httpClient.Transport.(*http.Transport)
+		if isHTTPTransport {
+			transport.Proxy = proxy
+		}
 
 		return nil
 	}

diff --git a/pkg/drivers/http/options.go b/pkg/drivers/http/options.go
@@ -31,7 +31,7 @@ type (
 		Concurrency       int
 		BodyLimit         int64
 		HTTPCodesFilter   []compiledStatusCodeFilter
-		HTTPTransport     *stdhttp.Transport
+		HTTPTransport     stdhttp.RoundTripper
 		Timeout           time.Duration
 		MaxRedirectsLimit uint8
 	}
@@ -144,7 +144,7 @@ func WithAllowedHTTPCodes(httpCodes []int) Option {
 	}
 }
 
-func WithCustomTransport(transport *stdhttp.Transport) Option {
+func WithCustomTransport(transport stdhttp.RoundTripper) Option {
 	return func(opts *Options) {
 		opts.HTTPTransport = transport
 	}