diff options
Diffstat (limited to 'middleware/etcd/multi_test.go')
| -rw-r--r-- | middleware/etcd/multi_test.go | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/middleware/etcd/multi_test.go b/middleware/etcd/multi_test.go index f4b59f50b..d4e3493a7 100644 --- a/middleware/etcd/multi_test.go +++ b/middleware/etcd/multi_test.go @@ -6,8 +6,8 @@ import ( "sort" "testing" - "github.com/miekg/coredns/middleware" "github.com/miekg/coredns/middleware/etcd/msg" + "github.com/miekg/coredns/middleware/pkg/dnsrecorder" "github.com/miekg/coredns/middleware/test" "github.com/miekg/dns" @@ -25,14 +25,14 @@ func TestMultiLookup(t *testing.T) { for _, tc := range dnsTestCasesMulti { m := tc.Msg() - rec := middleware.NewResponseRecorder(&test.ResponseWriter{}) + rec := dnsrecorder.New(&test.ResponseWriter{}) _, err := etc.ServeDNS(ctxt, rec, m) if err != nil { t.Errorf("expected no error, got %v\n", err) return } - resp := rec.Msg() + resp := rec.Msg sort.Sort(test.RRSet(resp.Answer)) sort.Sort(test.RRSet(resp.Ns)) sort.Sort(test.RRSet(resp.Extra)) |