bf3c8bcdc3
This diff forward ports b6db4f64dc83a2a27ee3ce6bba5ac93db922832d, whose original log message is the following: - - - We're now using ooni/oohttp as our HTTP library in most cases. A limitation of this library is that net/http/httptrace does not work very well and reliably because (1) we need to use oohttp's version of that code and (2) we cannot observe net events. I noticed this fact because an integration test for collecting HTTP performance metrics was broken. The best solution here is to remove this functionality, since it was basically unused in the repository. Only some integration tests inside urlgetter bothered with these metrics. A more clinical fix would have been to use ooni/oohttp/httptrace instead of net/http/httptrace in the stdlib, but it does not seem to be a good idea, given that those metrics were not used. With this diff applied, we'll further reduce the number of locally failing integration tests to just jafar-specific tests. This diff WILL need to be forwardported to `master`.
839 lines
22 KiB
Go
839 lines
22 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/model/mocks"
|
|
"github.com/ooni/probe-cli/v3/internal/netxlite"
|
|
)
|
|
|
|
func TestNewResolverVanilla(t *testing.T) {
|
|
r := netx.NewResolver(netx.Config{})
|
|
ir, ok := r.(*netxlite.ResolverIDNA)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ewr, ok := ir.Resolver.(*netxlite.ErrorWrapperResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ar, ok := ewr.Resolver.(*netxlite.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.(*netxlite.ResolverIDNA)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ewr, ok := ir.Resolver.(*netxlite.ErrorWrapperResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ar, ok := ewr.Resolver.(*netxlite.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.(*netxlite.ResolverIDNA)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ewr, ok := ir.Resolver.(*netxlite.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.(*netxlite.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.(*netxlite.ResolverIDNA)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
lr, ok := ir.Resolver.(*netxlite.ResolverLogger)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
if lr.Logger != log.Log {
|
|
t.Fatal("not the logger we expected")
|
|
}
|
|
ewr, ok := lr.Resolver.(*netxlite.ErrorWrapperResolver)
|
|
if !ok {
|
|
t.Fatalf("not the resolver we expected")
|
|
}
|
|
ar, ok := ewr.Resolver.(*netxlite.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.(*netxlite.ResolverIDNA)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
sr, ok := ir.Resolver.(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.(*netxlite.ErrorWrapperResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ar, ok := ewr.Resolver.(*netxlite.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.(*netxlite.ResolverIDNA)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ewr, ok := ir.Resolver.(*netxlite.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.(*netxlite.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.(*netxlite.ResolverIDNA)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
ewr, ok := ir.Resolver.(*netxlite.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.(*netxlite.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.(*netxlite.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.(*netxlite.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.DebugLogger != log.Log {
|
|
t.Fatal("not the Logger we expected")
|
|
}
|
|
ewth, ok := lth.TLSHandshaker.(*netxlite.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.(*netxlite.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.(*netxlite.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.(*netxlite.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{})
|
|
if _, ok := txp.(*httptransport.SystemTransportWrapper); !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,
|
|
})
|
|
bctxp, ok := txp.(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.HTTPTransport.(*httptransport.SystemTransportWrapper); !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewWithLogger(t *testing.T) {
|
|
txp := netx.NewHTTPTransport(netx.Config{
|
|
Logger: log.Log,
|
|
})
|
|
ltxp, ok := txp.(*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.(*httptransport.SystemTransportWrapper); !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
}
|
|
|
|
func TestNewWithSaver(t *testing.T) {
|
|
saver := new(trace.Saver)
|
|
txp := netx.NewHTTPTransport(netx.Config{
|
|
HTTPSaver: saver,
|
|
})
|
|
stxptxp, ok := txp.(httptransport.SaverTransactionHTTPTransport)
|
|
if !ok {
|
|
t.Fatal("not the transport we expected")
|
|
}
|
|
if stxptxp.Saver != saver {
|
|
t.Fatal("not the logger we expected")
|
|
}
|
|
sbtxp, ok := stxptxp.HTTPTransport.(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.HTTPTransport.(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.HTTPTransport.(*httptransport.SystemTransportWrapper); !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 != nil {
|
|
t.Fatal("expected nil resolver here")
|
|
}
|
|
}
|
|
|
|
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 != nil {
|
|
t.Fatal("expected nil resolver here")
|
|
}
|
|
}
|
|
|
|
func TestNewDNSClientSystemResolver(t *testing.T) {
|
|
dnsclient, err := netx.NewDNSClient(
|
|
netx.Config{}, "system:///")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, ok := dnsclient.(*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.(*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.(*netxlite.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
if _, ok := r.Transport().(*netxlite.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.(*netxlite.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
if _, ok := r.Transport().(*netxlite.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.(*netxlite.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
if _, ok := r.Transport().(*netxlite.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.(*netxlite.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.DNSTransport.(*netxlite.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.(*netxlite.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
if _, ok := r.Transport().(*netxlite.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.(*netxlite.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.DNSTransport.(*netxlite.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.(*netxlite.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
txp, ok := r.Transport().(*netxlite.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.(*netxlite.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.DNSTransport.(*netxlite.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.(*netxlite.SerialResolver)
|
|
if !ok {
|
|
t.Fatal("not the resolver we expected")
|
|
}
|
|
txp, ok := r.Transport().(*netxlite.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.(*netxlite.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.DNSTransport.(*netxlite.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.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.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.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)
|
|
}
|
|
}
|