Fix issue with vesting logic

evmos · Mar 28, 2023 · d1b70eb · d1b70eb
1 parent 4d5d3fe
commit d1b70eb
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/x/bank/keeper/keeper.go b/x/bank/keeper/keeper.go
@@ -152,9 +152,17 @@ func (k BaseKeeper) DelegateCoins(ctx sdk.Context, delegatorAddr, moduleAccAddr
 		return sdkerrors.Wrap(sdkerrors.ErrInvalidCoins, amt.String())
 	}
 
+	spendableBalances := k.SpendableCoins(ctx, delegatorAddr)
 	balances := sdk.NewCoins()
 
 	for _, coin := range amt {
+		ok, spendableBalance := spendableBalances.Find(coin.GetDenom())
+		if !ok || spendableBalance.IsLT(coin) {
+			return sdkerrors.Wrapf(
+				sdkerrors.ErrInsufficientFunds, "failed to delegate; %s is smaller than %s", spendableBalance, amt,
+			)
+		}
+
 		balance := k.GetBalance(ctx, delegatorAddr, coin.GetDenom())
 		if balance.IsLT(coin) {
 			return sdkerrors.Wrapf(