mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-07-27 15:16:43 +00:00
111 lines
2.3 KiB
Go
111 lines
2.3 KiB
Go
package network
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/p2p/adapters"
|
|
p2ptest "github.com/ethereum/go-ethereum/p2p/testing"
|
|
)
|
|
|
|
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 newHiveTester(t *testing.T, params *HiveParams) (*bzzTester, *Hive) {
|
|
// setup
|
|
addr := RandomAddr() // tested peers peer address
|
|
to := NewTestOverlay(addr.OverlayAddr()) // overlay topology drive
|
|
pp := NewHive(params, to) // hive
|
|
|
|
ct := BzzCodeMap(DiscoveryMsgs...) // bzz protocol code map
|
|
services := func(p Peer) error {
|
|
pp.Add(p)
|
|
p.DisconnectHook(func(err error) {
|
|
pp.Remove(p)
|
|
})
|
|
return nil
|
|
}
|
|
|
|
return newBzzBaseTester(t, 1, addr, ct, services), pp
|
|
}
|
|
|
|
func TestOverlayRegistration(t *testing.T) {
|
|
params := NewHiveParams()
|
|
params.Discovery = false
|
|
s, pp := newHiveTester(t, params)
|
|
defer s.Stop()
|
|
|
|
id := s.Ids[0]
|
|
raddr := NewPeerAddrFromNodeId(id)
|
|
|
|
s.runHandshakes()
|
|
|
|
// hive should have called the overlay
|
|
if pp.Overlay.(*testOverlay).posMap[string(raddr.OverlayAddr())] == nil {
|
|
t.Fatalf("Overlay#On not called on new peer")
|
|
}
|
|
|
|
}
|
|
|
|
func TestRegisterAndConnect(t *testing.T) {
|
|
params := NewHiveParams()
|
|
s, pp := newHiveTester(t, params)
|
|
defer s.Stop()
|
|
|
|
id := s.Ids[0]
|
|
raddr := NewPeerAddrFromNodeId(id)
|
|
|
|
pp.Register(raddr)
|
|
|
|
// 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)
|
|
defer pp.Stop()
|
|
tc.ticker <- time.Now()
|
|
|
|
s.runHandshakes()
|
|
|
|
if pp.Overlay.(*testOverlay).posMap[string(raddr.OverlayAddr())] == nil {
|
|
t.Fatalf("Overlay#On not called on new peer")
|
|
}
|
|
|
|
// retrieve and broadcast
|
|
ord := order(raddr.OverlayAddr())
|
|
o := 0
|
|
if ord == 0 {
|
|
o = 1
|
|
}
|
|
s.TestExchanges(p2ptest.Exchange{
|
|
Label: "getPeersMsg message",
|
|
Expects: []p2ptest.Expect{
|
|
p2ptest.Expect{
|
|
Code: 1,
|
|
Msg: &getPeersMsg{uint8(o), 5},
|
|
Peer: id,
|
|
},
|
|
},
|
|
})
|
|
}
|