package network import ( "testing" "github.com/multiformats/go-multiaddr" "github.com/nspcc-dev/neofs-api-go/pkg/netmap" "github.com/stretchr/testify/require" ) func TestAddressFromString(t *testing.T) { t.Run("valid addresses", func(t *testing.T) { testcases := []struct { inp string exp multiaddr.Multiaddr }{ {":8080", buildMultiaddr("/ip4/0.0.0.0/tcp/8080", t)}, {"example.com:7070", buildMultiaddr("/dns4/example.com/tcp/7070", t)}, {"213.44.87.1:32512", buildMultiaddr("/ip4/213.44.87.1/tcp/32512", t)}, {"[2004:eb1::1]:8080", buildMultiaddr("/ip6/2004:eb1::1/tcp/8080", t)}, } for _, testcase := range testcases { addr, err := AddressFromString(testcase.inp) require.NoError(t, err) require.Equal(t, testcase.exp, addr.ma, testcase.inp) } }) } func TestAddress_HostAddrString(t *testing.T) { t.Run("valid addresses", func(t *testing.T) { testcases := []struct { ma multiaddr.Multiaddr exp string }{ {buildMultiaddr("/dns4/neofs.bigcorp.com/tcp/8080", t), "neofs.bigcorp.com:8080"}, {buildMultiaddr("/ip4/172.16.14.1/tcp/8080", t), "172.16.14.1:8080"}, } for _, testcase := range testcases { addr := Address{testcase.ma} got := addr.HostAddr() require.Equal(t, testcase.exp, got) } }) t.Run("invalid addresses", func(t *testing.T) { testcases := []multiaddr.Multiaddr{ buildMultiaddr("/tcp/8080", t), } for _, testcase := range testcases { addr := Address{testcase} require.Panics(t, func() { addr.HostAddr() }) } }) } func buildMultiaddr(s string, t *testing.T) multiaddr.Multiaddr { ma, err := multiaddr.NewMultiaddr(s) require.NoError(t, err) return ma } func TestAddress_WriteToNodeInfo(t *testing.T) { a := "127.0.0.1:8080" addr, err := AddressFromString(a) require.NoError(t, err) var ni netmap.NodeInfo addr.WriteToNodeInfo(&ni) restored, err := AddressFromString(ni.Address()) require.NoError(t, err) require.True(t, restored.Equal(*addr)) }