woop/p2p/tests/address_test.go

package p2ptests

import (
	"strings"
	"testing"

	p2ptypes "github.com/harmony-one/harmony/p2p/types"
	"github.com/harmony-one/harmony/test/helpers"
	"github.com/stretchr/testify/assert"
)

func TestMultiAddressParsing(t *testing.T) {
	t.Parallel()

	multiAddresses, err := p2ptypes.StringsToMultiAddrs(helpers.Bootnodes)
	assert.NoError(t, err)
	assert.Equal(t, len(helpers.Bootnodes), len(multiAddresses))

	for index, multiAddress := range multiAddresses {
		assert.Equal(t, multiAddress.String(), helpers.Bootnodes[index])
	}
}

func TestAddressListConversionToString(t *testing.T) {
	t.Parallel()

	multiAddresses, err := p2ptypes.StringsToMultiAddrs(helpers.Bootnodes)
	assert.NoError(t, err)
	assert.Equal(t, len(helpers.Bootnodes), len(multiAddresses))

	expected := strings.Join(helpers.Bootnodes[:], ",")
	var addressList p2ptypes.AddrList = multiAddresses
	assert.Equal(t, expected, addressList.String())
}

func TestAddressListConversionFromString(t *testing.T) {
	t.Parallel()

	multiAddresses, err := p2ptypes.StringsToMultiAddrs(helpers.Bootnodes)
	assert.NoError(t, err)
	assert.Equal(t, len(helpers.Bootnodes), len(multiAddresses))

	addressString := strings.Join(helpers.Bootnodes[:], ",")
	var addressList p2ptypes.AddrList = multiAddresses
	addressList.Set(addressString)
	assert.Equal(t, len(addressList), len(multiAddresses))
	assert.Equal(t, addressList[0], multiAddresses[0])
}