diff --git a/p2p/protocol/ping/ping_test.go b/p2p/protocol/ping/ping_test.go index 75021b1406..51e7f18f0e 100644 --- a/p2p/protocol/ping/ping_test.go +++ b/p2p/protocol/ping/ping_test.go @@ -5,13 +5,14 @@ import ( "testing" "time" - "github.com/stretchr/testify/require" - bhost "github.com/libp2p/go-libp2p/p2p/host/basic" swarmt "github.com/libp2p/go-libp2p/p2p/net/swarm/testing" "github.com/libp2p/go-libp2p/p2p/protocol/ping" "github.com/libp2p/go-libp2p-core/peer" + + ma "github.com/multiformats/go-multiaddr" + "github.com/stretchr/testify/require" ) func TestPing(t *testing.T) { @@ -19,12 +20,14 @@ func TestPing(t *testing.T) { defer cancel() h1, err := bhost.NewHost(swarmt.GenSwarm(t), nil) require.NoError(t, err) + defer h1.Close() h2, err := bhost.NewHost(swarmt.GenSwarm(t), nil) require.NoError(t, err) + defer h2.Close() err = h1.Connect(ctx, peer.AddrInfo{ ID: h2.ID(), - Addrs: h2.Addrs(), + Addrs: []ma.Multiaddr{h2.Addrs()[0]}, }) require.NoError(t, err)