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

91 lines
2.6 KiB
Go
Raw Normal View History

2023-08-21 03:50:38 +00:00
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
require.NoError(t, h.Network().Listen(ma.StringCast("/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)
}