2572376fdb
This basically adapts already existing code inside websteps to instead be into the netxlite package, where it belongs. In the process, abstract the TLSDialer but keep a reference to the previous name to avoid refactoring existing code (just for now). While there, notice that the right name is CloseIdleConnections (i.e., plural not singular) and change the name. While there, since we abstracted TLSDialer to be an interface, create suitable factories for making a TLSDialer type from a Dialer and a TLSHandshaker. See https://github.com/ooni/probe/issues/1591
905 lines
24 KiB
Go
905 lines
24 KiB
Go
package netx_test
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"errors"
|
|
"net"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/apex/log"
|
|
"github.com/ooni/probe-cli/v3/internal/bytecounter"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/netx"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/httptransport"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/resolver"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/tlsdialer"
|
|
"github.com/ooni/probe-cli/v3/internal/engine/netx/trace"
|
|
"github.com/ooni/probe-cli/v3/internal/errorsx"
|
|
"github.com/ooni/probe-cli/v3/internal/netxlite"
|
|
"github.com/ooni/probe-cli/v3/internal/netxlite/mocks"
|
|
)
|
|
|
|
func TestNewResolverVanilla(t *testing.T) {
|
|
r := netx.NewResolver(netx.Config{})
|
|
ir, ok := r.(*resolver.IDNAResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
rla, ok := ir.Resolver.(*netxlite.ResolverLegacyAdapter)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ewr, ok := rla.ResolverLegacy.(*errorsx.ErrorWrapperResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ar, ok := ewr.Resolver.(resolver.AddressResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
_, ok = ar.Resolver.(*netxlite.ResolverSystem)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewResolverSpecificResolver(t *testing.T) {
|
|
r := netx.NewResolver(netx.Config{
|
|
BaseResolver: resolver.BogonResolver{
|
|
// not initialized because it doesn't matter in this context
|
|
},
|
|
})
|
|
ir, ok := r.(*resolver.IDNAResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
rla, ok := ir.Resolver.(*netxlite.ResolverLegacyAdapter)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ewr, ok := rla.ResolverLegacy.(*errorsx.ErrorWrapperResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ar, ok := ewr.Resolver.(resolver.AddressResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
_, ok = ar.Resolver.(resolver.BogonResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewResolverWithBogonFilter(t *testing.T) {
|
|
r := netx.NewResolver(netx.Config{
|
|
BogonIsError: true,
|
|
})
|
|
ir, ok := r.(*resolver.IDNAResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
rla, ok := ir.Resolver.(*netxlite.ResolverLegacyAdapter)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ewr, ok := rla.ResolverLegacy.(*errorsx.ErrorWrapperResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
br, ok := ewr.Resolver.(resolver.BogonResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ar, ok := br.Resolver.(resolver.AddressResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
_, ok = ar.Resolver.(*netxlite.ResolverSystem)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewResolverWithLogging(t *testing.T) {
|
|
r := netx.NewResolver(netx.Config{
|
|
Logger: log.Log,
|
|
})
|
|
ir, ok := r.(*resolver.IDNAResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
rla, ok := ir.Resolver.(*netxlite.ResolverLegacyAdapter)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
lr, ok := rla.ResolverLegacy.(*netxlite.ResolverLogger)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
if lr.Logger != log.Log {
|
|
t.Fatal("not the logger we expected")
|
|
}
|
|
rla, ok = ir.Resolver.(*netxlite.ResolverLegacyAdapter)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
lr, ok = rla.ResolverLegacy.(*netxlite.ResolverLogger)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
rla, ok = lr.Resolver.(*netxlite.ResolverLegacyAdapter)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ewr, ok := rla.ResolverLegacy.(*errorsx.ErrorWrapperResolver)
|
|
if !ok {
|
|
t.Fatalf("not the resolver we expected %T", rla.ResolverLegacy)
|
|
}
|
|
ar, ok := ewr.Resolver.(resolver.AddressResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
_, ok = ar.Resolver.(*netxlite.ResolverSystem)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewResolverWithSaver(t *testing.T) {
|
|
saver := new(trace.Saver)
|
|
r := netx.NewResolver(netx.Config{
|
|
ResolveSaver: saver,
|
|
})
|
|
ir, ok := r.(*resolver.IDNAResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
rla, ok := ir.Resolver.(*netxlite.ResolverLegacyAdapter)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
sr, ok := rla.ResolverLegacy.(resolver.SaverResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
if sr.Saver != saver {
|
|
t.Fatal("not the saver we expected")
|
|
}
|
|
ewr, ok := sr.Resolver.(*errorsx.ErrorWrapperResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ar, ok := ewr.Resolver.(resolver.AddressResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
_, ok = ar.Resolver.(*netxlite.ResolverSystem)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewResolverWithReadWriteCache(t *testing.T) {
|
|
r := netx.NewResolver(netx.Config{
|
|
CacheResolutions: true,
|
|
})
|
|
ir, ok := r.(*resolver.IDNAResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
rla, ok := ir.Resolver.(*netxlite.ResolverLegacyAdapter)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ewr, ok := rla.ResolverLegacy.(*errorsx.ErrorWrapperResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
cr, ok := ewr.Resolver.(*resolver.CacheResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
if cr.ReadOnly != false {
|
|
t.Fatal("expected readwrite cache here")
|
|
}
|
|
ar, ok := cr.Resolver.(resolver.AddressResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
_, ok = ar.Resolver.(*netxlite.ResolverSystem)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewResolverWithPrefilledReadonlyCache(t *testing.T) {
|
|
r := netx.NewResolver(netx.Config{
|
|
DNSCache: map[string][]string{
|
|
"dns.google.com": {"8.8.8.8"},
|
|
},
|
|
})
|
|
ir, ok := r.(*resolver.IDNAResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
rla, ok := ir.Resolver.(*netxlite.ResolverLegacyAdapter)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ewr, ok := rla.ResolverLegacy.(*errorsx.ErrorWrapperResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
cr, ok := ewr.Resolver.(*resolver.CacheResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
if cr.ReadOnly != true {
|
|
t.Fatal("expected readonly cache here")
|
|
}
|
|
if cr.Get("dns.google.com")[0] != "8.8.8.8" {
|
|
t.Fatal("cache not correctly prefilled")
|
|
}
|
|
ar, ok := cr.Resolver.(resolver.AddressResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
_, ok = ar.Resolver.(*netxlite.ResolverSystem)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewTLSDialerVanilla(t *testing.T) {
|
|
td := netx.NewTLSDialer(netx.Config{})
|
|
rtd, ok := td.(*netxlite.TLSDialerLegacy)
|
|
if !ok {
|
|
t.Fatal("not the TLSDialer we expected")
|
|
}
|
|
if len(rtd.Config.NextProtos) != 2 {
|
|
t.Fatal("invalid len(config.NextProtos)")
|
|
}
|
|
if rtd.Config.NextProtos[0] != "h2" || rtd.Config.NextProtos[1] != "http/1.1" {
|
|
t.Fatal("invalid Config.NextProtos")
|
|
}
|
|
if rtd.Config.RootCAs != netx.DefaultCertPool() {
|
|
t.Fatal("invalid Config.RootCAs")
|
|
}
|
|
if rtd.Dialer == nil {
|
|
t.Fatal("invalid Dialer")
|
|
}
|
|
if rtd.TLSHandshaker == nil {
|
|
t.Fatal("invalid TLSHandshaker")
|
|
}
|
|
ewth, ok := rtd.TLSHandshaker.(*errorsx.ErrorWrapperTLSHandshaker)
|
|
if !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
if _, ok := ewth.TLSHandshaker.(*netxlite.TLSHandshakerConfigurable); !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewTLSDialerWithConfig(t *testing.T) {
|
|
td := netx.NewTLSDialer(netx.Config{
|
|
TLSConfig: new(tls.Config),
|
|
})
|
|
rtd, ok := td.(*netxlite.TLSDialerLegacy)
|
|
if !ok {
|
|
t.Fatal("not the TLSDialer we expected")
|
|
}
|
|
if len(rtd.Config.NextProtos) != 0 {
|
|
t.Fatal("invalid len(config.NextProtos)")
|
|
}
|
|
if rtd.Config.RootCAs != netx.DefaultCertPool() {
|
|
t.Fatal("invalid Config.RootCAs")
|
|
}
|
|
if rtd.Dialer == nil {
|
|
t.Fatal("invalid Dialer")
|
|
}
|
|
if rtd.TLSHandshaker == nil {
|
|
t.Fatal("invalid TLSHandshaker")
|
|
}
|
|
ewth, ok := rtd.TLSHandshaker.(*errorsx.ErrorWrapperTLSHandshaker)
|
|
if !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
if _, ok := ewth.TLSHandshaker.(*netxlite.TLSHandshakerConfigurable); !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewTLSDialerWithLogging(t *testing.T) {
|
|
td := netx.NewTLSDialer(netx.Config{
|
|
Logger: log.Log,
|
|
})
|
|
rtd, ok := td.(*netxlite.TLSDialerLegacy)
|
|
if !ok {
|
|
t.Fatal("not the TLSDialer we expected")
|
|
}
|
|
if len(rtd.Config.NextProtos) != 2 {
|
|
t.Fatal("invalid len(config.NextProtos)")
|
|
}
|
|
if rtd.Config.NextProtos[0] != "h2" || rtd.Config.NextProtos[1] != "http/1.1" {
|
|
t.Fatal("invalid Config.NextProtos")
|
|
}
|
|
if rtd.Config.RootCAs != netx.DefaultCertPool() {
|
|
t.Fatal("invalid Config.RootCAs")
|
|
}
|
|
if rtd.Dialer == nil {
|
|
t.Fatal("invalid Dialer")
|
|
}
|
|
if rtd.TLSHandshaker == nil {
|
|
t.Fatal("invalid TLSHandshaker")
|
|
}
|
|
lth, ok := rtd.TLSHandshaker.(*netxlite.TLSHandshakerLogger)
|
|
if !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
if lth.Logger != log.Log {
|
|
t.Fatal("not the Logger we expected")
|
|
}
|
|
ewth, ok := lth.TLSHandshaker.(*errorsx.ErrorWrapperTLSHandshaker)
|
|
if !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
if _, ok := ewth.TLSHandshaker.(*netxlite.TLSHandshakerConfigurable); !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewTLSDialerWithSaver(t *testing.T) {
|
|
saver := new(trace.Saver)
|
|
td := netx.NewTLSDialer(netx.Config{
|
|
TLSSaver: saver,
|
|
})
|
|
rtd, ok := td.(*netxlite.TLSDialerLegacy)
|
|
if !ok {
|
|
t.Fatal("not the TLSDialer we expected")
|
|
}
|
|
if len(rtd.Config.NextProtos) != 2 {
|
|
t.Fatal("invalid len(config.NextProtos)")
|
|
}
|
|
if rtd.Config.NextProtos[0] != "h2" || rtd.Config.NextProtos[1] != "http/1.1" {
|
|
t.Fatal("invalid Config.NextProtos")
|
|
}
|
|
if rtd.Config.RootCAs != netx.DefaultCertPool() {
|
|
t.Fatal("invalid Config.RootCAs")
|
|
}
|
|
if rtd.Dialer == nil {
|
|
t.Fatal("invalid Dialer")
|
|
}
|
|
if rtd.TLSHandshaker == nil {
|
|
t.Fatal("invalid TLSHandshaker")
|
|
}
|
|
sth, ok := rtd.TLSHandshaker.(tlsdialer.SaverTLSHandshaker)
|
|
if !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
if sth.Saver != saver {
|
|
t.Fatal("not the Logger we expected")
|
|
}
|
|
ewth, ok := sth.TLSHandshaker.(*errorsx.ErrorWrapperTLSHandshaker)
|
|
if !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
if _, ok := ewth.TLSHandshaker.(*netxlite.TLSHandshakerConfigurable); !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewTLSDialerWithNoTLSVerifyAndConfig(t *testing.T) {
|
|
td := netx.NewTLSDialer(netx.Config{
|
|
TLSConfig: new(tls.Config),
|
|
NoTLSVerify: true,
|
|
})
|
|
rtd, ok := td.(*netxlite.TLSDialerLegacy)
|
|
if !ok {
|
|
t.Fatal("not the TLSDialer we expected")
|
|
}
|
|
if len(rtd.Config.NextProtos) != 0 {
|
|
t.Fatal("invalid len(config.NextProtos)")
|
|
}
|
|
if rtd.Config.InsecureSkipVerify != true {
|
|
t.Fatal("expected true InsecureSkipVerify")
|
|
}
|
|
if rtd.Config.RootCAs != netx.DefaultCertPool() {
|
|
t.Fatal("invalid Config.RootCAs")
|
|
}
|
|
if rtd.Dialer == nil {
|
|
t.Fatal("invalid Dialer")
|
|
}
|
|
if rtd.TLSHandshaker == nil {
|
|
t.Fatal("invalid TLSHandshaker")
|
|
}
|
|
ewth, ok := rtd.TLSHandshaker.(*errorsx.ErrorWrapperTLSHandshaker)
|
|
if !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
if _, ok := ewth.TLSHandshaker.(*netxlite.TLSHandshakerConfigurable); !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewTLSDialerWithNoTLSVerifyAndNoConfig(t *testing.T) {
|
|
td := netx.NewTLSDialer(netx.Config{
|
|
NoTLSVerify: true,
|
|
})
|
|
rtd, ok := td.(*netxlite.TLSDialerLegacy)
|
|
if !ok {
|
|
t.Fatal("not the TLSDialer we expected")
|
|
}
|
|
if len(rtd.Config.NextProtos) != 2 {
|
|
t.Fatal("invalid len(config.NextProtos)")
|
|
}
|
|
if rtd.Config.NextProtos[0] != "h2" || rtd.Config.NextProtos[1] != "http/1.1" {
|
|
t.Fatal("invalid Config.NextProtos")
|
|
}
|
|
if rtd.Config.InsecureSkipVerify != true {
|
|
t.Fatal("expected true InsecureSkipVerify")
|
|
}
|
|
if rtd.Config.RootCAs != netx.DefaultCertPool() {
|
|
t.Fatal("invalid Config.RootCAs")
|
|
}
|
|
if rtd.Dialer == nil {
|
|
t.Fatal("invalid Dialer")
|
|
}
|
|
if rtd.TLSHandshaker == nil {
|
|
t.Fatal("invalid TLSHandshaker")
|
|
}
|
|
ewth, ok := rtd.TLSHandshaker.(*errorsx.ErrorWrapperTLSHandshaker)
|
|
if !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
if _, ok := ewth.TLSHandshaker.(*netxlite.TLSHandshakerConfigurable); !ok {
|
|
t.Fatal("not the TLSHandshaker we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewVanilla(t *testing.T) {
|
|
txp := netx.NewHTTPTransport(netx.Config{})
|
|
uatxp, ok := txp.(httptransport.UserAgentTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
if _, ok := uatxp.RoundTripper.(*http.Transport); !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewWithDialer(t *testing.T) {
|
|
expected := errors.New("mocked error")
|
|
dialer := netx.FakeDialer{Err: expected}
|
|
txp := netx.NewHTTPTransport(netx.Config{
|
|
Dialer: dialer,
|
|
})
|
|
client := &http.Client{Transport: txp}
|
|
resp, err := client.Get("http://www.google.com")
|
|
if !errors.Is(err, expected) {
|
|
t.Fatal("not the error we expected")
|
|
}
|
|
if resp != nil {
|
|
t.Fatal("not the response we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewWithTLSDialer(t *testing.T) {
|
|
expected := errors.New("mocked error")
|
|
tlsDialer := &netxlite.TLSDialerLegacy{
|
|
Config: new(tls.Config),
|
|
Dialer: &mocks.Dialer{
|
|
MockDialContext: func(ctx context.Context, network string, address string) (net.Conn, error) {
|
|
return nil, expected
|
|
},
|
|
MockCloseIdleConnections: func() {
|
|
// nothing
|
|
},
|
|
},
|
|
TLSHandshaker: &netxlite.TLSHandshakerConfigurable{},
|
|
}
|
|
txp := netx.NewHTTPTransport(netx.Config{
|
|
TLSDialer: tlsDialer,
|
|
})
|
|
client := &http.Client{Transport: txp}
|
|
resp, err := client.Get("https://www.google.com")
|
|
if !errors.Is(err, expected) {
|
|
t.Fatal("not the error we expected")
|
|
}
|
|
if resp != nil {
|
|
t.Fatal("not the response we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewWithByteCounter(t *testing.T) {
|
|
counter := bytecounter.New()
|
|
txp := netx.NewHTTPTransport(netx.Config{
|
|
ByteCounter: counter,
|
|
})
|
|
uatxp, ok := txp.(httptransport.UserAgentTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
bctxp, ok := uatxp.RoundTripper.(httptransport.ByteCountingTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
if bctxp.Counter != counter {
|
|
t.Fatal("not the byte counter we expected")
|
|
}
|
|
if _, ok := bctxp.RoundTripper.(*http.Transport); !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewWithLogger(t *testing.T) {
|
|
txp := netx.NewHTTPTransport(netx.Config{
|
|
Logger: log.Log,
|
|
})
|
|
uatxp, ok := txp.(httptransport.UserAgentTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
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.HTTPTransport.(*http.Transport); !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewWithSaver(t *testing.T) {
|
|
saver := new(trace.Saver)
|
|
txp := netx.NewHTTPTransport(netx.Config{
|
|
HTTPSaver: saver,
|
|
})
|
|
uatxp, ok := txp.(httptransport.UserAgentTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
stxptxp, ok := uatxp.RoundTripper.(httptransport.SaverTransactionHTTPTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
if stxptxp.Saver != saver {
|
|
t.Fatal("not the logger we expected")
|
|
}
|
|
sptxp, ok := stxptxp.RoundTripper.(httptransport.SaverPerformanceHTTPTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
if sptxp.Saver != saver {
|
|
t.Fatal("not the logger we expected")
|
|
}
|
|
sbtxp, ok := sptxp.RoundTripper.(httptransport.SaverBodyHTTPTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
if sbtxp.Saver != saver {
|
|
t.Fatal("not the logger we expected")
|
|
}
|
|
smtxp, ok := sbtxp.RoundTripper.(httptransport.SaverMetadataHTTPTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
if smtxp.Saver != saver {
|
|
t.Fatal("not the logger we expected")
|
|
}
|
|
if _, ok := smtxp.RoundTripper.(*http.Transport); !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewDNSClientInvalidURL(t *testing.T) {
|
|
dnsclient, err := netx.NewDNSClient(netx.Config{}, "\t\t\t")
|
|
if err == nil || !strings.HasSuffix(err.Error(), "invalid control character in URL") {
|
|
t.Fatal("not the error we expected")
|
|
}
|
|
if dnsclient.Resolver != nil {
|
|
t.Fatal("expected nil resolver here")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSClientUnsupportedScheme(t *testing.T) {
|
|
dnsclient, err := netx.NewDNSClient(netx.Config{}, "antani:///")
|
|
if err == nil || err.Error() != "unsupported resolver scheme" {
|
|
t.Fatal("not the error we expected")
|
|
}
|
|
if dnsclient.Resolver != nil {
|
|
t.Fatal("expected nil resolver here")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSClientSystemResolver(t *testing.T) {
|
|
dnsclient, err := netx.NewDNSClient(
|
|
netx.Config{}, "system:///")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, ok := dnsclient.Resolver.(*netxlite.ResolverSystem); !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSClientEmpty(t *testing.T) {
|
|
dnsclient, err := netx.NewDNSClient(
|
|
netx.Config{}, "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, ok := dnsclient.Resolver.(*netxlite.ResolverSystem); !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSClientPowerdnsDoH(t *testing.T) {
|
|
dnsclient, err := netx.NewDNSClient(
|
|
netx.Config{}, "doh://powerdns")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
r, ok := dnsclient.Resolver.(resolver.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
if _, ok := r.Transport().(resolver.DNSOverHTTPS); !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSClientGoogleDoH(t *testing.T) {
|
|
dnsclient, err := netx.NewDNSClient(
|
|
netx.Config{}, "doh://google")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
r, ok := dnsclient.Resolver.(resolver.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
if _, ok := r.Transport().(resolver.DNSOverHTTPS); !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSClientCloudflareDoH(t *testing.T) {
|
|
dnsclient, err := netx.NewDNSClient(
|
|
netx.Config{}, "doh://cloudflare")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
r, ok := dnsclient.Resolver.(resolver.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
if _, ok := r.Transport().(resolver.DNSOverHTTPS); !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSClientCloudflareDoHSaver(t *testing.T) {
|
|
saver := new(trace.Saver)
|
|
dnsclient, err := netx.NewDNSClient(
|
|
netx.Config{ResolveSaver: saver}, "doh://cloudflare")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
r, ok := dnsclient.Resolver.(resolver.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
txp, ok := r.Transport().(resolver.SaverDNSTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
if _, ok := txp.RoundTripper.(resolver.DNSOverHTTPS); !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSClientUDP(t *testing.T) {
|
|
dnsclient, err := netx.NewDNSClient(
|
|
netx.Config{}, "udp://8.8.8.8:53")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
r, ok := dnsclient.Resolver.(resolver.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
if _, ok := r.Transport().(resolver.DNSOverUDP); !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSClientUDPDNSSaver(t *testing.T) {
|
|
saver := new(trace.Saver)
|
|
dnsclient, err := netx.NewDNSClient(
|
|
netx.Config{ResolveSaver: saver}, "udp://8.8.8.8:53")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
r, ok := dnsclient.Resolver.(resolver.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
txp, ok := r.Transport().(resolver.SaverDNSTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
if _, ok := txp.RoundTripper.(resolver.DNSOverUDP); !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSClientTCP(t *testing.T) {
|
|
dnsclient, err := netx.NewDNSClient(
|
|
netx.Config{}, "tcp://8.8.8.8:53")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
r, ok := dnsclient.Resolver.(resolver.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
txp, ok := r.Transport().(resolver.DNSOverTCP)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
if txp.Network() != "tcp" {
|
|
t.Fatal("not the Network we expected")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSClientTCPDNSSaver(t *testing.T) {
|
|
saver := new(trace.Saver)
|
|
dnsclient, err := netx.NewDNSClient(
|
|
netx.Config{ResolveSaver: saver}, "tcp://8.8.8.8:53")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
r, ok := dnsclient.Resolver.(resolver.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
txp, ok := r.Transport().(resolver.SaverDNSTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
dotcp, ok := txp.RoundTripper.(resolver.DNSOverTCP)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
if dotcp.Network() != "tcp" {
|
|
t.Fatal("not the Network we expected")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSClientDoT(t *testing.T) {
|
|
dnsclient, err := netx.NewDNSClient(
|
|
netx.Config{}, "dot://8.8.8.8:53")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
r, ok := dnsclient.Resolver.(resolver.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
txp, ok := r.Transport().(resolver.DNSOverTCP)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
if txp.Network() != "dot" {
|
|
t.Fatal("not the Network we expected")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSClientDoTDNSSaver(t *testing.T) {
|
|
saver := new(trace.Saver)
|
|
dnsclient, err := netx.NewDNSClient(
|
|
netx.Config{ResolveSaver: saver}, "dot://8.8.8.8:53")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
r, ok := dnsclient.Resolver.(resolver.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
txp, ok := r.Transport().(resolver.SaverDNSTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
dotls, ok := txp.RoundTripper.(resolver.DNSOverTCP)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
if dotls.Network() != "dot" {
|
|
t.Fatal("not the Network we expected")
|
|
}
|
|
dnsclient.CloseIdleConnections()
|
|
}
|
|
|
|
func TestNewDNSCLientDoTWithoutPort(t *testing.T) {
|
|
c, err := netx.NewDNSClientWithOverrides(
|
|
netx.Config{}, "dot://8.8.8.8", "", "8.8.8.8", "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if c.Resolver.Address() != "8.8.8.8:853" {
|
|
t.Fatal("expected default port to be added")
|
|
}
|
|
}
|
|
|
|
func TestNewDNSCLientTCPWithoutPort(t *testing.T) {
|
|
c, err := netx.NewDNSClientWithOverrides(
|
|
netx.Config{}, "tcp://8.8.8.8", "", "8.8.8.8", "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if c.Resolver.Address() != "8.8.8.8:53" {
|
|
t.Fatal("expected default port to be added")
|
|
}
|
|
}
|
|
|
|
func TestNewDNSCLientUDPWithoutPort(t *testing.T) {
|
|
c, err := netx.NewDNSClientWithOverrides(
|
|
netx.Config{}, "udp://8.8.8.8", "", "8.8.8.8", "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if c.Resolver.Address() != "8.8.8.8:53" {
|
|
t.Fatal("expected default port to be added")
|
|
}
|
|
}
|
|
|
|
func TestNewDNSClientBadDoTEndpoint(t *testing.T) {
|
|
_, err := netx.NewDNSClient(
|
|
netx.Config{}, "dot://bad:endpoint:53")
|
|
if err == nil || !strings.Contains(err.Error(), "too many colons in address") {
|
|
t.Fatal("expected error with bad endpoint")
|
|
}
|
|
}
|
|
|
|
func TestNewDNSClientBadTCPEndpoint(t *testing.T) {
|
|
_, err := netx.NewDNSClient(
|
|
netx.Config{}, "tcp://bad:endpoint:853")
|
|
if err == nil || !strings.Contains(err.Error(), "too many colons in address") {
|
|
t.Fatal("expected error with bad endpoint")
|
|
}
|
|
}
|
|
|
|
func TestNewDNSClientBadUDPEndpoint(t *testing.T) {
|
|
_, err := netx.NewDNSClient(
|
|
netx.Config{}, "udp://bad:endpoint:853")
|
|
if err == nil || !strings.Contains(err.Error(), "too many colons in address") {
|
|
t.Fatal("expected error with bad endpoint")
|
|
}
|
|
}
|
|
|
|
func TestNewDNSCLientWithInvalidTLSVersion(t *testing.T) {
|
|
_, err := netx.NewDNSClientWithOverrides(
|
|
netx.Config{}, "dot://8.8.8.8", "", "", "TLSv999")
|
|
if !errors.Is(err, netxlite.ErrInvalidTLSVersion) {
|
|
t.Fatalf("not the error we expected: %+v", err)
|
|
}
|
|
}
|