mirror of
				https://github.com/coredns/coredns.git
				synced 2025-10-31 02:03:20 -04:00 
			
		
		
		
	Change the CI setup for K8s to be simpler. Now it just creates a set of objects via a yaml file, making it very easy to modify the tests. Implement PTR for services.
		
			
				
	
	
		
			248 lines
		
	
	
		
			8.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			248 lines
		
	
	
		
			8.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // +build k8s
 | |
| 
 | |
| package test
 | |
| 
 | |
| import (
 | |
| 	"testing"
 | |
| 	"time"
 | |
| 
 | |
| 	"github.com/miekg/coredns/middleware/test"
 | |
| 
 | |
| 	"github.com/mholt/caddy"
 | |
| 	"github.com/miekg/dns"
 | |
| )
 | |
| 
 | |
| // Test data
 | |
| // TODO: Fix the actual RR values
 | |
| 
 | |
| var dnsTestCases = []test.Case{
 | |
| 	{
 | |
| 		Qname: "svc-1-a.test-1.svc.cluster.local.", Qtype: dns.TypeA,
 | |
| 		Rcode: dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.A("svc-1-a.test-1.svc.cluster.local.      303    IN      A       10.3.0.100"),
 | |
| 		},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "bogusservice.test-1.svc.cluster.local.", Qtype: dns.TypeA,
 | |
| 		Rcode:  dns.RcodeNameError,
 | |
| 		Answer: []dns.RR{},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "svc-1-a.*.svc.cluster.local.", Qtype: dns.TypeA,
 | |
| 		Rcode: dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.A("svc-1-a.test-1.svc.cluster.local.      303    IN      A       10.3.0.100"),
 | |
| 		},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "svc-1-a.any.svc.cluster.local.", Qtype: dns.TypeA,
 | |
| 		Rcode: dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.A("svc-1-a.test-1.svc.cluster.local.      303    IN      A       10.3.0.100"),
 | |
| 		},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "bogusservice.*.svc.cluster.local.", Qtype: dns.TypeA,
 | |
| 		Rcode:  dns.RcodeNameError,
 | |
| 		Answer: []dns.RR{},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "bogusservice.any.svc.cluster.local.", Qtype: dns.TypeA,
 | |
| 		Rcode:  dns.RcodeNameError,
 | |
| 		Answer: []dns.RR{},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "*.test-1.svc.cluster.local.", Qtype: dns.TypeA,
 | |
| 		Rcode: dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.A("svc-1-a.test-1.svc.cluster.local.      303    IN      A       10.3.0.100"),
 | |
| 			test.A("svc-1-b.test-1.svc.cluster.local.      303    IN      A       10.3.0.110"),
 | |
| 			test.A("svc-c.test-1.svc.cluster.local.        303    IN      A       10.3.0.115"),
 | |
| 		},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "any.test-1.svc.cluster.local.", Qtype: dns.TypeA,
 | |
| 		Rcode: dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.A("svc-1-a.test-1.svc.cluster.local.      303    IN      A       10.3.0.100"),
 | |
| 			test.A("svc-1-b.test-1.svc.cluster.local.      303    IN      A       10.3.0.110"),
 | |
| 			test.A("svc-c.test-1.svc.cluster.local.        303    IN      A       10.3.0.115"),
 | |
| 		},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "any.test-2.svc.cluster.local.", Qtype: dns.TypeA,
 | |
| 		Rcode:  dns.RcodeNameError,
 | |
| 		Answer: []dns.RR{},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "*.test-2.svc.cluster.local.", Qtype: dns.TypeA,
 | |
| 		Rcode:  dns.RcodeNameError,
 | |
| 		Answer: []dns.RR{},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "*.*.svc.cluster.local.", Qtype: dns.TypeA,
 | |
| 		Rcode: dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.A("svc-1-a.test-1.svc.cluster.local.      303    IN      A       10.3.0.100"),
 | |
| 			test.A("svc-1-b.test-1.svc.cluster.local.      303    IN      A       10.3.0.110"),
 | |
| 			test.A("svc-c.test-1.svc.cluster.local.        303    IN      A       10.3.0.115"),
 | |
| 		},
 | |
| 	},
 | |
| 	//TODO: Fix below to all use test.SRV not test.A!
 | |
