p2p/test: add test for EvtLocalAddressesUpdated event (#2016)

p2p/test/notifications/notification_test.go

@@ -0,0 +1,81 @@
+package notifications
+
+import (
+	"strconv"
+	"testing"
+	"time"
+
+	"github.com/libp2p/go-libp2p"
+	"github.com/libp2p/go-libp2p/core/event"
+	libp2pquic "github.com/libp2p/go-libp2p/p2p/transport/quic"
+	"github.com/libp2p/go-libp2p/p2p/transport/tcp"
+
+	ma "github.com/multiformats/go-multiaddr"
+	"github.com/stretchr/testify/require"
+)
+
+func portFromString(t *testing.T, s string) int {
+	t.Helper()
+	p, err := strconv.ParseInt(s, 10, 32)
+	require.NoError(t, err)
+	return int(p)
+}
+
+func TestListenAddressNotif(t *testing.T) {
+	h, err := libp2p.New(
+		libp2p.ListenAddrStrings("/ip4/127.0.0.1/tcp/0"),
+		libp2p.Transport(tcp.NewTCPTransport),
+		libp2p.Transport(libp2pquic.NewTransport),
+		libp2p.DisableRelay(),
+	)
+	require.NoError(t, err)
+	defer h.Close()
+	sub, err := h.EventBus().Subscribe(&event.EvtLocalAddressesUpdated{})
+	require.NoError(t, err)
+	defer sub.Close()
+
+	var initialAddr ma.Multiaddr
+	// make sure the event is emitted for the initial listen address
+	select {
+	case e := <-sub.Out():
+		ev := e.(event.EvtLocalAddressesUpdated)
+		require.Empty(t, ev.Removed)
+		require.Len(t, ev.Current, 1)
+		require.Equal(t, ev.Current[0].Action, event.Added)
+		initialAddr = ev.Current[0].Address
+		portStr, err := initialAddr.ValueForProtocol(ma.P_TCP)
+		require.NoError(t, err)
+		require.NotZero(t, portFromString(t, portStr))
+	case <-time.After(500 * time.Millisecond):
+		t.Fatal("timeout")
+	}
+	listenAddrs, err := h.Network().InterfaceListenAddresses()
+	require.NoError(t, err)
+	require.Equal(t, []ma.Multiaddr{initialAddr}, listenAddrs)
+
+	// now start listening on another address
+	var newAddr ma.Multiaddr
+	require.NoError(t, h.Network().Listen(ma.StringCast("/ip4/127.0.0.1/udp/0/quic-v1")))
+	select {
+	case e := <-sub.Out():
+		ev := e.(event.EvtLocalAddressesUpdated)
+		require.Empty(t, ev.Removed)
+		require.Len(t, ev.Current, 2)
+		require.Equal(t, ev.Current[0], event.UpdatedAddress{Address: initialAddr, Action: event.Maintained})
+		require.Equal(t, ev.Current[1].Action, event.Added)
+		newAddr = ev.Current[1].Address
+		_, err = newAddr.ValueForProtocol(ma.P_QUIC_V1)
+		require.NoError(t, err)
+		portStr, err := newAddr.ValueForProtocol(ma.P_UDP)
+		require.NoError(t, err)
+		require.NotZero(t, portFromString(t, portStr))
+	case <-time.After(500 * time.Millisecond):
+		t.Fatal("timeout")
+	}
+
+	listenAddrs, err = h.Network().InterfaceListenAddresses()
+	require.NoError(t, err)
+	require.Len(t, listenAddrs, 2)
+	require.Contains(t, listenAddrs, initialAddr)
+	require.Contains(t, listenAddrs, newAddr)
+}