diff options
Diffstat (limited to 'plugin/forward/proxy_test.go')
| -rw-r--r-- | plugin/forward/proxy_test.go | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/plugin/forward/proxy_test.go b/plugin/forward/proxy_test.go index d7af25aa0..794103516 100644 --- a/plugin/forward/proxy_test.go +++ b/plugin/forward/proxy_test.go @@ -5,6 +5,7 @@ import ( "testing" "github.com/coredns/coredns/plugin/pkg/dnstest" + "github.com/coredns/coredns/plugin/pkg/transport" "github.com/coredns/coredns/plugin/test" "github.com/coredns/coredns/request" @@ -26,7 +27,7 @@ func TestProxyClose(t *testing.T) { ctx := context.TODO() for i := 0; i < 100; i++ { - p := NewProxy(s.Addr, DNS) + p := NewProxy(s.Addr, transport.DNS) p.start(hcInterval) go func() { p.Connect(ctx, state, options{}) }() @@ -95,7 +96,7 @@ func TestProxyTLSFail(t *testing.T) { } func TestProtocolSelection(t *testing.T) { - p := NewProxy("bad_address", DNS) + p := NewProxy("bad_address", transport.DNS) stateUDP := request.Request{W: &test.ResponseWriter{}, Req: new(dns.Msg)} stateTCP := request.Request{W: &test.ResponseWriter{TCP: true}, Req: new(dns.Msg)} |