refactor: move httptransport w/ logging to netxlite (#411)
Part of https://github.com/ooni/probe/issues/1505
This commit is contained in:
@@ -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{}
|
||||
@@ -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()
|
||||
}
|
||||
Reference in New Issue
Block a user