391e619979
Close #1535 Squashed commit of the following: commit 7755d7a50e48db5d341029e1c21da95a396d9090 Merge: e24e945f5c2ca690Author: Simon Zolin <s.zolin@adguard.com> Date: Wed May 27 14:41:24 2020 +0300 Merge remote-tracking branch 'origin/master' into 1535-whois-resolve commit e24e945f9b949e3cabcddb3a8a62962bb7a5b0b9 Merge: db764710355e634fAuthor: Simon Zolin <s.zolin@adguard.com> Date: Wed May 27 14:29:44 2020 +0300 Merge remote-tracking branch 'origin/master' into 1535-whois-resolve commit db764710a0f0b60293f7bdf6b0d7b858d56819e0 Author: Simon Zolin <s.zolin@adguard.com> Date: Wed May 27 12:57:00 2020 +0300 fix commit dc38f668d5845cfd696b3107a8f02dda4f10298e Author: Simon Zolin <s.zolin@adguard.com> Date: Wed May 27 12:44:10 2020 +0300 * whois: use the upstream server to resolve whois server hostname
36 lines
843 B
Go
36 lines
843 B
Go
package home
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/AdguardTeam/AdGuardHome/dnsforward"
|
|
"github.com/AdguardTeam/dnsproxy/proxy"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func prepareTestDNSServer() error {
|
|
config.DNS.Port = 1234
|
|
Context.dnsServer = dnsforward.NewServer(nil, nil, nil)
|
|
conf := &dnsforward.ServerConfig{}
|
|
uc, err := proxy.ParseUpstreamsConfig([]string{"1.1.1.1"}, nil, time.Second*5)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
conf.UpstreamConfig = &uc
|
|
return Context.dnsServer.Prepare(conf)
|
|
}
|
|
|
|
func TestWhois(t *testing.T) {
|
|
err := prepareTestDNSServer()
|
|
assert.Nil(t, err)
|
|
|
|
w := Whois{timeoutMsec: 5000}
|
|
resp, err := w.queryAll("8.8.8.8")
|
|
assert.True(t, err == nil)
|
|
m := whoisParse(resp)
|
|
assert.True(t, m["orgname"] == "Google LLC")
|
|
assert.True(t, m["country"] == "US")
|
|
assert.True(t, m["city"] == "Mountain View")
|
|
}
|