From 0f25a4f13199ffc27cc4fba733d19a2b2e860d69 Mon Sep 17 00:00:00 2001 From: William Banfield Date: Thu, 23 Sep 2021 12:06:10 -0400 Subject: [PATCH] add ensureRelock to common_test --- internal/consensus/common_test.go | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/internal/consensus/common_test.go b/internal/consensus/common_test.go index 20afbfa3a..9f8ffe9d5 100644 --- a/internal/consensus/common_test.go +++ b/internal/consensus/common_test.go @@ -647,10 +647,17 @@ func ensureNewUnlock(t *testing.T, unlockCh <-chan tmpubsub.Message, height int6 } func ensureLock(t *testing.T, lockCh <-chan tmpubsub.Message, height int64, round int32) { + t.Helper() ensureNewEvent(t, lockCh, height, round, ensureTimeout, "Timeout expired while waiting for LockValue event") } +func ensureRelock(t *testing.T, relockCh <-chan tmpubsub.Message, height int64, round int32) { + t.Helper() + ensureNewEvent(t, relockCh, height, round, ensureTimeout, + "Timeout expired while waiting for RelockValue event") +} + func ensureProposal(t *testing.T, proposalCh <-chan tmpubsub.Message, height int64, round int32, propID types.BlockID) { t.Helper() select {