aboutsummaryrefslogtreecommitdiff
path: root/plugin/cache/do_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'plugin/cache/do_test.go')
-rw-r--r--plugin/cache/do_test.go75
1 files changed, 0 insertions, 75 deletions
diff --git a/plugin/cache/do_test.go b/plugin/cache/do_test.go
deleted file mode 100644
index 3cf87cabe..000000000
--- a/plugin/cache/do_test.go
+++ /dev/null
@@ -1,75 +0,0 @@
-package cache
-
-import (
- "context"
- "testing"
-
- "github.com/coredns/coredns/plugin"
- "github.com/coredns/coredns/plugin/pkg/dnstest"
- "github.com/coredns/coredns/plugin/test"
-
- "github.com/miekg/dns"
-)
-
-func TestDo(t *testing.T) {
- // cache sets Do and requests that don't have them.
- c := New()
- c.Next = echoHandler()
- req := new(dns.Msg)
- req.SetQuestion("example.org.", dns.TypeA)
- rec := dnstest.NewRecorder(&test.ResponseWriter{})
-
- // No DO set.
- c.ServeDNS(context.TODO(), rec, req)
- reply := rec.Msg
- opt := reply.Extra[len(reply.Extra)-1]
- if x, ok := opt.(*dns.OPT); !ok {
- t.Fatalf("Expected OPT RR, got %T", x)
- }
- if !opt.(*dns.OPT).Do() {
- t.Errorf("Expected DO bit to be set, got false")
- }
- if x := opt.(*dns.OPT).UDPSize(); x != defaultUDPBufSize {
- t.Errorf("Expected %d bufsize, got %d", defaultUDPBufSize, x)
- }
-
- // Do set - so left alone.
- const mysize = defaultUDPBufSize * 2
- setDo(req)
- // set bufsize to something else than default to see cache doesn't touch it
- req.Extra[len(req.Extra)-1].(*dns.OPT).SetUDPSize(mysize)
- c.ServeDNS(context.TODO(), rec, req)
- reply = rec.Msg
- opt = reply.Extra[len(reply.Extra)-1]
- if x, ok := opt.(*dns.OPT); !ok {
- t.Fatalf("Expected OPT RR, got %T", x)
- }
- if !opt.(*dns.OPT).Do() {
- t.Errorf("Expected DO bit to be set, got false")
- }
- if x := opt.(*dns.OPT).UDPSize(); x != mysize {
- t.Errorf("Expected %d bufsize, got %d", mysize, x)
- }
-
- // edns0 set, but not DO, so _not_ left alone.
- req.Extra[len(req.Extra)-1].(*dns.OPT).SetDo(false)
- c.ServeDNS(context.TODO(), rec, req)
- reply = rec.Msg
- opt = reply.Extra[len(reply.Extra)-1]
- if x, ok := opt.(*dns.OPT); !ok {
- t.Fatalf("Expected OPT RR, got %T", x)
- }
- if !opt.(*dns.OPT).Do() {
- t.Errorf("Expected DO bit to be set, got false")
- }
- if x := opt.(*dns.OPT).UDPSize(); x != defaultUDPBufSize {
- t.Errorf("Expected %d bufsize, got %d", defaultUDPBufSize, x)
- }
-}
-
-func echoHandler() plugin.Handler {
- return plugin.HandlerFunc(func(ctx context.Context, w dns.ResponseWriter, r *dns.Msg) (int, error) {
- w.WriteMsg(r)
- return dns.RcodeSuccess, nil
- })
-}