| 	{
 | |
| 		Qname: "svc-1-a.test-1.svc.cluster.local.", Qtype: dns.TypeSRV,
 | |
| 		Rcode: dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.SRV("_http._tcp.svc-1-a.test-1.svc.cluster.local.      303    IN    SRV 10 100 80 svc-1-a.test-1.svc.cluster.local."),
 | |
| 			test.SRV("_https._tcp.svc-1-a.test-1.svc.cluster.local.      303    IN    SRV 10 100 443 svc-1-a.test-1.svc.cluster.local."),
 | |
| 		},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "bogusservice.test-1.svc.cluster.local.", Qtype: dns.TypeSRV,
 | |
| 		Rcode:  dns.RcodeNameError,
 | |
| 		Answer: []dns.RR{},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "svc-1-a.*.svc.cluster.local.", Qtype: dns.TypeSRV,
 | |
| 		Rcode: dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.SRV("_http._tcp.svc-1-a.test-1.svc.cluster.local.      303    IN    SRV 10 100 80 svc-1-a.test-1.svc.cluster.local."),
 | |
| 			test.SRV("_https._tcp.svc-1-a.test-1.svc.cluster.local.      303    IN    SRV 10 100 443 svc-1-a.test-1.svc.cluster.local."),
 | |
| 		},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "svc-1-a.any.svc.cluster.local.", Qtype: dns.TypeSRV,
 | |
| 		Rcode: dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.SRV("_http._tcp.svc-1-a.test-1.svc.cluster.local.      303    IN    SRV 10 100 80 svc-1-a.test-1.svc.cluster.local."),
 | |
| 			test.SRV("_https._tcp.svc-1-a.test-1.svc.cluster.local.      303    IN    SRV 10 100 443 svc-1-a.test-1.svc.cluster.local."),
 | |
| 		},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "bogusservice.*.svc.cluster.local.", Qtype: dns.TypeSRV,
 | |
| 		Rcode:  dns.RcodeNameError,
 | |
| 		Answer: []dns.RR{},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "bogusservice.any.svc.cluster.local.", Qtype: dns.TypeSRV,
 | |
| 		Rcode:  dns.RcodeNameError,
 | |
| 		Answer: []dns.RR{},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "*.test-1.svc.cluster.local.", Qtype: dns.TypeSRV,
 | |
| 		Rcode: dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.SRV("_http._tcp.svc-1-a.test-1.svc.cluster.local.      303    IN    SRV 10 100 80 svc-1-a.test-1.svc.cluster.local."),
 | |
| 			test.SRV("_https._tcp.svc-1-a.test-1.svc.cluster.local.      303    IN    SRV 10 100 443 svc-1-a.test-1.svc.cluster.local."),
 | |
| 			test.SRV("_http._tcp.svc-1-b.test-1.svc.cluster.local.      303    IN    SRV 10 100 80 svc-1-b.test-1.svc.cluster.local."),
 | |
| 			test.SRV("_c-port._udp.svc-c.test-1.svc.cluster.local.      303    IN    SRV 10 100 1234 svc-c.test-1.svc.cluster.local."),
 | |
| 		},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "any.test-1.svc.cluster.local.", Qtype: dns.TypeSRV,
 | |
| 		Rcode: dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.SRV("_http._tcp.svc-1-a.test-1.svc.cluster.local.      303    IN    SRV 10 100 80 svc-1-a.test-1.svc.cluster.local."),
 | |
| 			test.SRV("_https._tcp.svc-1-a.test-1.svc.cluster.local.      303    IN    SRV 10 100 443 svc-1-a.test-1.svc.cluster.local."),
 | |
| 			test.SRV("_http._tcp.svc-1-b.test-1.svc.cluster.local.      303    IN    SRV 10 100 80 svc-1-b.test-1.svc.cluster.local."),
 | |
| 			test.SRV("_c-port._udp.svc-c.test-1.svc.cluster.local.      303    IN    SRV 10 100 1234 svc-c.test-1.svc.cluster.local."),
 | |
| 		},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "any.test-2.svc.cluster.local.", Qtype: dns.TypeSRV,
 | |
| 		Rcode:  dns.RcodeNameError,
 | |
| 		Answer: []dns.RR{},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "*.test-2.svc.cluster.local.", Qtype: dns.TypeSRV,
 | |
| 		Rcode:  dns.RcodeNameError,
 | |
| 		Answer: []dns.RR{},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "*.*.svc.cluster.local.", Qtype: dns.TypeSRV,
 | |
