diff --git a/internal/consensus/state_test.go b/internal/consensus/state_test.go index be1522cbf..fd01dad35 100644 --- a/internal/consensus/state_test.go +++ b/internal/consensus/state_test.go @@ -2353,12 +2353,11 @@ func TestVoteExtensionRequiredHeight(t *testing.T) { if testCase.expectSuccessfulRound { ensurePrecommit(t, voteCh, height, round) height++ + ensureNewRound(t, newRoundCh, height, round) } else { - ensureNewTimeout(t, timeoutCh, height, round, cs1.state.ConsensusParams.Timeout.VoteTimeout(round).Nanoseconds()) - round++ + ensureNoNewTimeout(t, timeoutCh, cs1.state.ConsensusParams.Timeout.VoteTimeout(round).Nanoseconds()) } - ensureNewRound(t, newRoundCh, height, round) m.AssertExpectations(t) }) }