mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-07-27 23:26:44 +00:00
119 lines
2.7 KiB
Go
119 lines
2.7 KiB
Go
package network
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/p2p/adapters"
|
|
// "github.com/ethereum/go-ethereum/p2p/discover"
|
|
"github.com/ethereum/go-ethereum/logger"
|
|
"github.com/ethereum/go-ethereum/logger/glog"
|
|
// "github.com/ethereum/go-ethereum/p2p/protocols"
|
|
p2ptest "github.com/ethereum/go-ethereum/p2p/testing"
|
|
)
|
|
|
|
func init() {
|
|
glog.SetV(logger.Detail)
|
|
glog.SetToStderr(true)
|
|
}
|
|
|
|
type testConnect struct {
|
|
mu sync.Mutex
|
|
conns []string
|
|
connectf func(c string) error
|
|
ticker chan time.Time
|
|
}
|
|
|
|
func (self *testConnect) ping() <-chan time.Time {
|
|
return self.ticker
|
|
}
|
|
|
|
func (self *testConnect) connect(na string) error {
|
|
self.mu.Lock()
|
|
defer self.mu.Unlock()
|
|
self.conns = append(self.conns, na)
|
|
self.connectf(na)
|
|
return nil
|
|
}
|
|
|
|
func TestOverlayRegistration(t *testing.T) {
|
|
// setup
|
|
addr := RandomAddr() // tested peers peer address
|
|
to := NewTestOverlay(addr.OverlayAddr()) // overlay topology driver
|
|
pp := NewHive(NewHiveParams(), to) // hive
|
|
ct := BzzCodeMap(HiveMsgs...) // bzz protocol code map
|
|
|
|
s := newBzzTester(t, 1, addr, ct, nil)
|
|
|
|
// connect to the other peer
|
|
id := s.Ids[0]
|
|
raddr := NewPeerAddrFromNodeId(id)
|
|
s.runHandshakes()
|
|
|
|
// hive should have called the overlay
|
|
if to.posMap[string(raddr.OverlayAddr())] == nil {
|
|
t.Fatalf("Overlay#On not called on new peer")
|
|
}
|
|
}
|
|
|
|
func TestRegisterAndConnect(t *testing.T) {
|
|
addr := RandomAddr()
|
|
to := NewTestOverlay(addr.OverlayAddr())
|
|
pp := NewHive(NewHiveParams(), to)
|
|
ct := BzzCodeMap(HiveMsgs...)
|
|
s := newBzzTester(t, 0, addr, pp, ct, nil)
|
|
|
|
// register the node with the peerPool
|
|
id := p2ptest.RandomNodeId()
|
|
// pretend to start the node
|
|
s.Start(id)
|
|
|
|
// register another address
|
|
raddr := NewPeerAddrFromNodeId(id)
|
|
pp.Register(raddr)
|
|
glog.V(5).Infof("%v", pp)
|
|
|
|
// start the hive and wait for the connection
|
|
tc := &testConnect{
|
|
connectf: func(c string) error {
|
|
s.Connect(adapters.NewNodeIdFromHex(c))
|
|
return nil
|
|
},
|
|
ticker: make(chan time.Time),
|
|
}
|
|
pp.Start(tc.connect, tc.ping)
|
|
tc.ticker <- time.Now()
|
|
|
|
// run bzz handshake
|
|
s.runHandshakes()
|
|
if to.posMap[string(raddr.OverlayAddr())] == nil {
|
|
t.Fatalf("Overlay#On not called on new peer")
|
|
}
|
|
|
|
// retrieve and broadcast
|
|
glog.V(6).Infof("check peer requests for %v", id)
|
|
ord := order(raddr.OverlayAddr())
|
|
o := 0
|
|
if ord == 0 {
|
|
o = 1
|
|
}
|
|
s.TestExchanges(p2ptest.Exchange{
|
|
Expects: []p2ptest.Expect{
|
|
p2ptest.Expect{
|
|
Code: 3,
|
|
Msg: &SubPeersMsg{ProxLimit: 0, MinProxBinSize: 8},
|
|
Peer: s.ExchangeSession.Id(1),
|
|
},
|
|
},
|
|
})
|
|
s.TestExchanges(p2ptest.Exchange{
|
|
Expects: []p2ptest.Expect{
|
|
p2ptest.Expect{
|
|
Code: 1,
|
|
Msg: &getPeersMsg{uint8(o), 5},
|
|
Peer: id,
|
|
},
|
|
},
|
|
})
|
|
}
|