tmp fix

Author: hieblmi

fsm/fsm.go

@@ -154,6 +154,35 @@ func NewStateMachineWithState(states States, current StateType,
 	}
 }
 
+// GetNextStateType returns the next state type for the event given the
+// machine's current state, or an error if the event can't be handled in the
+// given state.
+func (s *StateMachine) GetNextStateType(event EventType) (StateType, error) {
+	var (
+		state State
+		ok    bool
+	)
+
+	stateMap := s.States
+
+	if state, ok = stateMap[s.current]; !ok {
+		return "", NewErrConfigError("current state not found")
+	}
+
+	if state.Transitions == nil {
+		return "", NewErrConfigError("current state has no transitions")
+	}
+
+	var next StateType
+	if next, ok = state.Transitions[event]; !ok {
+		return "", NewErrConfigError(
+			"event not found in current transitions",
+		)
+	}
+
+	return next, nil
+}
+
 // getNextState returns the next state for the event given the machine's current
 // state, or an error if the event can't be handled in the given state.
 func (s *StateMachine) getNextState(event EventType) (State, error) {

staticaddr/deposit/manager.go

@@ -196,9 +196,19 @@ func (m *Manager) recoverDeposits(ctx context.Context) error {
 		go func(fsm *FSM) {
 			err := fsm.SendEvent(ctx, OnRecover, nil)
 			if err != nil {
-				log.Errorf("Error sending OnStart event: %v",
+				log.Errorf("Error sending OnRecover event: %v",
 					err)
 			}
+
+			nextState, err := fsm.GetNextStateType(OnRecover)
+			if err != nil {
+				log.Errorf("Error retrieving next state: %v",
+					err)
+			} else {
+				fsm.DefaultObserver.WaitForStateAsync(
+					ctx, nextState, false,
+				)
+			}
 		}(fsm)
 
 		m.mu.Lock()