diff options
| author | Sonic <sonic@cobinhood.com> | 2018-10-15 12:53:29 +0800 |
|---|---|---|
| committer | Wei-Ning Huang <w@dexon.org> | 2018-12-19 20:54:27 +0800 |
| commit | 3ecb633a4deb7324a0c0a3ad0d0dd98fccbe7e4e (patch) | |
| tree | f877cde0e4382cdf2cef8d4cf1986a93aa7991d1 | |
| parent | 5e32edda84a36b8675f58dbf87fb1c742df80d23 (diff) | |
| download | dexon-3ecb633a4deb7324a0c0a3ad0d0dd98fccbe7e4e.tar dexon-3ecb633a4deb7324a0c0a3ad0d0dd98fccbe7e4e.tar.gz dexon-3ecb633a4deb7324a0c0a3ad0d0dd98fccbe7e4e.tar.bz2 dexon-3ecb633a4deb7324a0c0a3ad0d0dd98fccbe7e4e.tar.lz dexon-3ecb633a4deb7324a0c0a3ad0d0dd98fccbe7e4e.tar.xz dexon-3ecb633a4deb7324a0c0a3ad0d0dd98fccbe7e4e.tar.zst dexon-3ecb633a4deb7324a0c0a3ad0d0dd98fccbe7e4e.zip | |
dex: fix flaky test
| -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 + } + } +} |