diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/file_cname_proxy_test.go | 44 |
1 files changed, 43 insertions, 1 deletions
diff --git a/test/file_cname_proxy_test.go b/test/file_cname_proxy_test.go index 61bd8764e..a6ab8e59e 100644 --- a/test/file_cname_proxy_test.go +++ b/test/file_cname_proxy_test.go @@ -12,7 +12,7 @@ import ( "github.com/miekg/dns" ) -func TestZoneExternalCNAMELookup(t *testing.T) { +func TestZoneExternalCNAMELookupWithoutProxy(t *testing.T) { t.Parallel() log.SetOutput(ioutil.Discard) @@ -50,3 +50,45 @@ func TestZoneExternalCNAMELookup(t *testing.T) { t.Fatalf("Expected 1 RR in answer section got %d", len(resp.Answer)) } } + +func TestZoneExternalCNAMELookupWithProxy(t *testing.T) { + t.Parallel() + log.SetOutput(ioutil.Discard) + + name, rm, err := TempFile(".", exampleOrg) + if err != nil { + t.Fatalf("Failed to create zone: %s", err) + } + defer rm() + + // Corefile with for example without proxy section. + corefile := `example.org:0 { + file ` + name + ` { + upstream 8.8.8.8 + } +} +` + i, err := CoreDNSServer(corefile) + if err != nil { + t.Fatalf("Could not get CoreDNS serving instance: %s", err) + } + + udp, _ := CoreDNSServerPorts(i, 0) + if udp == "" { + t.Fatalf("Could not get UDP listening port") + } + defer i.Stop() + + p := proxy.NewLookup([]string{udp}) + state := request.Request{W: &test.ResponseWriter{}, Req: new(dns.Msg)} + + resp, err := p.Lookup(state, "cname.example.org.", dns.TypeA) + if err != nil { + t.Fatalf("Expected to receive reply, but didn't: %s", err) + } + // There should be a CNAME *and* an IP address in the answer section. + // For now, just check that we have 2 RRs + if len(resp.Answer) != 2 { + t.Fatalf("Expected 2 RRs in answer section got %d", len(resp.Answer)) + } +} |