| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | package cache
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import (
 | 
					
						
							| 
									
										
										
										
											2016-10-28 07:50:16 +01:00
										 |  |  | 	"io/ioutil"
 | 
					
						
							|  |  |  | 	"log"
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | 	"testing"
 | 
					
						
							|  |  |  | 	"time"
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2017-02-21 22:51:47 -08:00
										 |  |  | 	"github.com/coredns/coredns/middleware"
 | 
					
						
							|  |  |  | 	"github.com/coredns/coredns/middleware/pkg/response"
 | 
					
						
							|  |  |  | 	"github.com/coredns/coredns/middleware/test"
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2016-10-02 08:31:44 +01:00
										 |  |  | 	lru "github.com/hashicorp/golang-lru"
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | 	"github.com/miekg/dns"
 | 
					
						
							|  |  |  | )
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | type cacheTestCase struct {
 | 
					
						
							|  |  |  | 	test.Case
 | 
					
						
							|  |  |  | 	in                 test.Case
 | 
					
						
							|  |  |  | 	AuthenticatedData  bool
 | 
					
						
							|  |  |  | 	Authoritative      bool
 | 
					
						
							|  |  |  | 	RecursionAvailable bool
 | 
					
						
							|  |  |  | 	Truncated          bool
 | 
					
						
							|  |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | var cacheTestCases = []cacheTestCase{
 | 
					
						
							|  |  |  | 	{
 | 
					
						
							|  |  |  | 		RecursionAvailable: true, AuthenticatedData: true, Authoritative: true,
 | 
					
						
							|  |  |  | 		Case: test.Case{
 | 
					
						
							|  |  |  | 			Qname: "miek.nl.", Qtype: dns.TypeMX,
 | 
					
						
							|  |  |  | 			Answer: []dns.RR{
 | 
					
						
							| 
									
										
										
										
											2016-10-02 08:31:44 +01:00
										 |  |  | 				test.MX("miek.nl.	3600	IN	MX	1 aspmx.l.google.com."),
 | 
					
						
							|  |  |  | 				test.MX("miek.nl.	3600	IN	MX	10 aspmx2.googlemail.com."),
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | 			},
 | 
					
						
							|  |  |  | 		},
 | 
					
						
							|  |  |  | 		in: test.Case{
 | 
					
						
							|  |  |  | 			Qname: "miek.nl.", Qtype: dns.TypeMX,
 | 
					
						
							|  |  |  | 			Answer: []dns.RR{
 | 
					
						
							| 
									
										
										
										
											2016-10-02 08:31:44 +01:00
										 |  |  | 				test.MX("miek.nl.	3601	IN	MX	1 aspmx.l.google.com."),
 | 
					
						
							|  |  |  | 				test.MX("miek.nl.	3601	IN	MX	10 aspmx2.googlemail.com."),
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | 			},
 | 
					
						
							|  |  |  | 		},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-10-08 15:22:31 +01:00
										 |  |  | 	{
 | 
					
						
							|  |  |  | 		RecursionAvailable: true, AuthenticatedData: true, Authoritative: true,
 | 
					
						
							|  |  |  | 		Case: test.Case{
 | 
					
						
							|  |  |  | 			Qname: "mIEK.nL.", Qtype: dns.TypeMX,
 | 
					
						
							|  |  |  | 			Answer: []dns.RR{
 | 
					
						
							|  |  |  | 				test.MX("mIEK.nL.	3600	IN	MX	1 aspmx.l.google.com."),
 | 
					
						
							|  |  |  | 				test.MX("mIEK.nL.	3600	IN	MX	10 aspmx2.googlemail.com."),
 | 
					
						
							|  |  |  | 			},
 | 
					
						
							|  |  |  | 		},
 | 
					
						
							|  |  |  | 		in: test.Case{
 | 
					
						
							|  |  |  | 			Qname: "mIEK.nL.", Qtype: dns.TypeMX,
 | 
					
						
							|  |  |  | 			Answer: []dns.RR{
 | 
					
						
							|  |  |  | 				test.MX("mIEK.nL.	3601	IN	MX	1 aspmx.l.google.com."),
 | 
					
						
							|  |  |  | 				test.MX("mIEK.nL.	3601	IN	MX	10 aspmx2.googlemail.com."),
 | 
					
						
							|  |  |  | 			},
 | 
					
						
							|  |  |  | 		},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | 	{
 | 
					
						
							|  |  |  | 		Truncated: true,
 | 
					
						
							|  |  |  | 		Case: test.Case{
 | 
					
						
							|  |  |  | 			Qname: "miek.nl.", Qtype: dns.TypeMX,
 | 
					
						
							|  |  |  | 			Answer: []dns.RR{test.MX("miek.nl.	1800	IN	MX	1 aspmx.l.google.com.")},
 | 
					
						
							|  |  |  | 		},
 | 
					
						
							|  |  |  | 		in: test.Case{},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-10-28 07:50:16 +01:00
										 |  |  | 	{
 | 
					
						
							|  |  |  | 		RecursionAvailable: true, Authoritative: true,
 | 
					
						
							|  |  |  | 		Case: test.Case{
 | 
					
						
							|  |  |  | 			Rcode: dns.RcodeNameError,
 | 
					
						
							|  |  |  | 			Qname: "example.org.", Qtype: dns.TypeA,
 | 
					
						
							|  |  |  | 			Ns: []dns.RR{
 | 
					
						
							|  |  |  | 				test.SOA("example.org. 3600 IN	SOA	sns.dns.icann.org. noc.dns.icann.org. 2016082540 7200 3600 1209600 3600"),
 | 
					
						
							|  |  |  | 			},
 | 
					
						
							|  |  |  | 		},
 | 
					
						
							|  |  |  | 		in: test.Case{
 | 
					
						
							|  |  |  | 			Rcode: dns.RcodeNameError,
 | 
					
						
							|  |  |  | 			Qname: "example.org.", Qtype: dns.TypeA,
 | 
					
						
							|  |  |  | 			Ns: []dns.RR{
 | 
					
						
							|  |  |  | 				test.SOA("example.org. 3600 IN	SOA	sns.dns.icann.org. noc.dns.icann.org. 2016082540 7200 3600 1209600 3600"),
 | 
					
						
							|  |  |  | 			},
 | 
					
						
							|  |  |  | 		},
 | 
					
						
							|  |  |  | 	},
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func cacheMsg(m *dns.Msg, tc cacheTestCase) *dns.Msg {
 | 
					
						
							|  |  |  | 	m.RecursionAvailable = tc.RecursionAvailable
 | 
					
						
							|  |  |  | 	m.AuthenticatedData = tc.AuthenticatedData
 | 
					
						
							|  |  |  | 	m.Authoritative = tc.Authoritative
 | 
					
						
							| 
									
										
										
										
											2016-10-28 07:50:16 +01:00
										 |  |  | 	m.Rcode = tc.Rcode
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | 	m.Truncated = tc.Truncated
 | 
					
						
							|  |  |  | 	m.Answer = tc.in.Answer
 | 
					
						
							|  |  |  | 	m.Ns = tc.in.Ns
 | 
					
						
							|  |  |  | 	//	m.Extra = tc.in.Extra , not the OPT record!
 | 
					
						
							|  |  |  | 	return m
 | 
					
						
							|  |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2016-10-02 08:31:44 +01:00
										 |  |  | func newTestCache(ttl time.Duration) (*Cache, *ResponseWriter) {
 | 
					
						
							|  |  |  | 	c := &Cache{Zones: []string{"."}, pcap: defaultCap, ncap: defaultCap, pttl: ttl, nttl: ttl}
 | 
					
						
							|  |  |  | 	c.pcache, _ = lru.New(c.pcap)
 | 
					
						
							|  |  |  | 	c.ncache, _ = lru.New(c.ncap)
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	crr := &ResponseWriter{nil, c}
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | 	return c, crr
 | 
					
						
							|  |  |  | }
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func TestCache(t *testing.T) {
 | 
					
						
							| 
									
										
										
										
											2016-10-02 08:31:44 +01:00
										 |  |  | 	c, crr := newTestCache(maxTTL)
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2016-10-28 07:50:16 +01:00
										 |  |  | 	log.SetOutput(ioutil.Discard)
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | 	for _, tc := range cacheTestCases {
 | 
					
						
							|  |  |  | 		m := tc.in.Msg()
 | 
					
						
							|  |  |  | 		m = cacheMsg(m, tc)
 | 
					
						
							|  |  |  | 		do := tc.in.Do
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2016-10-10 12:09:29 +01:00
										 |  |  | 		mt, _ := response.Typify(m)
 | 
					
						
							| 
									
										
										
										
											2016-10-02 08:31:44 +01:00
										 |  |  | 		k := key(m, mt, do)
 | 
					
						
							|  |  |  | 		crr.set(m, k, mt, c.pttl)
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 		name := middleware.Name(m.Question[0].Name).Normalize()
 | 
					
						
							|  |  |  | 		qtype := m.Question[0].Qtype
 | 
					
						
							| 
									
										
										
										
											2016-10-02 08:31:44 +01:00
										 |  |  | 		i, ok, _ := c.get(name, qtype, do)
 | 
					
						
							|  |  |  | 		if ok && m.Truncated {
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | 			t.Errorf("Truncated message should not have been cached")
 | 
					
						
							| 
									
										
										
										
											2016-10-02 08:31:44 +01:00
										 |  |  | 			continue
 | 
					
						
							| 
									
										
										
										
											2016-04-19 11:13:24 +01:00
										 |  |  | 		}
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		if ok {
 | 
					
						
							|  |  |  | 			resp := i.toMsg(m)
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 			if !test.Header(t, tc.Case, resp) {
 | 
					
						
							|  |  |  | 				t.Logf("%v\n", resp)
 | 
					
						
							|  |  |  | 				continue
 | 
					
						
							|  |  |  | 			}
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 			if !test.Section(t, tc.Case, test.Answer, resp.Answer) {
 | 
					
						
							|  |  |  | 				t.Logf("%v\n", resp)
 | 
					
						
							|  |  |  | 			}
 | 
					
						
							|  |  |  | 			if !test.Section(t, tc.Case, test.Ns, resp.Ns) {
 | 
					
						
							|  |  |  | 				t.Logf("%v\n", resp)
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 			}
 | 
					
						
							|  |  |  | 			if !test.Section(t, tc.Case, test.Extra, resp.Extra) {
 | 
					
						
							|  |  |  | 				t.Logf("%v\n", resp)
 | 
					
						
							|  |  |  | 			}
 | 
					
						
							|  |  |  | 		}
 | 
					
						
							|  |  |  | 	}
 | 
					
						
							|  |  |  | }
 |