diff options
Diffstat (limited to 'plugin/erratic/erratic_test.go')
| -rw-r--r-- | plugin/erratic/erratic_test.go | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/plugin/erratic/erratic_test.go b/plugin/erratic/erratic_test.go index 7a1a420da..6927c5248 100644 --- a/plugin/erratic/erratic_test.go +++ b/plugin/erratic/erratic_test.go @@ -3,7 +3,7 @@ package erratic import ( "testing" - "github.com/coredns/coredns/plugin/pkg/dnsrecorder" + "github.com/coredns/coredns/plugin/pkg/dnstest" "github.com/coredns/coredns/plugin/test" "github.com/miekg/dns" @@ -28,7 +28,7 @@ func TestErraticDrop(t *testing.T) { req := new(dns.Msg) req.SetQuestion("example.org.", dns.TypeA) - rec := dnsrecorder.New(&test.ResponseWriter{}) + rec := dnstest.NewRecorder(&test.ResponseWriter{}) code, err := e.ServeDNS(ctx, rec, req) if err != tc.expectedErr { @@ -62,7 +62,7 @@ func TestErraticTruncate(t *testing.T) { req := new(dns.Msg) req.SetQuestion("example.org.", dns.TypeA) - rec := dnsrecorder.New(&test.ResponseWriter{}) + rec := dnstest.NewRecorder(&test.ResponseWriter{}) code, err := e.ServeDNS(ctx, rec, req) if err != tc.expectedErr { |