mirror of
https://github.com/AdguardTeam/AdGuardHome.git
synced 2024-11-16 18:38:52 -07:00
c9d2436d77
Merge in DNS/adguard-home from 2574-external-tests-3 to master Updates #2574. Squashed commit of the following: commit 29d429c65dee2621ca503710a7ba9522f14f55f9 Author: Eugene Burkov <e.burkov@adguard.com> Date: Thu Feb 4 20:06:57 2021 +0300 all: finally fix spacing commit 9e3a3be63b74852a7802e3f1832648444b58e4d0 Author: Eugene Burkov <e.burkov@adguard.com> Date: Thu Feb 4 19:59:09 2021 +0300 aghtest: polish spacing commit 8a984159fe813b95b989803f5b8b78d01a41bd39 Author: Eugene Burkov <e.burkov@adguard.com> Date: Thu Feb 4 18:44:47 2021 +0300 all: fix linux tests, imp code quality commit 0c1b42bacba1b23fa847e1fa032579c525b3eaa1 Author: Eugene Burkov <e.burkov@adguard.com> Date: Thu Feb 4 17:33:12 2021 +0300 all: mv testutil to aghtest package, imp tests
35 lines
1.0 KiB
Go
35 lines
1.0 KiB
Go
package home
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/AdguardTeam/AdGuardHome/internal/dnsforward"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func prepareTestDNSServer() error {
|
|
config.DNS.Port = 1234
|
|
Context.dnsServer = dnsforward.NewServer(dnsforward.DNSCreateParams{})
|
|
conf := &dnsforward.ServerConfig{}
|
|
conf.UpstreamDNS = []string{"8.8.8.8"}
|
|
|
|
return Context.dnsServer.Prepare(conf)
|
|
}
|
|
|
|
// TODO(e.burkov): It's kind of complicated to get rid of network access in this
|
|
// test. The thing is that *Whois creates new *net.Dialer each time it requests
|
|
// the server, so it becomes hard to simulate handling of request from test even
|
|
// with substituted upstream. However, it must be done.
|
|
func TestWhois(t *testing.T) {
|
|
assert.Nil(t, prepareTestDNSServer())
|
|
|
|
w := Whois{timeoutMsec: 5000}
|
|
resp, err := w.queryAll(context.Background(), "8.8.8.8")
|
|
assert.Nil(t, err)
|
|
m := whoisParse(resp)
|
|
assert.Equal(t, "Google LLC", m["orgname"])
|
|
assert.Equal(t, "US", m["country"])
|
|
assert.Equal(t, "Mountain View", m["city"])
|
|
}
|