mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-22 19:40:37 +00:00
d17996f8b0
* Update V3 from V4 * Fix build v3 -> v4 * Update ssz * Update beacon_chain.pb.go * Fix formatter import * Update update-mockgen.sh comment to v4 * Fix conflicts. Pass build and tests * Fix test
50 lines
1.5 KiB
Go
50 lines
1.5 KiB
Go
package network_test
|
|
|
|
import (
|
|
"net"
|
|
"regexp"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/v4/network"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/require"
|
|
)
|
|
|
|
func TestExternalIPv4(t *testing.T) {
|
|
// Regular expression format for IPv4
|
|
IPv4Format := `\.\d{1,3}\.\d{1,3}\b`
|
|
test, err := network.ExternalIPv4()
|
|
require.NoError(t, err)
|
|
|
|
valid := regexp.MustCompile(IPv4Format)
|
|
assert.Equal(t, true, valid.MatchString(test))
|
|
}
|
|
|
|
func TestRetrieveIP(t *testing.T) {
|
|
ip, err := network.ExternalIP()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
retIP := net.ParseIP(ip)
|
|
if retIP.To4() == nil && retIP.To16() == nil {
|
|
t.Errorf("An invalid IP was retrieved: %s", ip)
|
|
}
|
|
}
|
|
|
|
func TestSortAddresses(t *testing.T) {
|
|
testAddresses := []net.IP{
|
|
{0xff, 0x02, 0xAA, 0, 0x1F, 0, 0, 0, 0, 0, 0x02, 0x2E, 0, 0, 0x36, 0x45},
|
|
{0xff, 0x02, 0xAA, 0, 0x1F, 0, 0x2E, 0, 0, 0x36, 0x45, 0, 0, 0, 0, 0x02},
|
|
{0xAA, 0x11, 0x33, 0x19},
|
|
{0x01, 0xBF, 0x33, 0x10},
|
|
{0x03, 0x89, 0x33, 0x13},
|
|
}
|
|
|
|
sortedAddrs := network.SortAddresses(testAddresses)
|
|
assert.Equal(t, true, sortedAddrs[0].To4() != nil, "expected ipv4 address")
|
|
assert.Equal(t, true, sortedAddrs[1].To4() != nil, "expected ipv4 address")
|
|
assert.Equal(t, true, sortedAddrs[2].To4() != nil, "expected ipv4 address")
|
|
assert.Equal(t, true, sortedAddrs[3].To16() != nil && sortedAddrs[3].To4() == nil, "expected ipv6 address")
|
|
assert.Equal(t, true, sortedAddrs[4].To16() != nil && sortedAddrs[4].To4() == nil, "expected ipv6 address")
|
|
}
|