12ddb0739e
* feat: obtain external address for rendezvous If the ext ip returned by geth is 127.0.0.1, it will attempt to obtain the external IP address via rendezvous and use that to register the ens record later * fix: failing test * fix: code review, and adding external ip address to logs
107 lines
3.3 KiB
Go
107 lines
3.3 KiB
Go
package discovery
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
"time"
|
|
|
|
lcrypto "github.com/libp2p/go-libp2p-core/crypto"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
"github.com/status-im/rendezvous/server"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/syndtr/goleveldb/leveldb"
|
|
"github.com/syndtr/goleveldb/leveldb/storage"
|
|
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/ethereum/go-ethereum/p2p/discv5"
|
|
"github.com/ethereum/go-ethereum/p2p/enode"
|
|
"github.com/ethereum/go-ethereum/p2p/enr"
|
|
)
|
|
|
|
func makeTestRendezvousServer(t *testing.T, addr string) *server.Server {
|
|
priv, _, err := lcrypto.GenerateKeyPair(lcrypto.Secp256k1, 0)
|
|
require.NoError(t, err)
|
|
laddr, err := ma.NewMultiaddr(addr)
|
|
require.NoError(t, err)
|
|
db, err := leveldb.Open(storage.NewMemStorage(), nil)
|
|
require.NoError(t, err)
|
|
srv := server.NewServer(laddr, priv, server.NewStorage(db))
|
|
require.NoError(t, srv.Start())
|
|
return srv
|
|
}
|
|
|
|
func TestRendezvousDiscovery(t *testing.T) {
|
|
srv := makeTestRendezvousServer(t, "/ip4/127.0.0.1/tcp/7777")
|
|
defer srv.Stop()
|
|
identity, err := crypto.GenerateKey()
|
|
require.NoError(t, err)
|
|
node := enode.NewV4(&identity.PublicKey, net.IP{10, 10, 10, 10}, 10, 20)
|
|
c, err := NewRendezvous([]ma.Multiaddr{srv.Addr()}, identity, node)
|
|
require.NoError(t, err)
|
|
require.NoError(t, c.Start())
|
|
require.True(t, c.Running())
|
|
|
|
topic := "test"
|
|
stop := make(chan struct{})
|
|
go func() { assert.NoError(t, c.Register(topic, stop)) }()
|
|
|
|
period := make(chan time.Duration, 1)
|
|
period <- 100 * time.Millisecond
|
|
found := make(chan *discv5.Node, 1)
|
|
lookup := make(chan bool)
|
|
go func() { assert.NoError(t, c.Discover(topic, period, found, lookup)) }()
|
|
|
|
select {
|
|
case n := <-found:
|
|
assert.Equal(t, discv5.PubkeyID(&identity.PublicKey), n.ID)
|
|
case <-time.After(10 * time.Second):
|
|
assert.Fail(t, "failed waiting to discover a node")
|
|
}
|
|
close(stop)
|
|
close(period)
|
|
}
|
|
|
|
func TestMakeRecordReturnsCachedRecord(t *testing.T) {
|
|
identity, err := crypto.GenerateKey()
|
|
require.NoError(t, err)
|
|
record := enr.Record{}
|
|
require.NoError(t, enode.SignV4(&record, identity))
|
|
c := NewRendezvousWithENR(nil, record)
|
|
rst, err := c.MakeRecord(nil)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, enode.V4ID{}.NodeAddr(&rst))
|
|
require.Equal(t, enode.V4ID{}.NodeAddr(&record), enode.V4ID{}.NodeAddr(&rst))
|
|
}
|
|
|
|
func TestRendezvousRegisterAndDiscoverExitGracefully(t *testing.T) {
|
|
r, err := NewRendezvous(make([]ma.Multiaddr, 1), nil, nil)
|
|
require.NoError(t, err)
|
|
require.NoError(t, r.Start())
|
|
require.NoError(t, r.Stop())
|
|
require.EqualError(t, errDiscoveryIsStopped, r.register(r.getRandomServer(), "", enr.Record{}).Error())
|
|
_, err = r.discoverRequest(nil, "")
|
|
require.EqualError(t, errDiscoveryIsStopped, err.Error())
|
|
}
|
|
|
|
func TestStopStoppedNode(t *testing.T) {
|
|
r, err := NewRendezvous(make([]ma.Multiaddr, 1), nil, nil)
|
|
require.NoError(t, err)
|
|
require.NoError(t, r.Stop())
|
|
}
|
|
|
|
func BenchmarkRendezvousStart(b *testing.B) {
|
|
identity, err := crypto.GenerateKey()
|
|
require.NoError(b, err)
|
|
addr, err := ma.NewMultiaddr("/ip4/127.0.0.1/tcp/7777")
|
|
require.NoError(b, err)
|
|
node := enode.NewV4(&identity.PublicKey, net.IP{10, 10, 10, 10}, 10, 20)
|
|
|
|
b.ResetTimer()
|
|
for n := 0; n < b.N; n++ {
|
|
c, err := NewRendezvous([]ma.Multiaddr{addr}, identity, node)
|
|
require.NoError(b, err)
|
|
require.NoError(b, c.Start())
|
|
}
|
|
}
|