ceremonyclient/go-libp2p/p2p/test/notifications/notification_test.go

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 tStringCast(str string) ma.Multiaddr {
	m, _ := ma.StringCast(str)
	return m
}

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, event.Added, ev.Current[0].Action)
		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
	require.NoError(t, h.Network().Listen(tStringCast("/ip4/127.0.0.1/udp/0/quic-v1")))
	var addedAddr ma.Multiaddr
	select {
	case e := <-sub.Out():
		ev := e.(event.EvtLocalAddressesUpdated)
		require.Empty(t, ev.Removed)
		require.Len(t, ev.Current, 2)
		var maintainedAddr ma.Multiaddr
		for _, e := range ev.Current {
			switch e.Action {
			case event.Added:
				addedAddr = e.Address
			case event.Maintained:
				maintainedAddr = e.Address
			default:
				t.Fatal("unexpected action")
			}
		}
		require.Equal(t, initialAddr, maintainedAddr)
		_, err = addedAddr.ValueForProtocol(ma.P_QUIC_V1)
		require.NoError(t, err)
		portStr, err := addedAddr.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, addedAddr)
}