mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 03:51:29 +00:00
d9fd2521af
* applies assertion funcs to p2p/encoder tests * applies assertion funcs to p2p/peers tests * addr_factory_test + broadcaster_test updated * connection_gater_test updated * applies assertion funcs to p2p/service tests * Merge branch 'master' into p2p-apply-testutils-assertions * minor fixes * Merge branch 'master' into p2p-apply-testutils-assertions * Merge refs/heads/master into p2p-apply-testutils-assertions
50 lines
1.7 KiB
Go
50 lines
1.7 KiB
Go
package p2p
|
|
|
|
import (
|
|
"testing"
|
|
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
)
|
|
|
|
func TestRelayAddrs_OnlyFactory(t *testing.T) {
|
|
relay := "/ip4/127.0.0.1/tcp/6660/p2p/QmQ7zhY7nGY66yK1n8hLGevfVyjbtvHSgtZuXkCH9oTrgi"
|
|
f := withRelayAddrs(relay)
|
|
|
|
a, err := ma.NewMultiaddr("/ip4/127.0.0.1/tcp/33201/p2p/QmaXZhW44pwQxBSeLkE5FNeLz8tGTTEsRciFg1DNWXXrWG")
|
|
require.NoError(t, err)
|
|
addrs := []ma.Multiaddr{a}
|
|
|
|
result := f(addrs)
|
|
assert.Equal(t, 2, len(result), "Unexpected number of addresses")
|
|
|
|
expected := "/ip4/127.0.0.1/tcp/6660/p2p/QmQ7zhY7nGY66yK1n8hLGevfVyjbtvHSgtZuXkCH9oTrgi/p2p-circuit/ip4/127.0.0.1/tcp/33201/p2p/QmaXZhW44pwQxBSeLkE5FNeLz8tGTTEsRciFg1DNWXXrWG"
|
|
if result[1].String() != expected {
|
|
t.Errorf("Address at index 1 (%s) is not the expected p2p-circuit address", result[1].String())
|
|
}
|
|
}
|
|
|
|
func TestRelayAddrs_UseNonRelayAddrs(t *testing.T) {
|
|
relay := "/ip4/127.0.0.1/tcp/6660/p2p/QmQ7zhY7nGY66yK1n8hLGevfVyjbtvHSgtZuXkCH9oTrgi"
|
|
f := withRelayAddrs(relay)
|
|
|
|
expected := []string{
|
|
"/ip4/127.0.0.1/tcp/6660/p2p/QmQ7zhY7nGY66yK1n8hLGevfVyjbtvHSgtZuXkCH9oTrgi/p2p-circuit/ip4/127.0.0.1/tcp/33201/p2p/QmaXZhW44pwQxBSeLkE5FNeLz8tGTTEsRciFg1DNWXXrWG",
|
|
"/ip4/127.0.0.1/tcp/6660/p2p/QmQ7zhY7nGY66yK1n8hLGevfVyjbtvHSgtZuXkCH9oTrgi/p2p-circuit/ip4/127.0.0.1/tcp/33203/p2p/QmaXZhW44pwQxBSeLkE5FNeLz8tGTTEsRciFg1DNWXXrWG",
|
|
}
|
|
|
|
addrs := make([]ma.Multiaddr, len(expected))
|
|
for i, addr := range expected {
|
|
a, err := ma.NewMultiaddr(addr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
addrs[i] = a
|
|
}
|
|
|
|
result := f(addrs)
|
|
assert.Equal(t, 2, len(result), "Unexpected number of addresses")
|
|
assert.DeepEqual(t, addrs, result)
|
|
}
|