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

swarm/network: Changed LightNode handhsake test to table driven

Browse source
This commit is contained in:
Attila Gazso 2018-07-31 12:15:52 +02:00
parent 975e96ceb1
commit 5270b80752
1 changed files with 24 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
swarm/network/protocol_test.go
View file

@ -255,44 +255,34 @@ func TestBzzHandshakeSuccess(t *testing.T) {
} }
} }
func TestBzzHandshakeLightNodeOff(t *testing.T) { var lightNodeTests = []struct {
name string
lightNode bool
}{
{"on", true},
{"off", false},
}
func TestBzzHandshakeLightNode(t *testing.T) {
for _, test := range lightNodeTests {
t.Run(test.name, func(t *testing.T) {
randomAddr := RandomAddr() randomAddr := RandomAddr()
pt := newBzzHandshakeTester(t, 1, randomAddr, false) pt := newBzzHandshakeTester(t, 1, randomAddr, false)
id := pt.IDs[0] id := pt.IDs[0]
addr := NewAddrFromNodeID(id) addr := NewAddrFromNodeID(id)
peerLightNode := false
err := pt.testHandshake( err := pt.testHandshake(
correctBzzHandshake(randomAddr, false), correctBzzHandshake(randomAddr, false),
&HandshakeMsg{Version: TestProtocolVersion, NetworkID: TestProtocolNetworkID, Addr: addr, LightNode: peerLightNode}, &HandshakeMsg{Version: TestProtocolVersion, NetworkID: TestProtocolNetworkID, Addr: addr, LightNode: test.lightNode},
) )
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
if pt.bzz.handshakes[id].LightNode != peerLightNode { if pt.bzz.handshakes[id].LightNode != test.lightNode {
t.Fatalf("peer LightNode flag is %v, should be %v", pt.bzz.handshakes[id].LightNode, peerLightNode) t.Fatalf("peer LightNode flag is %v, should be %v", pt.bzz.handshakes[id].LightNode, test.lightNode)
} }
} })
func TestBzzHandshakeLightNodeOn(t *testing.T) {
randomAddr := RandomAddr()
pt := newBzzHandshakeTester(t, 1, randomAddr, false)
id := pt.IDs[0]
addr := NewAddrFromNodeID(id)
peerLightNode := true
err := pt.testHandshake(
correctBzzHandshake(randomAddr, false),
&HandshakeMsg{Version: TestProtocolVersion, NetworkID: TestProtocolNetworkID, Addr: addr, LightNode: peerLightNode},
)
if err != nil {
t.Fatal(err)
}
if pt.bzz.handshakes[id].LightNode != peerLightNode {
t.Fatalf("peer LightNode flag is %v, should be %v", pt.bzz.handshakes[id].LightNode, peerLightNode)
} }
} }