diff options
Diffstat (limited to 'middleware/whoami/whoami_test.go')
-rw-r--r-- | middleware/whoami/whoami_test.go | 56 |
1 files changed, 0 insertions, 56 deletions
diff --git a/middleware/whoami/whoami_test.go b/middleware/whoami/whoami_test.go deleted file mode 100644 index d9357e497..000000000 --- a/middleware/whoami/whoami_test.go +++ /dev/null @@ -1,56 +0,0 @@ -package whoami - -import ( - "testing" - - "github.com/coredns/coredns/middleware/pkg/dnsrecorder" - "github.com/coredns/coredns/middleware/test" - - "github.com/miekg/dns" - "golang.org/x/net/context" -) - -func TestWhoami(t *testing.T) { - wh := Whoami{} - - tests := []struct { - qname string - qtype uint16 - expectedCode int - expectedReply []string // ownernames for the records in the additional section. - expectedErr error - }{ - { - qname: "example.org", - qtype: dns.TypeA, - expectedCode: dns.RcodeSuccess, - expectedReply: []string{"example.org.", "_udp.example.org."}, - expectedErr: nil, - }, - } - - ctx := context.TODO() - - for i, tc := range tests { - req := new(dns.Msg) - req.SetQuestion(dns.Fqdn(tc.qname), tc.qtype) - - rec := dnsrecorder.New(&test.ResponseWriter{}) - code, err := wh.ServeDNS(ctx, rec, req) - - if err != tc.expectedErr { - t.Errorf("Test %d: Expected error %v, but got %v", i, tc.expectedErr, err) - } - if code != int(tc.expectedCode) { - t.Errorf("Test %d: Expected status code %d, but got %d", i, tc.expectedCode, code) - } - if len(tc.expectedReply) != 0 { - for i, expected := range tc.expectedReply { - actual := rec.Msg.Extra[i].Header().Name - if actual != expected { - t.Errorf("Test %d: Expected answer %s, but got %s", i, expected, actual) - } - } - } - } -} |