diff options
Diffstat (limited to 'middleware/chaos/chaos_test.go')
| -rw-r--r-- | middleware/chaos/chaos_test.go | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/middleware/chaos/chaos_test.go b/middleware/chaos/chaos_test.go index f669eaa7c..6a3261754 100644 --- a/middleware/chaos/chaos_test.go +++ b/middleware/chaos/chaos_test.go @@ -4,6 +4,7 @@ import ( "testing" "github.com/miekg/coredns/middleware" + "github.com/miekg/coredns/middleware/pkg/dnsrecorder" "github.com/miekg/coredns/middleware/test" "github.com/miekg/dns" @@ -58,7 +59,7 @@ func TestChaos(t *testing.T) { req.Question[0].Qclass = dns.ClassCHAOS em.Next = tc.next - rec := middleware.NewResponseRecorder(&test.ResponseWriter{}) + rec := dnsrecorder.New(&test.ResponseWriter{}) code, err := em.ServeDNS(ctx, rec, req) if err != tc.expectedErr { @@ -68,7 +69,7 @@ func TestChaos(t *testing.T) { t.Errorf("Test %d: Expected status code %d, but got %d", i, tc.expectedCode, code) } if tc.expectedReply != "" { - answer := rec.Msg().Answer[0].(*dns.TXT).Txt[0] + answer := rec.Msg.Answer[0].(*dns.TXT).Txt[0] if answer != tc.expectedReply { t.Errorf("Test %d: Expected answer %s, but got %s", i, tc.expectedReply, answer) } |