forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-20 10:52:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

swarm/network: extract no light node local variable to constant

Browse source
This commit is contained in:
Matthew Halpern 2019-02-14 11:12:42 -08:00
parent ba90a4aaa4
commit 95eb97f137
1 changed files with 8 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
swarm/network/protocol_test.go
View file

@ -33,6 +33,8 @@ import (
const ( const (
TestProtocolVersion = 8 TestProtocolVersion = 8
TestProtocolNetworkID = 3 TestProtocolNetworkID = 3
noLightNode = false
) )
var ( var (
@ -164,13 +166,12 @@ func correctBzzHandshake(addr *BzzAddr, lightNode bool) *HandshakeMsg {
} }
func TestBzzHandshakeNetworkIDMismatch(t *testing.T) { func TestBzzHandshakeNetworkIDMismatch(t *testing.T) {
lightNode := false
addr := RandomAddr() addr := RandomAddr()
s := newBzzHandshakeTester(1, addr, lightNode) s := newBzzHandshakeTester(1, addr, noLightNode)
node := s.Nodes[0] node := s.Nodes[0]
err := s.testHandshake( err := s.testHandshake(
correctBzzHandshake(addr, lightNode), correctBzzHandshake(addr, noLightNode),
&HandshakeMsg{Version: TestProtocolVersion, NetworkID: 321, Addr: NewAddr(node)}, &HandshakeMsg{Version: TestProtocolVersion, NetworkID: 321, Addr: NewAddr(node)},
&p2ptest.Disconnect{Peer: node.ID(), Error: fmt.Errorf("Handshake error: Message handler error: (msg code 0): network id mismatch 321 (!= 3)")}, &p2ptest.Disconnect{Peer: node.ID(), Error: fmt.Errorf("Handshake error: Message handler error: (msg code 0): network id mismatch 321 (!= 3)")},
) )
@ -181,13 +182,12 @@ func TestBzzHandshakeNetworkIDMismatch(t *testing.T) {
} }
func TestBzzHandshakeVersionMismatch(t *testing.T) { func TestBzzHandshakeVersionMismatch(t *testing.T) {
lightNode := false
addr := RandomAddr() addr := RandomAddr()
s := newBzzHandshakeTester(1, addr, lightNode) s := newBzzHandshakeTester(1, addr, noLightNode)
node := s.Nodes[0] node := s.Nodes[0]
err := s.testHandshake( err := s.testHandshake(
correctBzzHandshake(addr, lightNode), correctBzzHandshake(addr, noLightNode),
&HandshakeMsg{Version: 0, NetworkID: TestProtocolNetworkID, Addr: NewAddr(node)}, &HandshakeMsg{Version: 0, NetworkID: TestProtocolNetworkID, Addr: NewAddr(node)},
&p2ptest.Disconnect{Peer: node.ID(), Error: fmt.Errorf("Handshake error: Message handler error: (msg code 0): version mismatch 0 (!= %d)", TestProtocolVersion)}, &p2ptest.Disconnect{Peer: node.ID(), Error: fmt.Errorf("Handshake error: Message handler error: (msg code 0): version mismatch 0 (!= %d)", TestProtocolVersion)},
) )
@ -198,13 +198,12 @@ func TestBzzHandshakeVersionMismatch(t *testing.T) {
} }
func TestBzzHandshakeSuccess(t *testing.T) { func TestBzzHandshakeSuccess(t *testing.T) {
lightNode := false
addr := RandomAddr() addr := RandomAddr()
s := newBzzHandshakeTester(1, addr, lightNode) s := newBzzHandshakeTester(1, addr, noLightNode)
node := s.Nodes[0] node := s.Nodes[0]
err := s.testHandshake( err := s.testHandshake(
correctBzzHandshake(addr, lightNode), correctBzzHandshake(addr, noLightNode),
&HandshakeMsg{Version: TestProtocolVersion, NetworkID: TestProtocolNetworkID, Addr: NewAddr(node)}, &HandshakeMsg{Version: TestProtocolVersion, NetworkID: TestProtocolNetworkID, Addr: NewAddr(node)},
) )