forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-18 01:43:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

swarm: Fix bzz initialization from tests

Browse source
This commit is contained in:
Balint Gabor 2018-03-05 18:16:46 +01:00
parent 79abcac95e
commit 86df48a2a1
6 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
swarm/network/protocol.go
View file

@ -154,7 +154,7 @@ func (b *Bzz) NodeInfo() interface{} {
// * handshake/hive // * handshake/hive
// * discovery // * discovery
func (b *Bzz) Protocols() []p2p.Protocol { func (b *Bzz) Protocols() []p2p.Protocol {
return []p2p.Protocol{ protocol := []p2p.Protocol{
{ {
Name: BzzSpec.Name, Name: BzzSpec.Name,
Version: BzzSpec.Version, Version: BzzSpec.Version,
@ -170,13 +170,16 @@ func (b *Bzz) Protocols() []p2p.Protocol {
NodeInfo: b.Hive.NodeInfo, NodeInfo: b.Hive.NodeInfo,
PeerInfo: b.Hive.PeerInfo, PeerInfo: b.Hive.PeerInfo,
}, },
{ }
if b.streamerSpec != nil && b.streamerRun != nil {
protocol = append(protocol, p2p.Protocol{
Name: b.streamerSpec.Name, Name: b.streamerSpec.Name,
Version: b.streamerSpec.Version, Version: b.streamerSpec.Version,
Length: b.streamerSpec.Length(), Length: b.streamerSpec.Length(),
Run: b.RunProtocol(b.streamerSpec, b.streamerRun), Run: b.RunProtocol(b.streamerSpec, b.streamerRun),
}, })
} }
return protocol
} }
// APIs returns the APIs offered by bzz // APIs returns the APIs offered by bzz

2
swarm/network/protocol_test.go
View file

@ -137,7 +137,7 @@ func newBzzHandshakeTester(t *testing.T, n int, addr *BzzAddr) *bzzTester {
HiveParams: NewHiveParams(), HiveParams: NewHiveParams(),
} }
kad := NewKademlia(addr.OAddr, NewKadParams()) kad := NewKademlia(addr.OAddr, NewKadParams())
bzz := NewBzz(config, kad, nil) bzz := NewBzz(config, kad, nil, nil, nil)
s := p2ptest.NewProtocolTester(t, NewNodeIDFromAddr(addr), 1, bzz.runBzz) s := p2ptest.NewProtocolTester(t, NewNodeIDFromAddr(addr), 1, bzz.runBzz)

2
swarm/network/simulations/discovery/discovery_test.go
View file

@ -331,5 +331,5 @@ func newService(ctx *adapters.ServiceContext) (node.Service, error) {
HiveParams: hp, HiveParams: hp,
} }
return network.NewBzz(config, kad, nil), nil return network.NewBzz(config, kad, nil, nil, nil), nil
} }

2
swarm/network/simulations/overlay.go
View file

@ -69,7 +69,7 @@ func (s *Simulation) NewService(ctx *adapters.ServiceContext) (node.Service, err
HiveParams: hp, HiveParams: hp,
} }
return network.NewBzz(config, kad, store), nil return network.NewBzz(config, kad, store, nil, nil), nil
} }
func createMockers() map[string]*simulations.MockerConfig { func createMockers() map[string]*simulations.MockerConfig {

2
swarm/pss/client/client_test.go
View file

@ -263,7 +263,7 @@ func newServices() adapters.Services {
UnderlayAddr: addr.Under(), UnderlayAddr: addr.Under(),
HiveParams: hp, HiveParams: hp,
} }
return network.NewBzz(config, kademlia(ctx.Config.ID), stateStore), nil return network.NewBzz(config, kademlia(ctx.Config.ID), stateStore, nil, nil), nil
}, },
} }
} }

2
swarm/pss/pss_test.go
View file

@ -1192,7 +1192,7 @@ func newServices() adapters.Services {
UnderlayAddr: addr.Under(), UnderlayAddr: addr.Under(),
HiveParams: hp, HiveParams: hp,
} }
return network.NewBzz(config, kademlia(ctx.Config.ID), stateStore), nil return network.NewBzz(config, kademlia(ctx.Config.ID), stateStore, nil, nil), nil
}, },
} }
} }