fungible::Unbalanced::decrease_balance: Handle precision properly (#2823)

This commit is contained in:
Bastian Köcher
2024-01-05 19:55:32 +01:00
committed by GitHub
parent 19de1c9660
commit 1c95310a66
3 changed files with 46 additions and 3 deletions
+11
View File
@@ -0,0 +1,11 @@
title: "`fungible::Unbalanced::decrease_balance`: Handle `precision` properly"
doc:
- audience: Runtime Dev
description: |
`fungible::Unbalanced::decrease_balance` will now handle `precision` properly. This means when
passing `Exact`, it will ensure that the available balance is bigger or equal to the `amount`
that should be deducted.
crates:
- name: "frame-support"
@@ -146,7 +146,7 @@ fn unbalanced_trait_decrease_balance_works_2() {
assert_eq!(Balances::total_balance_on_hold(&1337), 60);
assert_noop!(
Balances::decrease_balance(&1337, 40, Exact, Expendable, Polite),
Error::<Test>::InsufficientBalance
TokenError::FundsUnavailable
);
assert_eq!(Balances::decrease_balance(&1337, 39, Exact, Expendable, Polite), Ok(39));
assert_eq!(<Balances as fungible::Inspect<_>>::balance(&1337), 1);
@@ -468,3 +468,28 @@ fn emit_events_with_changing_freezes() {
assert_eq!(events(), [RuntimeEvent::Balances(crate::Event::Thawed { who: 1, amount: 15 })]);
});
}
#[test]
fn withdraw_precision_exact_works() {
ExtBuilder::default()
.existential_deposit(1)
.monied(true)
.build_and_execute_with(|| {
assert_ok!(Balances::set_freeze(&TestId::Foo, &1, 10));
assert_eq!(Balances::account(&1).free, 10);
assert_eq!(Balances::account(&1).frozen, 10);
// `BestEffort` will not reduce anything
assert_ok!(<Balances as fungible::Balanced<_>>::withdraw(
&1, 5, BestEffort, Preserve, Polite
));
assert_eq!(Balances::account(&1).free, 10);
assert_eq!(Balances::account(&1).frozen, 10);
assert_noop!(
<Balances as fungible::Balanced<_>>::withdraw(&1, 5, Exact, Preserve, Polite),
TokenError::FundsUnavailable
);
});
}
@@ -181,9 +181,16 @@ pub trait Unbalanced<AccountId>: Inspect<AccountId> {
) -> Result<Self::Balance, DispatchError> {
let old_balance = Self::balance(who);
let free = Self::reducible_balance(who, preservation, force);
if let BestEffort = precision {
amount = amount.min(free);
match precision {
BestEffort => {
amount = amount.min(free);
},
Exact =>
if free < amount {
return Err(TokenError::FundsUnavailable.into())
},
}
let new_balance = old_balance.checked_sub(&amount).ok_or(TokenError::FundsUnavailable)?;
if let Some(dust) = Self::write_balance(who, new_balance)? {
Self::handle_dust(Dust(dust));