refactor: move httptransport w/ logging to netxlite (#411)

Part of https://github.com/ooni/probe/issues/1505

internal/engine/netx/dialer/dialer.go

@@ -25,6 +25,9 @@ type Resolver interface {
 type Logger interface {
 	// Debugf formats and emits a debug message.
 	Debugf(format string, v ...interface{})
+
+	// Debug emits a debug message.
+	Debug(msg string)
 }
 
 // Config contains the settings for New.

internal/engine/netx/httptransport/logging.go

@@ -1,50 +0,0 @@
-package httptransport
-
-import "net/http"
-
-// Logger is the logger assumed by this package
-type Logger interface {
-	Debugf(format string, v ...interface{})
-	Debug(message string)
-}
-
-// LoggingTransport is a logging transport
-type LoggingTransport struct {
-	RoundTripper
-	Logger Logger
-}
-
-// RoundTrip implements RoundTripper.RoundTrip
-func (txp LoggingTransport) RoundTrip(req *http.Request) (*http.Response, error) {
-	host := req.Host
-	if host == "" {
-		host = req.URL.Host
-	}
-	req.Header.Set("Host", host) // anticipate what Go would do
-	return txp.logTrip(req)
-}
-
-func (txp LoggingTransport) logTrip(req *http.Request) (*http.Response, error) {
-	txp.Logger.Debugf("> %s %s", req.Method, req.URL.String())
-	for key, values := range req.Header {
-		for _, value := range values {
-			txp.Logger.Debugf("> %s: %s", key, value)
-		}
-	}
-	txp.Logger.Debug(">")
-	resp, err := txp.RoundTripper.RoundTrip(req)
-	if err != nil {
-		txp.Logger.Debugf("< %s", err)
-		return nil, err
-	}
-	txp.Logger.Debugf("< %d", resp.StatusCode)
-	for key, values := range resp.Header {
-		for _, value := range values {
-			txp.Logger.Debugf("< %s: %s", key, value)
-		}
-	}
-	txp.Logger.Debug("<")
-	return resp, nil
-}
-
-var _ RoundTripper = LoggingTransport{}

internal/engine/netx/httptransport/logging_test.go

@@ -1,78 +0,0 @@
-package httptransport_test
-
-import (
-	"context"
-	"errors"
-	"io"
-	"net/http"
-	"net/url"
-	"strings"
-	"testing"
-
-	"github.com/apex/log"
-	"github.com/ooni/probe-cli/v3/internal/engine/netx/httptransport"
-	"github.com/ooni/probe-cli/v3/internal/iox"
-)
-
-func TestLoggingFailure(t *testing.T) {
-	txp := httptransport.LoggingTransport{
-		Logger: log.Log,
-		RoundTripper: httptransport.FakeTransport{
-			Err: io.EOF,
-		},
-	}
-	client := &http.Client{Transport: txp}
-	resp, err := client.Get("https://www.google.com")
-	if !errors.Is(err, io.EOF) {
-		t.Fatal("not the error we expected")
-	}
-	if resp != nil {
-		t.Fatal("expected nil response here")
-	}
-}
-
-func TestLoggingFailureWithNoHostHeader(t *testing.T) {
-	txp := httptransport.LoggingTransport{
-		Logger: log.Log,
-		RoundTripper: httptransport.FakeTransport{
-			Err: io.EOF,
-		},
-	}
-	req := &http.Request{
-		Header: http.Header{},
-		URL: &url.URL{
-			Scheme: "https",
-			Host:   "www.google.com",
-			Path:   "/",
-		},
-	}
-	resp, err := txp.RoundTrip(req)
-	if !errors.Is(err, io.EOF) {
-		t.Fatal("not the error we expected")
-	}
-	if resp != nil {
-		t.Fatal("expected nil response here")
-	}
-}
-
-func TestLoggingSuccess(t *testing.T) {
-	txp := httptransport.LoggingTransport{
-		Logger: log.Log,
-		RoundTripper: httptransport.FakeTransport{
-			Resp: &http.Response{
-				Body: io.NopCloser(strings.NewReader("")),
-				Header: http.Header{
-					"Server": []string{"antani/0.1.0"},
-				},
-				StatusCode: 200,
-			},
-		},
-	}
-	client := &http.Client{Transport: txp}
-	resp, err := client.Get("https://www.google.com")
-	if err != nil {
-		t.Fatal(err)
-	}
-	iox.ReadAllContext(context.Background(), resp.Body)
-	resp.Body.Close()
-}

internal/engine/netx/netx.go

@@ -231,7 +231,7 @@ func NewHTTPTransport(config Config) HTTPRoundTripper {
 			Counter: config.ByteCounter, RoundTripper: txp}
 	}
 	if config.Logger != nil {
-		txp = httptransport.LoggingTransport{Logger: config.Logger, RoundTripper: txp}
+		txp = &netxlite.HTTPTransportLogger{Logger: config.Logger, HTTPTransport: txp}
 	}
 	if config.HTTPSaver != nil {
 		txp = httptransport.SaverMetadataHTTPTransport{

internal/engine/netx/netx_test.go

@@ -491,14 +491,14 @@ func TestNewWithLogger(t *testing.T) {
 	if !ok {
 		t.Fatal("not the transport we expected")
 	}
-	ltxp, ok := uatxp.RoundTripper.(httptransport.LoggingTransport)
+	ltxp, ok := uatxp.RoundTripper.(*netxlite.HTTPTransportLogger)
 	if !ok {
 		t.Fatal("not the transport we expected")
 	}
 	if ltxp.Logger != log.Log {
 		t.Fatal("not the logger we expected")
 	}
-	if _, ok := ltxp.RoundTripper.(*http.Transport); !ok {
+	if _, ok := ltxp.HTTPTransport.(*http.Transport); !ok {
 		t.Fatal("not the transport we expected")
 	}
 }