fluencelabs/tendermint
1
0
Fork 0
mirror of https://github.com/fluencelabs/tendermint synced 2025-05-30 14:41:20 +00:00
Code Issues Projects Releases Wiki Activity

p2p/pex: wait to connect to all peers in reactor test

Browse Source
This commit is contained in:
Ethan Buchman 2018-01-23 21:30:53 -05:00
parent 21ce5856b3
commit 775bb85efb
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
p2p/pex/pex_reactor_test.go
View File

@ -111,7 +111,7 @@ func TestPEXReactorRunning(t *testing.T) {
require.Nil(t, err) require.Nil(t, err)
} }
assertSomePeersWithTimeout(t, switches, 10*time.Millisecond, 10*time.Second) assertPeersWithTimeout(t, switches, 10*time.Millisecond, 10*time.Second, N-1)
// stop them // stop them
for _, s := range switches { for _, s := range switches {
@ -119,7 +119,7 @@ func TestPEXReactorRunning(t *testing.T) {
} }
} }
func assertSomePeersWithTimeout(t *testing.T, switches []*p2p.Switch, checkPeriod, timeout time.Duration) { func assertPeersWithTimeout(t *testing.T, switches []*p2p.Switch, checkPeriod, timeout time.Duration, nPeers int) {
ticker := time.NewTicker(checkPeriod) ticker := time.NewTicker(checkPeriod)
remaining := timeout remaining := timeout
for { for {
@ -129,7 +129,7 @@ func assertSomePeersWithTimeout(t *testing.T, switches []*p2p.Switch, checkPerio
allGood := true allGood := true
for _, s := range switches { for _, s := range switches {
outbound, inbound, _ := s.NumPeers() outbound, inbound, _ := s.NumPeers()
if outbound+inbound == 0 { if outbound+inbound < nPeers {
allGood = false allGood = false
} }
} }
@ -296,7 +296,7 @@ func TestPEXReactorUsesSeedsIfNeeded(t *testing.T) {
defer sw.Stop() defer sw.Stop()
// 3. check that peer at least connects to seed // 3. check that peer at least connects to seed
assertSomePeersWithTimeout(t, []*p2p.Switch{sw}, 10*time.Millisecond, 10*time.Second) assertPeersWithTimeout(t, []*p2p.Switch{sw}, 10*time.Millisecond, 10*time.Second, 1)
} }
func TestPEXReactorCrawlStatus(t *testing.T) { func TestPEXReactorCrawlStatus(t *testing.T) {