fix: use updateRoundStep to track consensus state changes (#1464)

We're currently using `newStep` to indicate whether to write a new
trace, however this actually doesn't track when the round and step
change happens and thus misses certain steps (like NewRound). This PR
fixes it

consensus/state.go

@@ -538,6 +538,7 @@ func (cs *State) updateRoundStep(round int32, step cstypes.RoundStepType) {
 		}
 		if cs.Step != step {
 			cs.metrics.MarkStep(cs.Step)
+			schema.WriteRoundState(cs.traceClient, cs.Height, round, uint8(step))
 		}
 	}
 	cs.Round = round
@@ -706,10 +707,6 @@ func (cs *State) newStep() {
 
 	cs.nSteps++
 
-	step := uint8(cs.RoundState.Step)
-
-	schema.WriteRoundState(cs.traceClient, cs.Height, cs.Round, step)
-
 	// newStep is called by updateToState in NewState before the eventBus is set!
 	if cs.eventBus != nil {
 		if err := cs.eventBus.PublishEventNewRoundStep(rs); err != nil {
@@ -1518,6 +1515,7 @@ func (cs *State) enterPrecommitWait(height int64, round int32) {
 	defer func() {
 		// Done enterPrecommitWait:
 		cs.TriggeredTimeoutPrecommit = true
+		cs.updateRoundStep(round, cstypes.RoundStepPrecommitWait)
 		cs.newStep()
 	}()