From 3f66df222e55f3fcec8077bd0e1edcdb93990bdb Mon Sep 17 00:00:00 2001 From: Daniel Date: Wed, 22 May 2019 16:05:57 +0200 Subject: [PATCH] Fix intel tests --- intel/clients.go | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/intel/clients.go b/intel/clients.go index aa281cd0..4e6302bd 100644 --- a/intel/clients.go +++ b/intel/clients.go @@ -20,6 +20,13 @@ func SetLocalAddrFactory(laf func(network string) net.Addr) { } } +func getLocalAddr(network string) net.Addr { + if localAddrFactory != nil { + return localAddrFactory(network) + } + return nil +} + type clientManager struct { dnsClient *dns.Client factory func() *dns.Client @@ -38,7 +45,7 @@ func newDNSClientManager(resolver *Resolver) *clientManager { return &dns.Client{ Timeout: 5 * time.Second, Dialer: &net.Dialer{ - LocalAddr: localAddrFactory("udp"), + LocalAddr: getLocalAddr("udp"), }, } }, @@ -53,7 +60,7 @@ func newTCPClientManager(resolver *Resolver) *clientManager { Net: "tcp", Timeout: 5 * time.Second, Dialer: &net.Dialer{ - LocalAddr: localAddrFactory("tcp"), + LocalAddr: getLocalAddr("tcp"), }, } }, @@ -74,7 +81,7 @@ func newTLSClientManager(resolver *Resolver) *clientManager { }, Timeout: 5 * time.Second, Dialer: &net.Dialer{ - LocalAddr: localAddrFactory("tcp"), + LocalAddr: getLocalAddr("tcp"), }, } }, @@ -94,7 +101,7 @@ func newHTTPSClientManager(resolver *Resolver) *clientManager { }, Timeout: 5 * time.Second, Dialer: &net.Dialer{ - LocalAddr: localAddrFactory("tcp"), + LocalAddr: getLocalAddr("tcp"), }, } if resolver.VerifyDomain != "" {