mirror of
https://source.quilibrium.com/quilibrium/ceremonyclient.git
synced 2024-12-26 08:35:17 +00:00
96 lines
2.6 KiB
Go
96 lines
2.6 KiB
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)
|
|
}
|