mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-08-06 12:03:47 +00:00
46 lines
859 B
Go
46 lines
859 B
Go
package network
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/log"
|
|
p2ptest "github.com/ethereum/go-ethereum/p2p/testing"
|
|
)
|
|
|
|
/***
|
|
*
|
|
* - after connect, that outgoing subpeersmsg is sent
|
|
*
|
|
*/
|
|
func TestDiscovery(t *testing.T) {
|
|
addr := RandomAddr()
|
|
to := NewKademlia(addr.OAddr, NewKadParams())
|
|
ct := BzzCodeMap(DiscoveryMsgs...)
|
|
|
|
services := func(p Peer) error {
|
|
dp := NewDiscovery(p, to)
|
|
to.On(dp)
|
|
log.Trace(fmt.Sprintf("kademlia on %v", p))
|
|
p.DisconnectHook(func(err error) {
|
|
to.Off(p)
|
|
})
|
|
return nil
|
|
}
|
|
|
|
s := newBzzBaseTester(t, 1, addr, ct, services)
|
|
defer s.Stop()
|
|
|
|
s.runHandshakes()
|
|
// o := 0
|
|
s.TestExchanges(p2ptest.Exchange{
|
|
Label: "outgoing SubPeersMsg",
|
|
Expects: []p2ptest.Expect{
|
|
p2ptest.Expect{
|
|
Code: 3,
|
|
Msg: &subPeersMsg{ProxLimit: 0},
|
|
Peer: s.ProtocolTester.Ids[0],
|
|
},
|
|
},
|
|
})
|
|
}
|