|
|
@ -614,13 +614,16 @@ func testBroadcastBlock(t *testing.T, totalPeers, broadcastExpected int) { |
|
|
|
select { |
|
|
|
select { |
|
|
|
case <-doneCh: |
|
|
|
case <-doneCh: |
|
|
|
received++ |
|
|
|
received++ |
|
|
|
|
|
|
|
if received > broadcastExpected { |
|
|
|
case <-time.After(time.Second): |
|
|
|
// We can bail early here
|
|
|
|
|
|
|
|
t.Errorf("broadcast count mismatch: have %d > want %d", received, broadcastExpected) |
|
|
|
|
|
|
|
return |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
case <-time.After(2 * time.Second): |
|
|
|
if received != broadcastExpected { |
|
|
|
if received != broadcastExpected { |
|
|
|
t.Errorf("broadcast count mismatch: have %d, want %d", received, broadcastExpected) |
|
|
|
t.Errorf("broadcast count mismatch: have %d, want %d", received, broadcastExpected) |
|
|
|
} |
|
|
|
} |
|
|
|
return |
|
|
|
return |
|
|
|
|
|
|
|
|
|
|
|
case err = <-errCh: |
|
|
|
case err = <-errCh: |
|
|
|
t.Fatalf("broadcast failed: %v", err) |
|
|
|
t.Fatalf("broadcast failed: %v", err) |
|
|
|
} |
|
|
|
} |
|
|
|