forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-07-28 23:56:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

feat: fix discovery test

Browse source 
Signed-off-by: Chen Kai <281165273grape@gmail.com>
This commit is contained in:
Chen Kai 2024-02-23 14:42:30 +08:00
parent c8d7aa56ce
commit 00fcda62fa
1 changed files with 33 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

60
p2p/discover/portal_protocol_test.go
View file

@ -12,6 +12,7 @@ import (
"time"
"github.com/prysmaticlabs/go-bitfield"
"golang.org/x/exp/slices"
"github.com/ethereum/go-ethereum/internal/testlog"
"github.com/ethereum/go-ethereum/log"
@ -200,50 +201,55 @@ func TestPortalWireProtocolUdp(t *testing.T) {
func TestPortalWireProtocol(t *testing.T) {
node1, err := setupLocalPortalNode(":7777", nil)
assert.NoError(t, err)
node1.log = testlog.Logger(t, log.LvlTrace)
node1.log = testlog.Logger(t, log.LevelDebug)
err = node1.Start()
assert.NoError(t, err)
fmt.Println(node1.localNode.Node().String())
time.Sleep(15 * time.Second)
node2, err := setupLocalPortalNode(":7778", []*enode.Node{node1.localNode.Node()})
assert.NoError(t, err)
node2.log = testlog.Logger(t, log.LvlTrace)
node2.log = testlog.Logger(t, log.LevelDebug)
err = node2.Start()
assert.NoError(t, err)
fmt.Println(node2.localNode.Node().String())
time.Sleep(15 * time.Second)
node3, err := setupLocalPortalNode(":7779", []*enode.Node{node1.localNode.Node()})
assert.NoError(t, err)
node3.log = testlog.Logger(t, log.LvlTrace)
node3.log = testlog.Logger(t, log.LevelDebug)
err = node3.Start()
assert.NoError(t, err)
fmt.Println(node3.localNode.Node().String())
time.Sleep(10 * time.Second)
//assert.Equal(t, 2, len(node1.table.Nodes()))
//assert.Equal(t, 2, len(node2.table.Nodes()))
//assert.Equal(t, 2, len(node3.table.Nodes()))
time.Sleep(15 * time.Second)
//slices.ContainsFunc(node1.table.Nodes(), func(n *enode.Node) bool {
// return n.ID() == node2.localNode.Node().ID()
//})
//slices.ContainsFunc(node1.table.Nodes(), func(n *enode.Node) bool {
// return n.ID() == node3.localNode.Node().ID()
//})
//
//slices.ContainsFunc(node2.table.Nodes(), func(n *enode.Node) bool {
// return n.ID() == node1.localNode.Node().ID()
//})
//slices.ContainsFunc(node2.table.Nodes(), func(n *enode.Node) bool {
// return n.ID() == node3.localNode.Node().ID()
//})
//
//slices.ContainsFunc(node3.table.Nodes(), func(n *enode.Node) bool {
// return n.ID() == node1.localNode.Node().ID()
//})
//slices.ContainsFunc(node3.table.Nodes(), func(n *enode.Node) bool {
// return n.ID() == node2.localNode.Node().ID()
//})
assert.Equal(t, 2, len(node1.table.Nodes()))
assert.Equal(t, 2, len(node2.table.Nodes()))
assert.Equal(t, 2, len(node3.table.Nodes()))
slices.ContainsFunc(node1.table.Nodes(), func(n *enode.Node) bool {
return n.ID() == node2.localNode.Node().ID()
})
slices.ContainsFunc(node1.table.Nodes(), func(n *enode.Node) bool {
return n.ID() == node3.localNode.Node().ID()
})
slices.ContainsFunc(node2.table.Nodes(), func(n *enode.Node) bool {
return n.ID() == node1.localNode.Node().ID()
})
slices.ContainsFunc(node2.table.Nodes(), func(n *enode.Node) bool {
return n.ID() == node3.localNode.Node().ID()
})
slices.ContainsFunc(node3.table.Nodes(), func(n *enode.Node) bool {
return n.ID() == node1.localNode.Node().ID()
})
slices.ContainsFunc(node3.table.Nodes(), func(n *enode.Node) bool {
return n.ID() == node2.localNode.Node().ID()
})
err = node1.storage.Put(node1.toContentId([]byte("test_key")), []byte("test_value"))
assert.NoError(t, err)