63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
|
package dns
|
||
|
|
|
||
|
|
// Tests that solve that an specific issue.
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestTCPRtt(t *testing.T) {
|
||
|
|
m := new(Msg)
|
||
|
|
m.RecursionDesired = true
|
||
|
|
m.SetQuestion("example.org.", TypeA)
|
||
|
|
|
||
|
|
c := &Client{}
|
||
|
|
for _, proto := range []string{"udp", "tcp"} {
|
||
|
|
c.Net = proto
|
||
|
|
_, rtt, err := c.Exchange(m, "8.8.4.4:53")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if rtt == 0 {
|
||
|
|
t.Fatalf("expecting non zero rtt %s, got zero", c.Net)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNSEC3MissingSalt(t *testing.T) {
|
||
|
|
rr := testRR("ji6neoaepv8b5o6k4ev33abha8ht9fgc.example. NSEC3 1 1 12 aabbccdd K8UDEMVP1J2F7EG6JEBPS17VP3N8I58H")
|
||
|
|
m := new(Msg)
|
||
|
|
m.Answer = []RR{rr}
|
||
|
|
mb, err := m.Pack()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("expected to pack message. err: %s", err)
|
||
|
|
}
|
||
|
|
if err := m.Unpack(mb); err != nil {
|
||
|
|
t.Fatalf("expected to unpack message. missing salt? err: %s", err)
|
||
|
|
}
|
||
|
|
in := rr.(*NSEC3).Salt
|
||
|
|
out := m.Answer[0].(*NSEC3).Salt
|
||
|
|
if in != out {
|
||
|
|
t.Fatalf("expected salts to match. packed: `%s`. returned: `%s`", in, out)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNSEC3MixedNextDomain(t *testing.T) {
|
||
|
|
rr := testRR("ji6neoaepv8b5o6k4ev33abha8ht9fgc.example. NSEC3 1 1 12 - k8udemvp1j2f7eg6jebps17vp3n8i58h")
|
||
|
|
m := new(Msg)
|
||
|
|
m.Answer = []RR{rr}
|
||
|
|
mb, err := m.Pack()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("expected to pack message. err: %s", err)
|
||
|
|
}
|
||
|
|
if err := m.Unpack(mb); err != nil {
|
||
|
|
t.Fatalf("expected to unpack message. err: %s", err)
|
||
|
|
}
|
||
|
|
in := strings.ToUpper(rr.(*NSEC3).NextDomain)
|
||
|
|
out := m.Answer[0].(*NSEC3).NextDomain
|
||
|
|
if in != out {
|
||
|
|
t.Fatalf("expected round trip to produce NextDomain `%s`, instead `%s`", in, out)
|
||
|
|
}
|
||
|
|
}
|