| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | // +build etcd
 | 
					
						
							| 
									
										
										
										
											2016-03-23 10:46:33 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | package etcd
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import (
 | 
					
						
							| 
									
										
										
										
											2017-08-13 18:16:25 +01:00
										 |  |  | 	"context"
 | 
					
						
							|  |  |  | 	"encoding/json"
 | 
					
						
							|  |  |  | 	"testing"
 | 
					
						
							|  |  |  | 	"time"
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2017-02-21 22:51:47 -08:00
										 |  |  | 	"github.com/coredns/coredns/middleware/etcd/msg"
 | 
					
						
							| 
									
										
										
										
											2017-08-13 18:16:25 +01:00
										 |  |  | 	"github.com/coredns/coredns/middleware/pkg/dnsrecorder"
 | 
					
						
							|  |  |  | 	"github.com/coredns/coredns/middleware/pkg/singleflight"
 | 
					
						
							|  |  |  | 	"github.com/coredns/coredns/middleware/pkg/tls"
 | 
					
						
							|  |  |  | 	"github.com/coredns/coredns/middleware/proxy"
 | 
					
						
							| 
									
										
										
										
											2017-02-21 22:51:47 -08:00
										 |  |  | 	"github.com/coredns/coredns/middleware/test"
 | 
					
						
							| 
									
										
										
										
											2016-03-28 10:49:28 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2017-08-13 18:16:25 +01:00
										 |  |  | 	etcdc "github.com/coreos/etcd/client"
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 	"github.com/miekg/dns"
 | 
					
						
							|  |  |  | )
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2017-08-13 18:16:25 +01:00
										 |  |  | func init() {
 | 
					
						
							|  |  |  | 	ctxt, _ = context.WithTimeout(context.Background(), etcdTimeout)
 | 
					
						
							|  |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | // Note the key is encoded as DNS name, while in "reality" it is a etcd path.
 | 
					
						
							|  |  |  | var services = []*msg.Service{
 | 
					
						
							| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | 	{Host: "dev.server1", Port: 8080, Key: "a.server1.dev.region1.skydns.test."},
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 	{Host: "10.0.0.1", Port: 8080, Key: "a.server1.prod.region1.skydns.test."},
 | 
					
						
							|  |  |  | 	{Host: "10.0.0.2", Port: 8080, Key: "b.server1.prod.region1.skydns.test."},
 | 
					
						
							|  |  |  | 	{Host: "::1", Port: 8080, Key: "b.server6.prod.region1.skydns.test."},
 | 
					
						
							| 
									
										
										
										
											2016-06-08 10:29:46 +01:00
										 |  |  | 	// Unresolvable internal name.
 | 
					
						
							| 
									
										
										
										
											2016-03-23 22:59:05 +00:00
										 |  |  | 	{Host: "unresolvable.skydns.test", Key: "cname.prod.region1.skydns.test."},
 | 
					
						
							| 
									
										
										
										
											2016-06-08 10:29:46 +01:00
										 |  |  | 	// Priority.
 | 
					
						
							| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | 	{Host: "priority.server1", Priority: 333, Port: 8080, Key: "priority.skydns.test."},
 | 
					
						
							| 
									
										
										
										
											2016-06-08 10:29:46 +01:00
										 |  |  | 	// Subdomain.
 | 
					
						
							| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | 	{Host: "sub.server1", Port: 0, Key: "a.sub.region1.skydns.test."},
 | 
					
						
							|  |  |  | 	{Host: "sub.server2", Port: 80, Key: "b.sub.region1.skydns.test."},
 | 
					
						
							| 
									
										
										
										
											2016-03-24 08:22:24 +00:00
										 |  |  | 	{Host: "10.0.0.1", Port: 8080, Key: "c.sub.region1.skydns.test."},
 | 
					
						
							| 
									
										
										
										
											2016-06-08 10:29:46 +01:00
										 |  |  | 	// Cname loop.
 | 
					
						
							| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | 	{Host: "a.cname.skydns.test", Key: "b.cname.skydns.test."},
 | 
					
						
							|  |  |  | 	{Host: "b.cname.skydns.test", Key: "a.cname.skydns.test."},
 | 
					
						
							| 
									
										
										
										
											2016-04-13 08:03:56 +01:00
										 |  |  | 	// Nameservers.
 | 
					
						
							|  |  |  | 	{Host: "10.0.0.2", Key: "a.ns.dns.skydns.test."},
 | 
					
						
							|  |  |  | 	{Host: "10.0.0.3", Key: "b.ns.dns.skydns.test."},
 | 
					
						
							| 
									
										
										
										
											2016-06-08 10:29:46 +01:00
										 |  |  | 	// Reverse.
 | 
					
						
							|  |  |  | 	{Host: "reverse.example.com", Key: "1.0.0.10.in-addr.arpa."}, // 10.0.0.1
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | var dnsTestCases = []test.Case{
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 	// SRV Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "a.server1.dev.region1.skydns.test.", Qtype: dns.TypeSRV,
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 		Answer: []dns.RR{test.SRV("a.server1.dev.region1.skydns.test. 300 SRV 10 100 8080 dev.server1.")},
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-03-25 10:32:12 +00:00
										 |  |  | 	// SRV Test (case test)
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "a.SERVer1.dEv.region1.skydns.tEst.", Qtype: dns.TypeSRV,
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 		Answer: []dns.RR{test.SRV("a.SERVer1.dEv.region1.skydns.tEst. 300 SRV 10 100 8080 dev.server1.")},
 | 
					
						
							| 
									
										
										
										
											2016-03-25 10:32:12 +00:00
										 |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-03-23 23:06:16 +00:00
										 |  |  | 	// NXDOMAIN Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "doesnotexist.skydns.test.", Qtype: dns.TypeA,
 | 
					
						
							|  |  |  | 		Rcode: dns.RcodeNameError,
 | 
					
						
							|  |  |  | 		Ns: []dns.RR{
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 			test.SOA("skydns.test. 300 SOA ns.dns.skydns.test. hostmaster.skydns.test. 0 0 0 0 0"),
 | 
					
						
							| 
									
										
										
										
											2016-03-23 23:06:16 +00:00
										 |  |  | 		},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 	// A Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "a.server1.prod.region1.skydns.test.", Qtype: dns.TypeA,
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 		Answer: []dns.RR{test.A("a.server1.prod.region1.skydns.test. 300 A 10.0.0.1")},
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 	},
 | 
					
						
							|  |  |  | 	// SRV Test where target is IP address
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "a.server1.prod.region1.skydns.test.", Qtype: dns.TypeSRV,
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 		Answer: []dns.RR{test.SRV("a.server1.prod.region1.skydns.test. 300 SRV 10 100 8080 a.server1.prod.region1.skydns.test.")},
 | 
					
						
							|  |  |  | 		Extra:  []dns.RR{test.A("a.server1.prod.region1.skydns.test. 300 A 10.0.0.1")},
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 	},
 | 
					
						
							|  |  |  | 	// AAAA Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "b.server6.prod.region1.skydns.test.", Qtype: dns.TypeAAAA,
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 		Answer: []dns.RR{test.AAAA("b.server6.prod.region1.skydns.test. 300 AAAA ::1")},
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 	},
 | 
					
						
							|  |  |  | 	// Multiple A Record Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "server1.prod.region1.skydns.test.", Qtype: dns.TypeA,
 | 
					
						
							|  |  |  | 		Answer: []dns.RR{
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 			test.A("server1.prod.region1.skydns.test. 300 A 10.0.0.1"),
 | 
					
						
							|  |  |  | 			test.A("server1.prod.region1.skydns.test. 300 A 10.0.0.2"),
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 		},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-03-24 08:22:24 +00:00
										 |  |  | 	// Priority Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "priority.skydns.test.", Qtype: dns.TypeSRV,
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 		Answer: []dns.RR{test.SRV("priority.skydns.test. 300 SRV 333 100 8080 priority.server1.")},
 | 
					
						
							| 
									
										
										
										
											2016-03-24 08:22:24 +00:00
										 |  |  | 	},
 | 
					
						
							|  |  |  | 	// Subdomain Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "sub.region1.skydns.test.", Qtype: dns.TypeSRV,
 | 
					
						
							|  |  |  | 		Answer: []dns.RR{
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 			test.SRV("sub.region1.skydns.test. 300 IN SRV 10 33 0 sub.server1."),
 | 
					
						
							|  |  |  | 			test.SRV("sub.region1.skydns.test. 300 IN SRV 10 33 80 sub.server2."),
 | 
					
						
							|  |  |  | 			test.SRV("sub.region1.skydns.test. 300 IN SRV 10 33 8080 c.sub.region1.skydns.test."),
 | 
					
						
							| 
									
										
										
										
											2016-03-24 08:22:24 +00:00
										 |  |  | 		},
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 		Extra: []dns.RR{test.A("c.sub.region1.skydns.test. 300 IN A 10.0.0.1")},
 | 
					
						
							| 
									
										
										
										
											2016-03-24 08:22:24 +00:00
										 |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-03-23 22:59:05 +00:00
										 |  |  | 	// CNAME (unresolvable internal name)
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "cname.prod.region1.skydns.test.", Qtype: dns.TypeA,
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 		Ns: []dns.RR{test.SOA("skydns.test. 300 SOA ns.dns.skydns.test. hostmaster.skydns.test. 0 0 0 0 0")},
 | 
					
						
							| 
									
										
										
										
											2016-03-23 22:59:05 +00:00
										 |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | 	// Wildcard Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "*.region1.skydns.test.", Qtype: dns.TypeSRV,
 | 
					
						
							|  |  |  | 		Answer: []dns.RR{
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 			test.SRV("*.region1.skydns.test.	300	IN	SRV	10 12 0 sub.server1."),
 | 
					
						
							|  |  |  | 			test.SRV("*.region1.skydns.test.	300	IN	SRV	10 12 0 unresolvable.skydns.test."),
 | 
					
						
							|  |  |  | 			test.SRV("*.region1.skydns.test.	300	IN	SRV	10 12 80 sub.server2."),
 | 
					
						
							|  |  |  | 			test.SRV("*.region1.skydns.test.	300	IN	SRV	10 12 8080 a.server1.prod.region1.skydns.test."),
 | 
					
						
							|  |  |  | 			test.SRV("*.region1.skydns.test.	300	IN	SRV	10 12 8080 b.server1.prod.region1.skydns.test."),
 | 
					
						
							|  |  |  | 			test.SRV("*.region1.skydns.test.	300	IN	SRV	10 12 8080 b.server6.prod.region1.skydns.test."),
 | 
					
						
							|  |  |  | 			test.SRV("*.region1.skydns.test.	300	IN	SRV	10 12 8080 c.sub.region1.skydns.test."),
 | 
					
						
							|  |  |  | 			test.SRV("*.region1.skydns.test.	300	IN	SRV	10 12 8080 dev.server1."),
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 		},
 | 
					
						
							| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | 		Extra: []dns.RR{
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 			test.A("a.server1.prod.region1.skydns.test.	300	IN	A	10.0.0.1"),
 | 
					
						
							|  |  |  | 			test.A("b.server1.prod.region1.skydns.test.	300	IN	A	10.0.0.2"),
 | 
					
						
							|  |  |  | 			test.AAAA("b.server6.prod.region1.skydns.test.	300	IN	AAAA	::1"),
 | 
					
						
							|  |  |  | 			test.A("c.sub.region1.skydns.test.	300	IN	A	10.0.0.1"),
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 		},
 | 
					
						
							| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | 	},
 | 
					
						
							|  |  |  | 	// Wildcard Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "prod.*.skydns.test.", Qtype: dns.TypeSRV,
 | 
					
						
							|  |  |  | 		Answer: []dns.RR{
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 			test.SRV("prod.*.skydns.test.	300	IN	SRV	10 25 0 unresolvable.skydns.test."),
 | 
					
						
							|  |  |  | 			test.SRV("prod.*.skydns.test.	300	IN	SRV	10 25 8080 a.server1.prod.region1.skydns.test."),
 | 
					
						
							|  |  |  | 			test.SRV("prod.*.skydns.test.	300	IN	SRV	10 25 8080 b.server1.prod.region1.skydns.test."),
 | 
					
						
							|  |  |  | 			test.SRV("prod.*.skydns.test.	300	IN	SRV	10 25 8080 b.server6.prod.region1.skydns.test."),
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 		},
 | 
					
						
							| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | 		Extra: []dns.RR{
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 			test.A("a.server1.prod.region1.skydns.test.	300	IN	A	10.0.0.1"),
 | 
					
						
							|  |  |  | 			test.A("b.server1.prod.region1.skydns.test.	300	IN	A	10.0.0.2"),
 | 
					
						
							|  |  |  | 			test.AAAA("b.server6.prod.region1.skydns.test.	300	IN	AAAA	::1"),
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 		},
 | 
					
						
							| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | 	},
 | 
					
						
							|  |  |  | 	// Wildcard Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "prod.any.skydns.test.", Qtype: dns.TypeSRV,
 | 
					
						
							|  |  |  | 		Answer: []dns.RR{
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 			test.SRV("prod.any.skydns.test.	300	IN	SRV	10 25 0 unresolvable.skydns.test."),
 | 
					
						
							|  |  |  | 			test.SRV("prod.any.skydns.test.	300	IN	SRV	10 25 8080 a.server1.prod.region1.skydns.test."),
 | 
					
						
							|  |  |  | 			test.SRV("prod.any.skydns.test.	300	IN	SRV	10 25 8080 b.server1.prod.region1.skydns.test."),
 | 
					
						
							|  |  |  | 			test.SRV("prod.any.skydns.test.	300	IN	SRV	10 25 8080 b.server6.prod.region1.skydns.test."),
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 		},
 | 
					
						
							| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | 		Extra: []dns.RR{
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 			test.A("a.server1.prod.region1.skydns.test.	300	IN	A	10.0.0.1"),
 | 
					
						
							|  |  |  | 			test.A("b.server1.prod.region1.skydns.test.	300	IN	A	10.0.0.2"),
 | 
					
						
							|  |  |  | 			test.AAAA("b.server6.prod.region1.skydns.test.	300	IN	AAAA	::1"),
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | 		},
 | 
					
						
							| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | 	},
 | 
					
						
							|  |  |  | 	// CNAME loop detection
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "a.cname.skydns.test.", Qtype: dns.TypeA,
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 		Ns: []dns.RR{test.SOA("skydns.test. 300 SOA ns.dns.skydns.test. hostmaster.skydns.test. 1407441600 28800 7200 604800 60")},
 | 
					
						
							| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | 	},
 | 
					
						
							|  |  |  | 	// NODATA Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "a.server1.dev.region1.skydns.test.", Qtype: dns.TypeTXT,
 | 
					
						
							| 
									
										
										
										
											2016-04-11 10:22:27 +01:00
										 |  |  | 		Ns: []dns.RR{test.SOA("skydns.test. 300 SOA ns.dns.skydns.test. hostmaster.skydns.test. 0 0 0 0 0")},
 | 
					
						
							| 
									
										
										
										
											2016-03-24 17:31:01 +00:00
										 |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-04-12 23:26:46 +01:00
										 |  |  | 	// NODATA Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "a.server1.dev.region1.skydns.test.", Qtype: dns.TypeHINFO,
 | 
					
						
							|  |  |  | 		Ns: []dns.RR{test.SOA("skydns.test. 300 SOA ns.dns.skydns.test. hostmaster.skydns.test. 0 0 0 0 0")},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							|  |  |  | 	// NXDOMAIN Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "a.server1.nonexistent.region1.skydns.test.", Qtype: dns.TypeHINFO, Rcode: dns.RcodeNameError,
 | 
					
						
							|  |  |  | 		Ns: []dns.RR{test.SOA("skydns.test. 300 SOA ns.dns.skydns.test. hostmaster.skydns.test. 0 0 0 0 0")},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "skydns.test.", Qtype: dns.TypeSOA,
 | 
					
						
							|  |  |  | 		Answer: []dns.RR{test.SOA("skydns.test.	300	IN	SOA	ns.dns.skydns.test. hostmaster.skydns.test. 1460498836 14400 3600 604800 60")},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-04-13 08:03:56 +01:00
										 |  |  | 	// NS Record Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "skydns.test.", Qtype: dns.TypeNS,
 | 
					
						
							|  |  |  | 		Answer: []dns.RR{
 | 
					
						
							|  |  |  | 			test.NS("skydns.test. 300 NS a.ns.dns.skydns.test."),
 | 
					
						
							|  |  |  | 			test.NS("skydns.test. 300 NS b.ns.dns.skydns.test."),
 | 
					
						
							|  |  |  | 		},
 | 
					
						
							|  |  |  | 		Extra: []dns.RR{
 | 
					
						
							|  |  |  | 			test.A("a.ns.dns.skydns.test. 300 A 10.0.0.2"),
 | 
					
						
							|  |  |  | 			test.A("b.ns.dns.skydns.test. 300 A 10.0.0.3"),
 | 
					
						
							|  |  |  | 		},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							|  |  |  | 	// NS Record Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "a.skydns.test.", Qtype: dns.TypeNS, Rcode: dns.RcodeNameError,
 | 
					
						
							|  |  |  | 		Ns: []dns.RR{test.SOA("skydns.test.	300	IN	SOA	ns.dns.skydns.test. hostmaster.skydns.test. 1460498836 14400 3600 604800 60")},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							|  |  |  | 	// A Record For NS Record Test
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "ns.dns.skydns.test.", Qtype: dns.TypeA,
 | 
					
						
							|  |  |  | 		Answer: []dns.RR{
 | 
					
						
							|  |  |  | 			test.A("ns.dns.skydns.test. 300 A 10.0.0.2"),
 | 
					
						
							|  |  |  | 			test.A("ns.dns.skydns.test. 300 A 10.0.0.3"),
 | 
					
						
							|  |  |  | 		},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-04-12 23:26:46 +01:00
										 |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "skydns_extra.test.", Qtype: dns.TypeSOA,
 | 
					
						
							|  |  |  | 		Answer: []dns.RR{test.SOA("skydns_extra.test. 300 IN SOA ns.dns.skydns_extra.test. hostmaster.skydns_extra.test. 1460498836 14400 3600 604800 60")},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-06-08 10:29:46 +01:00
										 |  |  | 	// Reverse lookup
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		Qname: "1.0.0.10.in-addr.arpa.", Qtype: dns.TypePTR,
 | 
					
						
							|  |  |  | 		Answer: []dns.RR{test.PTR("1.0.0.10.in-addr.arpa. 300 PTR reverse.example.com.")},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-03-20 17:44:58 +00:00
										 |  |  | }
 | 
					
						
							| 
									
										
										
										
											2017-08-13 18:16:25 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | func newEtcdMiddleware() *Etcd {
 | 
					
						
							|  |  |  | 	ctxt, _ = context.WithTimeout(context.Background(), etcdTimeout)
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	endpoints := []string{"http://localhost:2379"}
 | 
					
						
							|  |  |  | 	tlsc, _ := tls.NewTLSConfigFromArgs()
 | 
					
						
							|  |  |  | 	client, _ := newEtcdClient(endpoints, tlsc)
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	return &Etcd{
 | 
					
						
							|  |  |  | 		Proxy:      proxy.NewLookup([]string{"8.8.8.8:53"}),
 | 
					
						
							|  |  |  | 		PathPrefix: "skydns",
 | 
					
						
							|  |  |  | 		Ctx:        context.Background(),
 | 
					
						
							|  |  |  | 		Inflight:   &singleflight.Group{},
 | 
					
						
							|  |  |  | 		Zones:      []string{"skydns.test.", "skydns_extra.test.", "in-addr.arpa."},
 | 
					
						
							|  |  |  | 		Client:     client,
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func set(t *testing.T, e *Etcd, k string, ttl time.Duration, m *msg.Service) {
 | 
					
						
							|  |  |  | 	b, err := json.Marshal(m)
 | 
					
						
							|  |  |  | 	if err != nil {
 | 
					
						
							|  |  |  | 		t.Fatal(err)
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | 	path, _ := msg.PathWithWildcard(k, e.PathPrefix)
 | 
					
						
							|  |  |  | 	e.Client.Set(ctxt, path, string(b), &etcdc.SetOptions{TTL: ttl})
 | 
					
						
							|  |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func delete(t *testing.T, e *Etcd, k string) {
 | 
					
						
							|  |  |  | 	path, _ := msg.PathWithWildcard(k, e.PathPrefix)
 | 
					
						
							|  |  |  | 	e.Client.Delete(ctxt, path, &etcdc.DeleteOptions{Recursive: false})
 | 
					
						
							|  |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func TestLookup(t *testing.T) {
 | 
					
						
							|  |  |  | 	etc := newEtcdMiddleware()
 | 
					
						
							|  |  |  | 	for _, serv := range services {
 | 
					
						
							|  |  |  | 		set(t, etc, serv.Key, 0, serv)
 | 
					
						
							|  |  |  | 		defer delete(t, etc, serv.Key)
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	for _, tc := range dnsTestCases {
 | 
					
						
							|  |  |  | 		m := tc.Msg()
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		rec := dnsrecorder.New(&test.ResponseWriter{})
 | 
					
						
							|  |  |  | 		etc.ServeDNS(ctxt, rec, m)
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		resp := rec.Msg
 | 
					
						
							| 
									
										
										
										
											2017-08-16 15:30:58 +01:00
										 |  |  | 		test.SortAndCheck(t, resp, tc)
 | 
					
						
							| 
									
										
										
										
											2017-08-13 18:16:25 +01:00
										 |  |  | 	}
 | 
					
						
							|  |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | var ctxt context.Context
 |