diff options
author | Janos Guljas <janos@resenje.org> | 2018-09-25 22:57:31 +0800 |
---|---|---|
committer | Janos Guljas <janos@resenje.org> | 2018-09-25 22:57:31 +0800 |
commit | 24349144b6c0642755569268bab56b9033743212 (patch) | |
tree | 9d9d2b6659fd8a56512dfc807aafe4b733165ae1 /swarm/pss/pss_test.go | |
parent | 7d56602391e155e2ce9ba7c261300a1804ab9972 (diff) | |
parent | d3441ebb563439bac0837d70591f92e2c6080303 (diff) | |
download | go-tangerine-24349144b6c0642755569268bab56b9033743212.tar go-tangerine-24349144b6c0642755569268bab56b9033743212.tar.gz go-tangerine-24349144b6c0642755569268bab56b9033743212.tar.bz2 go-tangerine-24349144b6c0642755569268bab56b9033743212.tar.lz go-tangerine-24349144b6c0642755569268bab56b9033743212.tar.xz go-tangerine-24349144b6c0642755569268bab56b9033743212.tar.zst go-tangerine-24349144b6c0642755569268bab56b9033743212.zip |
Merge branch 'master' into max-stream-peer-servers
Diffstat (limited to 'swarm/pss/pss_test.go')
-rw-r--r-- | swarm/pss/pss_test.go | 45 |
1 files changed, 20 insertions, 25 deletions
diff --git a/swarm/pss/pss_test.go b/swarm/pss/pss_test.go index 6ba04cb5d..574714114 100644 --- a/swarm/pss/pss_test.go +++ b/swarm/pss/pss_test.go @@ -42,7 +42,7 @@ import ( "github.com/ethereum/go-ethereum/metrics/influxdb" "github.com/ethereum/go-ethereum/node" "github.com/ethereum/go-ethereum/p2p" - "github.com/ethereum/go-ethereum/p2p/discover" + "github.com/ethereum/go-ethereum/p2p/enode" "github.com/ethereum/go-ethereum/p2p/protocols" "github.com/ethereum/go-ethereum/p2p/simulations" "github.com/ethereum/go-ethereum/p2p/simulations/adapters" @@ -576,7 +576,7 @@ func TestMismatch(t *testing.T) { Name: pssProtocolName, Version: 0, } - nid, _ := discover.HexID("0x01") + nid := enode.ID{0x01} wrongpsspeer := network.NewPeer(&network.BzzPeer{ Peer: protocols.NewPeer(p2p.NewPeer(nid, common.ToHex(wrongpssaddr.Over()), []p2p.Cap{wrongpsscap}), rw, nil), BzzAddr: &network.BzzAddr{OAddr: wrongpssaddr.Over(), UAddr: nil}, @@ -588,7 +588,7 @@ func TestMismatch(t *testing.T) { Name: "nopss", Version: 1, } - nid, _ = discover.HexID("0x02") + nid = enode.ID{0x02} nopsspeer := network.NewPeer(&network.BzzPeer{ Peer: protocols.NewPeer(p2p.NewPeer(nid, common.ToHex(nopssaddr.Over()), []p2p.Cap{nopsscap}), rw, nil), BzzAddr: &network.BzzAddr{OAddr: nopssaddr.Over(), UAddr: nil}, @@ -923,11 +923,11 @@ func testSendAsym(t *testing.T) { type Job struct { Msg []byte - SendNode discover.NodeID - RecvNode discover.NodeID + SendNode enode.ID + RecvNode enode.ID } -func worker(id int, jobs <-chan Job, rpcs map[discover.NodeID]*rpc.Client, pubkeys map[discover.NodeID]string, topic string) { +func worker(id int, jobs <-chan Job, rpcs map[enode.ID]*rpc.Client, pubkeys map[enode.ID]string, topic string) { for j := range jobs { rpcs[j.SendNode].Call(nil, "pss_sendAsym", pubkeys[j.RecvNode], topic, hexutil.Encode(j.Msg)) } @@ -977,7 +977,7 @@ func TestNetwork10000(t *testing.T) { func testNetwork(t *testing.T) { type msgnotifyC struct { - id discover.NodeID + id enode.ID msgIdx int } @@ -989,16 +989,16 @@ func testNetwork(t *testing.T) { log.Info("network test", "nodecount", nodecount, "msgcount", msgcount, "addrhintsize", addrsize) - nodes := make([]discover.NodeID, nodecount) - bzzaddrs := make(map[discover.NodeID]string, nodecount) - rpcs := make(map[discover.NodeID]*rpc.Client, nodecount) - pubkeys := make(map[discover.NodeID]string, nodecount) + nodes := make([]enode.ID, nodecount) + bzzaddrs := make(map[enode.ID]string, nodecount) + rpcs := make(map[enode.ID]*rpc.Client, nodecount) + pubkeys := make(map[enode.ID]string, nodecount) sentmsgs := make([][]byte, msgcount) recvmsgs := make([]bool, msgcount) - nodemsgcount := make(map[discover.NodeID]int, nodecount) + nodemsgcount := make(map[enode.ID]int, nodecount) - trigger := make(chan discover.NodeID) + trigger := make(chan enode.ID) var a adapters.NodeAdapter if adapter == "exec" { @@ -1038,7 +1038,7 @@ func testNetwork(t *testing.T) { time.Sleep(1 * time.Second) - triggerChecks := func(trigger chan discover.NodeID, id discover.NodeID, rpcclient *rpc.Client, topic string) error { + triggerChecks := func(trigger chan enode.ID, id enode.ID, rpcclient *rpc.Client, topic string) error { msgC := make(chan APIMsg) ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel() @@ -1542,12 +1542,11 @@ func setupNetwork(numnodes int, allowRaw bool) (clients []*rpc.Client, err error func newServices(allowRaw bool) adapters.Services { stateStore := state.NewInmemoryStore() - kademlias := make(map[discover.NodeID]*network.Kademlia) - kademlia := func(id discover.NodeID) *network.Kademlia { + kademlias := make(map[enode.ID]*network.Kademlia) + kademlia := func(id enode.ID) *network.Kademlia { if k, ok := kademlias[id]; ok { return k } - addr := network.NewAddrFromNodeID(id) params := network.NewKadParams() params.MinProxBinSize = 2 params.MaxBinSize = 3 @@ -1555,7 +1554,7 @@ func newServices(allowRaw bool) adapters.Services { params.MaxRetries = 1000 params.RetryExponent = 2 params.RetryInterval = 1000000 - kademlias[id] = network.NewKademlia(addr.Over(), params) + kademlias[id] = network.NewKademlia(id[:], params) return kademlias[id] } return adapters.Services{ @@ -1606,7 +1605,7 @@ func newServices(allowRaw bool) adapters.Services { return ps, nil }, "bzz": func(ctx *adapters.ServiceContext) (node.Service, error) { - addr := network.NewAddrFromNodeID(ctx.Config.ID) + addr := network.NewAddr(ctx.Config.Node()) hp := network.NewHiveParams() hp.Discovery = false config := &network.BzzConfig{ @@ -1620,16 +1619,12 @@ func newServices(allowRaw bool) adapters.Services { } func newTestPss(privkey *ecdsa.PrivateKey, kad *network.Kademlia, ppextra *PssParams) *Pss { - - var nid discover.NodeID - copy(nid[:], crypto.FromECDSAPub(&privkey.PublicKey)) - addr := network.NewAddrFromNodeID(nid) - + nid := enode.PubkeyToIDV4(&privkey.PublicKey) // set up routing if kademlia is not passed to us if kad == nil { kp := network.NewKadParams() kp.MinProxBinSize = 3 - kad = network.NewKademlia(addr.Over(), kp) + kad = network.NewKademlia(nid[:], kp) } // create pss |