mirror of
				https://github.com/coredns/coredns.git
				synced 2025-11-04 03:03:14 -05:00 
			
		
		
		
	These are found with: `pcregrep -M "}\n\n}" **/*.go` Sometimes a unneeded newline is inserted, remove those. Signed-off-by: Miek Gieben <miek@miek.nl>
		
			
				
	
	
		
			90 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			90 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package test
 | 
						|
 | 
						|
import (
 | 
						|
	"testing"
 | 
						|
 | 
						|
	"github.com/coredns/coredns/plugin/test"
 | 
						|
 | 
						|
	"github.com/miekg/dns"
 | 
						|
)
 | 
						|
 | 
						|
func TestLookupWildcard(t *testing.T) {
 | 
						|
	t.Parallel()
 | 
						|
	name, rm, err := test.TempFile(".", exampleOrg)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatalf("Failed to create zone: %s", err)
 | 
						|
	}
 | 
						|
	defer rm()
 | 
						|
 | 
						|
	corefile := `example.org:0 {
 | 
						|
		file ` + name + `
 | 
						|
	}`
 | 
						|
 | 
						|
	i, udp, _, err := CoreDNSServerAndPorts(corefile)
 | 
						|
	if err != nil {
 | 
						|
		t.Fatalf("Could not get CoreDNS serving instance: %s", err)
 | 
						|
	}
 | 
						|
	defer i.Stop()
 | 
						|
 | 
						|
	for _, lookup := range []string{"a.w.example.org.", "a.a.w.example.org."} {
 | 
						|
		m := new(dns.Msg)
 | 
						|
		m.SetQuestion(lookup, dns.TypeTXT)
 | 
						|
		resp, err := dns.Exchange(m, udp)
 | 
						|
		if err != nil || resp == nil {
 | 
						|
			t.Fatalf("Expected to receive reply, but didn't for %s", lookup)
 | 
						|
		}
 | 
						|
 | 
						|
		// ;; ANSWER SECTION:
 | 
						|
		// a.w.example.org.          1800    IN      TXT     "Wildcard"
 | 
						|
		if resp.Rcode != dns.RcodeSuccess {
 | 
						|
			t.Errorf("Expected NOERROR RCODE, got %s for %s", dns.RcodeToString[resp.Rcode], lookup)
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		if len(resp.Answer) == 0 {
 | 
						|
			t.Errorf("Expected to at least one RR in the answer section, got none for %s TXT", lookup)
 | 
						|
			t.Logf("%s", resp)
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		if resp.Answer[0].Header().Name != lookup {
 | 
						|
			t.Errorf("Expected name to be %s, got: %s for TXT", lookup, resp.Answer[0].Header().Name)
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		if resp.Answer[0].Header().Rrtype != dns.TypeTXT {
 | 
						|
			t.Errorf("Expected RR to be TXT, got: %d, for %s TXT", resp.Answer[0].Header().Rrtype, lookup)
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		if resp.Answer[0].(*dns.TXT).Txt[0] != "Wildcard" {
 | 
						|
			t.Errorf("Expected Wildcard, got: %s, for %s TXT", resp.Answer[0].(*dns.TXT).Txt[0], lookup)
 | 
						|
			continue
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	for _, lookup := range []string{"w.example.org.", "a.w.example.org.", "a.a.w.example.org."} {
 | 
						|
		m := new(dns.Msg)
 | 
						|
		m.SetQuestion(lookup, dns.TypeSRV)
 | 
						|
		resp, err := dns.Exchange(m, udp)
 | 
						|
		if err != nil || resp == nil {
 | 
						|
			t.Fatal("Expected to receive reply, but didn't", lookup)
 | 
						|
		}
 | 
						|
 | 
						|
		// ;; AUTHORITY SECTION:
 | 
						|
		// example.org.              1800    IN      SOA     linode.atoom.net. miek.miek.nl. 1454960557 14400 3600 604800 14400
 | 
						|
		if resp.Rcode != dns.RcodeSuccess {
 | 
						|
			t.Errorf("Expected NOERROR RCODE, got %s for %s", dns.RcodeToString[resp.Rcode], lookup)
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		if len(resp.Answer) != 0 {
 | 
						|
			t.Errorf("Expected zero RRs in the answer section, got some, for %s SRV", lookup)
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		if len(resp.Ns) == 0 {
 | 
						|
			t.Errorf("Expected to at least one RR in the authority section, got none, for %s SRV", lookup)
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		if resp.Ns[0].Header().Rrtype != dns.TypeSOA {
 | 
						|
			t.Errorf("Expected RR to be SOA, got: %d, for %s SRV", resp.Ns[0].Header().Rrtype, lookup)
 | 
						|
			continue
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 |