mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-22 11:32:09 +00:00
f49637b09b
* Add network pkg * Go fmt * Update discovery_test.go Co-authored-by: Raul Jordan <raul@prysmaticlabs.com>
50 lines
1.5 KiB
Go
50 lines
1.5 KiB
Go
package network_test
|
|
|
|
import (
|
|
"net"
|
|
"regexp"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/network"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/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")
|
|
}
|