mirror of
https://github.com/coredns/coredns.git
synced 2025-11-04 03:03:14 -05:00
49 lines
979 B
Go
49 lines
979 B
Go
|
package test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"log"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
// Plug in CoreDNS, needed for AppVersion and AppName in this test.
|
||
|
|
_ "github.com/coredns/coredns/coremain"
|
||
|
|
|
||
|
|
"github.com/mholt/caddy"
|
||
|
|
"github.com/miekg/dns"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestChaos(t *testing.T) {
|
||
|
|
corefile := `.:0 {
|
||
|
|
chaos
|
||
|
|
}
|
||
|
|
`
|
||
|
|
|
||
|
|
i, err := CoreDNSServer(corefile)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Could not get CoreDNS serving instance: %s", err)
|
||
|
|
}
|
||
|
|
// Stop the server.
|
||
|
|
defer i.Stop()
|
||
|
|
|
||
|
|
udp, _ := CoreDNSServerPorts(i, 0)
|
||
|
|
if udp == "" {
|
||
|
|
t.Fatalf("Could not get UDP listening port")
|
||
|
|
}
|
||
|
|
|
||
|
|
log.SetOutput(ioutil.Discard)
|
||
|
|
|
||
|
|
m := new(dns.Msg)
|
||
|
|
m.SetQuestion("version.bind.", dns.TypeTXT)
|
||
|
|
m.Question[0].Qclass = dns.ClassCHAOS
|
||
|
|
|
||
|
|
resp, err := dns.Exchange(m, udp)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Expected to receive reply, but didn't: %v", err)
|
||
|
|
}
|
||
|
|
chTxt := resp.Answer[0].(*dns.TXT).Txt[0]
|
||
|
|
version := caddy.AppName + "-" + caddy.AppVersion
|
||
|
|
if chTxt != version {
|
||
|
|
t.Fatalf("Expected version to bo %s, got %s", version, chTxt)
|
||
|
|
}
|
||
|
|
}
|