Browse Source

change MakeConnectedSwitches to not connect to itself

and a test for it
pull/812/head
Anton Kaliaev 7 years ago
parent
commit
7869e541f6
No known key found for this signature in database GPG Key ID: 7B6881D965918214
2 changed files with 19 additions and 2 deletions
  1. +4
    -2
      p2p/switch.go
  2. +15
    -0
      p2p/switch_test.go

+ 4
- 2
p2p/switch.go View File

@ -509,8 +509,10 @@ func MakeConnectedSwitches(cfg *cfg.P2PConfig, n int, initSwitch func(int, *Swit
panic(err)
}
for i := 1; i < n; i++ {
connect(switches, 0, i)
for i := 0; i < n; i++ {
for j := i + 1; j < n; j++ {
connect(switches, i, j)
}
}
return switches


+ 15
- 0
p2p/switch_test.go View File

@ -286,6 +286,21 @@ func TestSwitchReconnectsToPersistentPeer(t *testing.T) {
assert.False(peer.IsRunning())
}
func TestSwitchFullConnectivity(t *testing.T) {
switches := MakeConnectedSwitches(config, 3, initSwitchFunc, Connect2Switches)
defer func() {
for _, sw := range switches {
sw.Stop()
}
}()
for i, sw := range switches {
if sw.Peers().Size() != 2 {
t.Fatalf("Expected each switch to be connected to 2 other, but %d switch only connected to %d", sw.Peers().Size(), i)
}
}
}
func BenchmarkSwitches(b *testing.B) {
b.StopTimer()


Loading…
Cancel
Save