Improve simulation handling for tax transactions (#504)

classic-terra · Jul 30, 2024 · c8c04d9 · c8c04d9
1 parent 51a080f
commit c8c04d9
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/custom/auth/ante/fee.go b/custom/auth/ante/fee.go
@@ -99,19 +99,24 @@ func (fd FeeDecorator) checkDeductFee(ctx sdk.Context, feeTx sdk.FeeTx, taxes sd
 		return sdkerrors.ErrUnknownAddress.Wrapf("fee payer address: %s does not exist", deductFeesFrom)
 	}
 
+	feesOrTax := fee
+
 	// deduct the fees
-	if !fee.IsZero() {
-		err := DeductFees(fd.bankKeeper, ctx, deductFeesFromAcc, fee)
+	if fee.IsZero() && simulate {
+		feesOrTax = taxes
+	}
+
+	if !feesOrTax.IsZero() {
+		err := DeductFees(fd.bankKeeper, ctx, deductFeesFromAcc, feesOrTax)
 		if err != nil {
 			return err
 		}
 
-		if !taxes.IsZero() && !simulate {
+		if !taxes.IsZero() {
 			err := fd.BurnTaxSplit(ctx, taxes)
 			if err != nil {
 				return err
 			}
-
 			// Record tax proceeds
 			fd.treasuryKeeper.RecordEpochTaxProceeds(ctx, taxes)
 		}