tracker: Fix network error handling in unit test

This commit is contained in:
Matt Joiner 2019-01-08 19:20:53 +11:00
parent 885af9d29d
commit 0047eefef4
1 changed files with 2 additions and 1 deletions

View File

@ -15,6 +15,7 @@ import (
"github.com/anacrolix/dht/krpc"
_ "github.com/anacrolix/envpprof"
"github.com/pkg/errors"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
@ -141,7 +142,7 @@ func TestUDPTracker(t *testing.T) {
Request: req,
}.Do()
// Skip any net errors as we don't control the server.
if _, ok := err.(net.Error); ok {
if _, ok := errors.Cause(err).(net.Error); ok {
t.Skip(err)
}
require.NoError(t, err)