diff options
| author | Sonic <sonic@cobinhood.com> | 2018-10-15 12:53:29 +0800 |
|---|---|---|
| committer | Wei-Ning Huang <w@dexon.org> | 2019-03-12 12:19:09 +0800 |
| commit | b2ea7ebaf4b10540ea7c1da06dae605eb6e36f84 (patch) | |
| tree | 29e2795debd1e5796d1eb70e95328ca8056483ce /dex/protocol_test.go | |
| parent | 4a1fdfc4ad4c52764d94f1c27a9cf18e36d04385 (diff) | |
| download | dexon-b2ea7ebaf4b10540ea7c1da06dae605eb6e36f84.tar dexon-b2ea7ebaf4b10540ea7c1da06dae605eb6e36f84.tar.gz dexon-b2ea7ebaf4b10540ea7c1da06dae605eb6e36f84.tar.bz2 dexon-b2ea7ebaf4b10540ea7c1da06dae605eb6e36f84.tar.lz dexon-b2ea7ebaf4b10540ea7c1da06dae605eb6e36f84.tar.xz dexon-b2ea7ebaf4b10540ea7c1da06dae605eb6e36f84.tar.zst dexon-b2ea7ebaf4b10540ea7c1da06dae605eb6e36f84.zip | |
dex: fix flaky test
Diffstat (limited to 'dex/protocol_test.go')
| -rw-r--r-- | dex/protocol_test.go | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/dex/protocol_test.go b/dex/protocol_test.go index a26a40feb..4bb3dc9e8 100644 --- a/dex/protocol_test.go +++ b/dex/protocol_test.go @@ -406,6 +406,7 @@ func TestSendLatticeBlock(t *testing.T) { }, } + waitForRegister(pm, 1) pm.BroadcastLatticeBlock(&block) msg, err := p.app.ReadMsg() if err != nil { @@ -552,6 +553,7 @@ func TestSendVote(t *testing.T) { wg.Add(1) go checkvote(p, tt.isReceiver) } + waitForRegister(pm, len(testPeers)) pm.BroadcastVote(&vote) wg.Wait() } @@ -725,6 +727,7 @@ func TestSendAgreement(t *testing.T) { Votes: []coreTypes.Vote{vote}, } + waitForRegister(pm, 1) pm.BroadcastAgreementResult(&agreement) msg, err := p.app.ReadMsg() if err != nil { @@ -785,6 +788,7 @@ func TestSendRandomness(t *testing.T) { Randomness: []byte{7, 7, 7, 7}, } + waitForRegister(pm, 1) pm.BroadcastRandomnessResult(&randomness) msg, err := p.app.ReadMsg() if err != nil { @@ -802,3 +806,11 @@ func TestSendRandomness(t *testing.T) { t.Errorf("agreement mismatch") } } + +func waitForRegister(pm *ProtocolManager, num int) { + for { + if pm.peers.Len() >= num { + return + } + } +} |