aboutsummaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authorGravatar Miek Gieben <miek@miek.nl> 2016-04-11 15:56:22 +0100
committerGravatar Miek Gieben <miek@miek.nl> 2016-04-11 15:56:22 +0100
commit9f651a397baabcd7d1876771f8568e14d7b4b6ef (patch)
tree8328a477dc20f17709b2233a8ec0ed76d6816280 /test
parent31ce53f514411fb6d113fd439663a710dc02fafc (diff)
downloadcoredns-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.go59
-rw-r--r--test/server_test.go8
-rw-r--r--test/tests.go8
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