diff options
author | 2016-04-11 15:56:22 +0100 | |
---|---|---|
committer | 2016-04-11 15:56:22 +0100 | |
commit | 9f651a397baabcd7d1876771f8568e14d7b4b6ef (patch) | |
tree | 8328a477dc20f17709b2233a8ec0ed76d6816280 /test | |
parent | 31ce53f514411fb6d113fd439663a710dc02fafc (diff) | |
download | coredns-9f651a397baabcd7d1876771f8568e14d7b4b6ef.tar.gz coredns-9f651a397baabcd7d1876771f8568e14d7b4b6ef.tar.zst coredns-9f651a397baabcd7d1876771f8568e14d7b4b6ef.zip |
All (non etcd) tests are now local (#105)
We don't need to network to do tests, we up enough local servers
to we don't need to forward to,s say 8.8.8.8
Diffstat (limited to 'test')
-rw-r--r-- | test/proxy_test.go | 59 | ||||
-rw-r--r-- | test/server_test.go | 8 | ||||
-rw-r--r-- | test/tests.go | 8 |
3 files changed, 67 insertions, 8 deletions
diff --git a/test/proxy_test.go b/test/proxy_test.go new file mode 100644 index 000000000..9365570b6 --- /dev/null +++ b/test/proxy_test.go @@ -0,0 +1,59 @@ +package test + +import ( + "io/ioutil" + "log" + "os" + "testing" + + "github.com/miekg/coredns/middleware" + "github.com/miekg/coredns/middleware/proxy" + "github.com/miekg/coredns/middleware/test" + + "github.com/miekg/dns" +) + +const exampleOrg = `; example.org test file +example.org. IN SOA sns.dns.icann.org. noc.dns.icann.org. 2015082541 7200 3600 1209600 3600 +example.org. IN NS b.iana-servers.net. +example.org. IN NS a.iana-servers.net. +example.org. IN A 127.0.0.1 +` + +func TestLookupProxy(t *testing.T) { + name, rm, err := test.Zone(t, ".", exampleOrg) + if err != nil { + t.Fatalf("failed to created zone: %s", err) + } + defer rm() + + corefile := `example.org:0 { + file ` + name + ` +} +` + ex, _, udp, err := Server(t, corefile) + if err != nil { + t.Fatalf("Could get server: %s", err) + } + defer ex.Stop() + + log.SetOutput(ioutil.Discard) + defer log.SetOutput(os.Stderr) + + p := proxy.New([]string{udp}) + state := middleware.State{W: &test.ResponseWriter{}, Req: new(dns.Msg)} + resp, err := p.Lookup(state, "example.org.", dns.TypeA) + if err != nil { + t.Error("Expected to receive reply, but didn't") + } + // expect answer section with A record in it + if len(resp.Answer) == 0 { + t.Error("Expected to at least one RR in the answer section, got none") + } + if resp.Answer[0].Header().Rrtype != dns.TypeA { + t.Error("Expected RR to A, got: %d", resp.Answer[0].Header().Rrtype) + } + if resp.Answer[0].(*dns.A).A.String() != "127.0.0.1" { + t.Error("Expected 127.0.0.1, got: %d", resp.Answer[0].(*dns.A).A.String()) + } +} diff --git a/test/server_test.go b/test/server_test.go index 6020e76d3..6a86d022b 100644 --- a/test/server_test.go +++ b/test/server_test.go @@ -12,7 +12,7 @@ func TestProxyToChaosServer(t *testing.T) { chaos CoreDNS-001 miek@miek.nl } ` - chaos, tcpCH, udpCH, err := testServer(t, corefile) + chaos, tcpCH, udpCH, err := Server(t, corefile) if err != nil { t.Fatalf("Could get server: %s", err) } @@ -22,7 +22,7 @@ func TestProxyToChaosServer(t *testing.T) { proxy . ` + udpCH + ` } ` - proxy, _, udp, err := testServer(t, corefileProxy) + proxy, _, udp, err := Server(t, corefileProxy) if err != nil { t.Fatalf("Could get server: %s", err) } @@ -37,10 +37,10 @@ func TestProxyToChaosServer(t *testing.T) { } func chaosTest(t *testing.T, server, net string) { - m := testMsg("version.bind.", dns.TypeTXT, nil) + m := Msg("version.bind.", dns.TypeTXT, nil) m.Question[0].Qclass = dns.ClassCHAOS - r, err := testExchange(m, server, net) + r, err := Exchange(m, server, net) if err != nil { t.Fatalf("Could not send message: %s", err) } diff --git a/test/tests.go b/test/tests.go index 2b11a5ac9..a97541f8e 100644 --- a/test/tests.go +++ b/test/tests.go @@ -11,7 +11,7 @@ import ( "github.com/miekg/dns" ) -func testMsg(zone string, typ uint16, o *dns.OPT) *dns.Msg { +func Msg(zone string, typ uint16, o *dns.OPT) *dns.Msg { m := new(dns.Msg) m.SetQuestion(zone, typ) if o != nil { @@ -20,14 +20,14 @@ func testMsg(zone string, typ uint16, o *dns.OPT) *dns.Msg { return m } -func testExchange(m *dns.Msg, server, net string) (*dns.Msg, error) { +func Exchange(m *dns.Msg, server, net string) (*dns.Msg, error) { c := new(dns.Client) c.Net = net return middleware.Exchange(c, m, server) } -// testServer returns a test server and the tcp and udp listeners addresses. -func testServer(t *testing.T, corefile string) (*server.Server, string, string, error) { +// Server returns a test server and the tcp and udp listeners addresses. +func Server(t *testing.T, corefile string) (*server.Server, string, string, error) { srv, err := core.TestServer(t, corefile) if err != nil { return nil, "", "", err |