staticaddr: wait until all deposits recovered

Author: hieblmi

staticaddr/deposit/manager.go

@@ -177,6 +177,7 @@ func (m *Manager) recoverDeposits(ctx context.Context) error {
 		return err
 	}
 
+	var wg sync.WaitGroup
 	for i, d := range deposits {
 		m.deposits[d.OutPoint] = deposits[i]
 
@@ -196,7 +197,9 @@ func (m *Manager) recoverDeposits(ctx context.Context) error {
 		}
 
 		// Send the OnRecover event to the state machine.
+		wg.Add(1)
 		go func(fsm *FSM) {
+			defer wg.Done()
 			err := fsm.SendEvent(ctx, OnRecover, nil)
 			if err != nil {
 				log.Errorf("Error sending OnStart event: %v",
@@ -208,6 +211,7 @@ func (m *Manager) recoverDeposits(ctx context.Context) error {
 		m.activeDeposits[d.OutPoint] = fsm
 		m.mu.Unlock()
 	}
+	wg.Wait()
 
 	return nil
 }