| 		Rcode: dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.SRV("_http._tcp.svc-1-a.test-1.svc.cluster.local.      303    IN    SRV 10 100 80 svc-1-a.test-1.svc.cluster.local."),
 | |
| 			test.SRV("_https._tcp.svc-1-a.test-1.svc.cluster.local.      303    IN    SRV 10 100 443 svc-1-a.test-1.svc.cluster.local."),
 | |
| 			test.SRV("_http._tcp.svc-1-b.test-1.svc.cluster.local.      303    IN    SRV 10 100 80 svc-1-b.test-1.svc.cluster.local."),
 | |
| 			test.SRV("_c-port._udp.svc-c.test-1.svc.cluster.local.      303    IN    SRV 10 100 1234 svc-c.test-1.svc.cluster.local."),
 | |
| 		},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "123.0.3.10.in-addr.arpa.", Qtype: dns.TypePTR,
 | |
| 		Rcode:  dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "100.0.3.10.in-addr.arpa.", Qtype: dns.TypePTR,
 | |
| 		Rcode:  dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.PTR("100.0.3.10.in-addr.arpa.      303    IN      PTR       svc-1-a.test-1.svc.cluster.local."),
 | |
| 		},
 | |
| 	},
 | |
| 	{
 | |
| 		Qname: "115.0.3.10.in-addr.arpa.", Qtype: dns.TypePTR,
 | |
| 		Rcode:  dns.RcodeSuccess,
 | |
| 		Answer: []dns.RR{
 | |
| 			test.PTR("115.0.3.10.in-addr.arpa.      303    IN      PTR       svc-c.test-1.svc.cluster.local."),
 | |
| 		},
 | |
| 	},
 | |
| }
 | |
| 
 | |
| func createTestServer(t *testing.T, corefile string) (*caddy.Instance, string) {
 | |
| 	server, err := CoreDNSServer(corefile)
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("Could not get CoreDNS serving instance: %s", err)
 | |
| 	}
 | |
| 
 | |
| 	udp, _ := CoreDNSServerPorts(server, 0)
 | |
| 	if udp == "" {
 | |
| 		t.Fatalf("Could not get UDP listening port")
 | |
| 	}
 | |
| 
 | |
| 	return server, udp
 | |
| }
 | |
| 
 | |
| func TestKubernetesIntegration(t *testing.T) {
 | |
| 	corefile :=
 | |
| 		`.:0 {
 | |
|     kubernetes cluster.local 0.3.10.in-addr.arpa {
 | |
|                 endpoint http://localhost:8080
 | |
| 		#endpoint https://kubernetes/ 
 | |
| 		#tls admin.pem admin-key.pem ca.pem
 | |
| 		#tls k8s_auth/client2.crt k8s_auth/client2.key k8s_auth/ca2.crt
 | |
| 		namespaces test-1
 | |
|     }
 | |
| `
 | |
| 	server, udp := createTestServer(t, corefile)
 | |
| 	defer server.Stop()
 | |
| 
 | |
| 	// Work-around for timing condition that results in no-data being returned in
 | |
| 	// test environment.
 | |
| 	time.Sleep(5 * time.Second)
 | |
| 
 | |
| 	for _, tc := range dnsTestCases {
 | |
| 		dnsClient := new(dns.Client)
 | |
| 		dnsMessage := new(dns.Msg)
 | |
| 
 | |
| 		dnsMessage.SetQuestion(tc.Qname, tc.Qtype)
 | |
| 
 | |
| 		res, _, err := dnsClient.Exchange(dnsMessage, udp)
 | |
| 		if err != nil {
 | |
| 			t.Fatalf("Could not send query: %s", err)
 | |
| 		}
 | |
| 
 | |
| 		// check the answer
 | |
| 		if res.Rcode != tc.Rcode {
 | |
| 			t.Errorf("Expected rcode %d but got %d for query %s, %d", tc.Rcode, res.Rcode, tc.Qname, tc.Qtype)
 | |
| 		}
 | |
| 
 | |
| 		if len(res.Answer) != len(tc.Answer) {
 | |
| 			t.Errorf("Expected %d answers but got %d for query %s, %d", len(tc.Answer), len(res.Answer), tc.Qname, tc.Qtype)
 | |
| 		}
 | |
| 
 | |
| 		//TODO: Check the actual RR values
 | |
| 	}
 | |
| }
 |