diff options
Diffstat (limited to 'middleware/chaos/chaos_test.go')
| -rw-r--r-- | middleware/chaos/chaos_test.go | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/middleware/chaos/chaos_test.go b/middleware/chaos/chaos_test.go index e8cdef5f1..fc7d4845f 100644 --- a/middleware/chaos/chaos_test.go +++ b/middleware/chaos/chaos_test.go @@ -4,6 +4,7 @@ import ( "testing" "github.com/miekg/coredns/middleware" + coretest "github.com/miekg/coredns/middleware/testing" "github.com/miekg/dns" "golang.org/x/net/context" @@ -57,7 +58,7 @@ func TestChaos(t *testing.T) { req.Question[0].Qclass = dns.ClassCHAOS em.Next = test.next - rec := middleware.NewResponseRecorder(&middleware.TestResponseWriter{}) + rec := middleware.NewResponseRecorder(&coretest.ResponseWriter{}) code, err := em.ServeDNS(ctx, rec, req) if err != test.expectedErr { |