Add debug roundtripper to discovery and identity clients for easier debugging

Signed-off-by: Richard Wall <[email protected]>

Author: wallrj

pkg/internal/cyberark/dataupload/dataupload.go

@@ -46,7 +46,7 @@ func NewCyberArkClient(trustedCAs *x509.CertPool, baseURL string, authenticateRe
 	if trustedCAs != nil {
 		tr.TLSClientConfig.RootCAs = trustedCAs
 	}
-	cyberClient.Transport = transport.DebugWrappers(tr)
+	cyberClient.Transport = transport.NewDebuggingRoundTripper(http.DefaultTransport, transport.DebugByContext)
 
 	return &CyberArkClient{
 		baseURL:             baseURL,

pkg/internal/cyberark/dataupload/dataupload_test.go

@@ -137,7 +137,7 @@ func TestPostDataReadingsWithOptionsWithRealAPI(t *testing.T) {
 		return
 	}
 
-	logger := ktesting.NewLogger(t, ktesting.NewConfig())
+	logger := ktesting.NewLogger(t, ktesting.NewConfig(ktesting.Verbosity(6)))
 	ctx := klog.NewContext(t.Context(), logger)
 
 	const (
@@ -165,7 +165,7 @@ func TestPostDataReadingsWithOptionsWithRealAPI(t *testing.T) {
 	cyberArkClient, err := dataupload.NewCyberArkClient(nil, serviceURL, identityClient.AuthenticateRequest)
 	require.NoError(t, err)
 
-	err = cyberArkClient.PostDataReadingsWithOptions(t.Context(), api.DataReadingsPost{}, dataupload.Options{
+	err = cyberArkClient.PostDataReadingsWithOptions(ctx, api.DataReadingsPost{}, dataupload.Options{
 		ClusterName: "bb068932-c80d-460d-88df-34bc7f3f3297",
 	})
 	require.NoError(t, err)

pkg/internal/cyberark/identity/identity.go

@@ -12,6 +12,7 @@ import (
 	"time"
 
 	"github.com/cenkalti/backoff/v5"
+	"k8s.io/client-go/transport"
 	"k8s.io/klog/v2"
 
 	"github.com/jetstack/preflight/pkg/internal/cyberark/servicediscovery"
@@ -212,7 +213,8 @@ func NewWithDiscoveryClient(ctx context.Context, discoveryClient *servicediscove
 
 	return &Client{
 		client: &http.Client{
-			Timeout: 10 * time.Second,
+			Timeout:   10 * time.Second,
+			Transport: transport.NewDebuggingRoundTripper(http.DefaultTransport, transport.DebugByContext),
 		},
 
 		endpoint:  endpoint,

pkg/internal/cyberark/servicediscovery/discovery.go

@@ -9,6 +9,8 @@ import (
 	"net/url"
 	"time"
 
+	"k8s.io/client-go/transport"
+
 	"github.com/jetstack/preflight/pkg/version"
 )
 
@@ -62,7 +64,8 @@ func WithCustomEndpoint(endpoint string) ClientOpt {
 func New(clientOpts ...ClientOpt) *Client {
 	client := &Client{
 		client: &http.Client{
-			Timeout: 10 * time.Second,
+			Timeout:   10 * time.Second,
+			Transport: transport.NewDebuggingRoundTripper(http.DefaultTransport, transport.DebugByContext),
 		},
 		endpoint: prodDiscoveryEndpoint,
 	}