mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-06-14 09:51:10 +00:00
Run cargo fmt on the whole code base (#9394)
* Run cargo fmt on the whole code base * Second run * Add CI check * Fix compilation * More unnecessary braces * Handle weights * Use --all * Use correct attributes... * Fix UI tests * AHHHHHHHHH * 🤦 * Docs * Fix compilation * 🤷 * Please stop * 🤦 x 2 * More * make rustfmt.toml consistent with polkadot Co-authored-by: André Silva <andrerfosilva@gmail.com>
This commit is contained in:
Bastian Köcher
@@ -24,336 +24,312 @@ use crate::mock::{Balances, ExtBuilder, System, Test, Vesting};
|
||||
|
||||
#[test]
|
||||
fn check_vesting_status() {
|
||||
ExtBuilder::default()
|
||||
.existential_deposit(256)
|
||||
.build()
|
||||
.execute_with(|| {
|
||||
let user1_free_balance = Balances::free_balance(&1);
|
||||
let user2_free_balance = Balances::free_balance(&2);
|
||||
let user12_free_balance = Balances::free_balance(&12);
|
||||
assert_eq!(user1_free_balance, 256 * 10); // Account 1 has free balance
|
||||
assert_eq!(user2_free_balance, 256 * 20); // Account 2 has free balance
|
||||
assert_eq!(user12_free_balance, 256 * 10); // Account 12 has free balance
|
||||
let user1_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 128, // Vesting over 10 blocks
|
||||
starting_block: 0,
|
||||
};
|
||||
let user2_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 20,
|
||||
per_block: 256, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
let user12_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 64, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_eq!(Vesting::vesting(&1), Some(user1_vesting_schedule)); // Account 1 has a vesting schedule
|
||||
assert_eq!(Vesting::vesting(&2), Some(user2_vesting_schedule)); // Account 2 has a vesting schedule
|
||||
assert_eq!(Vesting::vesting(&12), Some(user12_vesting_schedule)); // Account 12 has a vesting schedule
|
||||
ExtBuilder::default().existential_deposit(256).build().execute_with(|| {
|
||||
let user1_free_balance = Balances::free_balance(&1);
|
||||
let user2_free_balance = Balances::free_balance(&2);
|
||||
let user12_free_balance = Balances::free_balance(&12);
|
||||
assert_eq!(user1_free_balance, 256 * 10); // Account 1 has free balance
|
||||
assert_eq!(user2_free_balance, 256 * 20); // Account 2 has free balance
|
||||
assert_eq!(user12_free_balance, 256 * 10); // Account 12 has free balance
|
||||
let user1_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 128, // Vesting over 10 blocks
|
||||
starting_block: 0,
|
||||
};
|
||||
let user2_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 20,
|
||||
per_block: 256, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
let user12_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 64, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_eq!(Vesting::vesting(&1), Some(user1_vesting_schedule)); // Account 1 has a vesting schedule
|
||||
assert_eq!(Vesting::vesting(&2), Some(user2_vesting_schedule)); // Account 2 has a vesting schedule
|
||||
assert_eq!(Vesting::vesting(&12), Some(user12_vesting_schedule)); // Account 12 has a vesting schedule
|
||||
|
||||
// Account 1 has only 128 units vested from their illiquid 256 * 5 units at block 1
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(128 * 9));
|
||||
// Account 2 has their full balance locked
|
||||
assert_eq!(Vesting::vesting_balance(&2), Some(user2_free_balance));
|
||||
// Account 12 has only their illiquid funds locked
|
||||
assert_eq!(Vesting::vesting_balance(&12), Some(user12_free_balance - 256 * 5));
|
||||
// Account 1 has only 128 units vested from their illiquid 256 * 5 units at block 1
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(128 * 9));
|
||||
// Account 2 has their full balance locked
|
||||
assert_eq!(Vesting::vesting_balance(&2), Some(user2_free_balance));
|
||||
// Account 12 has only their illiquid funds locked
|
||||
assert_eq!(Vesting::vesting_balance(&12), Some(user12_free_balance - 256 * 5));
|
||||
|
||||
System::set_block_number(10);
|
||||
assert_eq!(System::block_number(), 10);
|
||||
System::set_block_number(10);
|
||||
assert_eq!(System::block_number(), 10);
|
||||
|
||||
// Account 1 has fully vested by block 10
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(0));
|
||||
// Account 2 has started vesting by block 10
|
||||
assert_eq!(Vesting::vesting_balance(&2), Some(user2_free_balance));
|
||||
// Account 12 has started vesting by block 10
|
||||
assert_eq!(Vesting::vesting_balance(&12), Some(user12_free_balance - 256 * 5));
|
||||
// Account 1 has fully vested by block 10
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(0));
|
||||
// Account 2 has started vesting by block 10
|
||||
assert_eq!(Vesting::vesting_balance(&2), Some(user2_free_balance));
|
||||
// Account 12 has started vesting by block 10
|
||||
assert_eq!(Vesting::vesting_balance(&12), Some(user12_free_balance - 256 * 5));
|
||||
|
||||
System::set_block_number(30);
|
||||
assert_eq!(System::block_number(), 30);
|
||||
System::set_block_number(30);
|
||||
assert_eq!(System::block_number(), 30);
|
||||
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(0)); // Account 1 is still fully vested, and not negative
|
||||
assert_eq!(Vesting::vesting_balance(&2), Some(0)); // Account 2 has fully vested by block 30
|
||||
assert_eq!(Vesting::vesting_balance(&12), Some(0)); // Account 2 has fully vested by block 30
|
||||
|
||||
});
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(0)); // Account 1 is still fully vested, and not negative
|
||||
assert_eq!(Vesting::vesting_balance(&2), Some(0)); // Account 2 has fully vested by block 30
|
||||
assert_eq!(Vesting::vesting_balance(&12), Some(0)); // Account 2 has fully vested by block 30
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn unvested_balance_should_not_transfer() {
|
||||
ExtBuilder::default()
|
||||
.existential_deposit(10)
|
||||
.build()
|
||||
.execute_with(|| {
|
||||
let user1_free_balance = Balances::free_balance(&1);
|
||||
assert_eq!(user1_free_balance, 100); // Account 1 has free balance
|
||||
// Account 1 has only 5 units vested at block 1 (plus 50 unvested)
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(45));
|
||||
assert_noop!(
|
||||
Balances::transfer(Some(1).into(), 2, 56),
|
||||
pallet_balances::Error::<Test, _>::LiquidityRestrictions,
|
||||
); // Account 1 cannot send more than vested amount
|
||||
});
|
||||
ExtBuilder::default().existential_deposit(10).build().execute_with(|| {
|
||||
let user1_free_balance = Balances::free_balance(&1);
|
||||
assert_eq!(user1_free_balance, 100); // Account 1 has free balance
|
||||
// Account 1 has only 5 units vested at block 1 (plus 50 unvested)
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(45));
|
||||
assert_noop!(
|
||||
Balances::transfer(Some(1).into(), 2, 56),
|
||||
pallet_balances::Error::<Test, _>::LiquidityRestrictions,
|
||||
); // Account 1 cannot send more than vested amount
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn vested_balance_should_transfer() {
|
||||
ExtBuilder::default()
|
||||
.existential_deposit(10)
|
||||
.build()
|
||||
.execute_with(|| {
|
||||
let user1_free_balance = Balances::free_balance(&1);
|
||||
assert_eq!(user1_free_balance, 100); // Account 1 has free balance
|
||||
// Account 1 has only 5 units vested at block 1 (plus 50 unvested)
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(45));
|
||||
assert_ok!(Vesting::vest(Some(1).into()));
|
||||
assert_ok!(Balances::transfer(Some(1).into(), 2, 55));
|
||||
});
|
||||
ExtBuilder::default().existential_deposit(10).build().execute_with(|| {
|
||||
let user1_free_balance = Balances::free_balance(&1);
|
||||
assert_eq!(user1_free_balance, 100); // Account 1 has free balance
|
||||
// Account 1 has only 5 units vested at block 1 (plus 50 unvested)
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(45));
|
||||
assert_ok!(Vesting::vest(Some(1).into()));
|
||||
assert_ok!(Balances::transfer(Some(1).into(), 2, 55));
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn vested_balance_should_transfer_using_vest_other() {
|
||||
ExtBuilder::default()
|
||||
.existential_deposit(10)
|
||||
.build()
|
||||
.execute_with(|| {
|
||||
let user1_free_balance = Balances::free_balance(&1);
|
||||
assert_eq!(user1_free_balance, 100); // Account 1 has free balance
|
||||
// Account 1 has only 5 units vested at block 1 (plus 50 unvested)
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(45));
|
||||
assert_ok!(Vesting::vest_other(Some(2).into(), 1));
|
||||
assert_ok!(Balances::transfer(Some(1).into(), 2, 55));
|
||||
});
|
||||
ExtBuilder::default().existential_deposit(10).build().execute_with(|| {
|
||||
let user1_free_balance = Balances::free_balance(&1);
|
||||
assert_eq!(user1_free_balance, 100); // Account 1 has free balance
|
||||
// Account 1 has only 5 units vested at block 1 (plus 50 unvested)
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(45));
|
||||
assert_ok!(Vesting::vest_other(Some(2).into(), 1));
|
||||
assert_ok!(Balances::transfer(Some(1).into(), 2, 55));
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn extra_balance_should_transfer() {
|
||||
ExtBuilder::default()
|
||||
.existential_deposit(10)
|
||||
.build()
|
||||
.execute_with(|| {
|
||||
assert_ok!(Balances::transfer(Some(3).into(), 1, 100));
|
||||
assert_ok!(Balances::transfer(Some(3).into(), 2, 100));
|
||||
ExtBuilder::default().existential_deposit(10).build().execute_with(|| {
|
||||
assert_ok!(Balances::transfer(Some(3).into(), 1, 100));
|
||||
assert_ok!(Balances::transfer(Some(3).into(), 2, 100));
|
||||
|
||||
let user1_free_balance = Balances::free_balance(&1);
|
||||
assert_eq!(user1_free_balance, 200); // Account 1 has 100 more free balance than normal
|
||||
let user1_free_balance = Balances::free_balance(&1);
|
||||
assert_eq!(user1_free_balance, 200); // Account 1 has 100 more free balance than normal
|
||||
|
||||
let user2_free_balance = Balances::free_balance(&2);
|
||||
assert_eq!(user2_free_balance, 300); // Account 2 has 100 more free balance than normal
|
||||
let user2_free_balance = Balances::free_balance(&2);
|
||||
assert_eq!(user2_free_balance, 300); // Account 2 has 100 more free balance than normal
|
||||
|
||||
// Account 1 has only 5 units vested at block 1 (plus 150 unvested)
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(45));
|
||||
assert_ok!(Vesting::vest(Some(1).into()));
|
||||
assert_ok!(Balances::transfer(Some(1).into(), 3, 155)); // Account 1 can send extra units gained
|
||||
// Account 1 has only 5 units vested at block 1 (plus 150 unvested)
|
||||
assert_eq!(Vesting::vesting_balance(&1), Some(45));
|
||||
assert_ok!(Vesting::vest(Some(1).into()));
|
||||
assert_ok!(Balances::transfer(Some(1).into(), 3, 155)); // Account 1 can send extra units gained
|
||||
|
||||
// Account 2 has no units vested at block 1, but gained 100
|
||||
assert_eq!(Vesting::vesting_balance(&2), Some(200));
|
||||
assert_ok!(Vesting::vest(Some(2).into()));
|
||||
assert_ok!(Balances::transfer(Some(2).into(), 3, 100)); // Account 2 can send extra units gained
|
||||
});
|
||||
// Account 2 has no units vested at block 1, but gained 100
|
||||
assert_eq!(Vesting::vesting_balance(&2), Some(200));
|
||||
assert_ok!(Vesting::vest(Some(2).into()));
|
||||
assert_ok!(Balances::transfer(Some(2).into(), 3, 100)); // Account 2 can send extra units gained
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn liquid_funds_should_transfer_with_delayed_vesting() {
|
||||
ExtBuilder::default()
|
||||
.existential_deposit(256)
|
||||
.build()
|
||||
.execute_with(|| {
|
||||
let user12_free_balance = Balances::free_balance(&12);
|
||||
ExtBuilder::default().existential_deposit(256).build().execute_with(|| {
|
||||
let user12_free_balance = Balances::free_balance(&12);
|
||||
|
||||
assert_eq!(user12_free_balance, 2560); // Account 12 has free balance
|
||||
// Account 12 has liquid funds
|
||||
assert_eq!(Vesting::vesting_balance(&12), Some(user12_free_balance - 256 * 5));
|
||||
assert_eq!(user12_free_balance, 2560); // Account 12 has free balance
|
||||
// Account 12 has liquid funds
|
||||
assert_eq!(Vesting::vesting_balance(&12), Some(user12_free_balance - 256 * 5));
|
||||
|
||||
// Account 12 has delayed vesting
|
||||
let user12_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 64, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_eq!(Vesting::vesting(&12), Some(user12_vesting_schedule));
|
||||
// Account 12 has delayed vesting
|
||||
let user12_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 64, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_eq!(Vesting::vesting(&12), Some(user12_vesting_schedule));
|
||||
|
||||
// Account 12 can still send liquid funds
|
||||
assert_ok!(Balances::transfer(Some(12).into(), 3, 256 * 5));
|
||||
});
|
||||
// Account 12 can still send liquid funds
|
||||
assert_ok!(Balances::transfer(Some(12).into(), 3, 256 * 5));
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn vested_transfer_works() {
|
||||
ExtBuilder::default()
|
||||
.existential_deposit(256)
|
||||
.build()
|
||||
.execute_with(|| {
|
||||
let user3_free_balance = Balances::free_balance(&3);
|
||||
let user4_free_balance = Balances::free_balance(&4);
|
||||
assert_eq!(user3_free_balance, 256 * 30);
|
||||
assert_eq!(user4_free_balance, 256 * 40);
|
||||
// Account 4 should not have any vesting yet.
|
||||
assert_eq!(Vesting::vesting(&4), None);
|
||||
// Make the schedule for the new transfer.
|
||||
let new_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 64, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_ok!(Vesting::vested_transfer(Some(3).into(), 4, new_vesting_schedule));
|
||||
// Now account 4 should have vesting.
|
||||
assert_eq!(Vesting::vesting(&4), Some(new_vesting_schedule));
|
||||
// Ensure the transfer happened correctly.
|
||||
let user3_free_balance_updated = Balances::free_balance(&3);
|
||||
assert_eq!(user3_free_balance_updated, 256 * 25);
|
||||
let user4_free_balance_updated = Balances::free_balance(&4);
|
||||
assert_eq!(user4_free_balance_updated, 256 * 45);
|
||||
// Account 4 has 5 * 256 locked.
|
||||
assert_eq!(Vesting::vesting_balance(&4), Some(256 * 5));
|
||||
ExtBuilder::default().existential_deposit(256).build().execute_with(|| {
|
||||
let user3_free_balance = Balances::free_balance(&3);
|
||||
let user4_free_balance = Balances::free_balance(&4);
|
||||
assert_eq!(user3_free_balance, 256 * 30);
|
||||
assert_eq!(user4_free_balance, 256 * 40);
|
||||
// Account 4 should not have any vesting yet.
|
||||
assert_eq!(Vesting::vesting(&4), None);
|
||||
// Make the schedule for the new transfer.
|
||||
let new_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 64, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_ok!(Vesting::vested_transfer(Some(3).into(), 4, new_vesting_schedule));
|
||||
// Now account 4 should have vesting.
|
||||
assert_eq!(Vesting::vesting(&4), Some(new_vesting_schedule));
|
||||
// Ensure the transfer happened correctly.
|
||||
let user3_free_balance_updated = Balances::free_balance(&3);
|
||||
assert_eq!(user3_free_balance_updated, 256 * 25);
|
||||
let user4_free_balance_updated = Balances::free_balance(&4);
|
||||
assert_eq!(user4_free_balance_updated, 256 * 45);
|
||||
// Account 4 has 5 * 256 locked.
|
||||
assert_eq!(Vesting::vesting_balance(&4), Some(256 * 5));
|
||||
|
||||
System::set_block_number(20);
|
||||
assert_eq!(System::block_number(), 20);
|
||||
System::set_block_number(20);
|
||||
assert_eq!(System::block_number(), 20);
|
||||
|
||||
// Account 4 has 5 * 64 units vested by block 20.
|
||||
assert_eq!(Vesting::vesting_balance(&4), Some(10 * 64));
|
||||
// Account 4 has 5 * 64 units vested by block 20.
|
||||
assert_eq!(Vesting::vesting_balance(&4), Some(10 * 64));
|
||||
|
||||
System::set_block_number(30);
|
||||
assert_eq!(System::block_number(), 30);
|
||||
System::set_block_number(30);
|
||||
assert_eq!(System::block_number(), 30);
|
||||
|
||||
// Account 4 has fully vested.
|
||||
assert_eq!(Vesting::vesting_balance(&4), Some(0));
|
||||
});
|
||||
// Account 4 has fully vested.
|
||||
assert_eq!(Vesting::vesting_balance(&4), Some(0));
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn vested_transfer_correctly_fails() {
|
||||
ExtBuilder::default()
|
||||
.existential_deposit(256)
|
||||
.build()
|
||||
.execute_with(|| {
|
||||
let user2_free_balance = Balances::free_balance(&2);
|
||||
let user4_free_balance = Balances::free_balance(&4);
|
||||
assert_eq!(user2_free_balance, 256 * 20);
|
||||
assert_eq!(user4_free_balance, 256 * 40);
|
||||
// Account 2 should already have a vesting schedule.
|
||||
let user2_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 20,
|
||||
per_block: 256, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_eq!(Vesting::vesting(&2), Some(user2_vesting_schedule));
|
||||
ExtBuilder::default().existential_deposit(256).build().execute_with(|| {
|
||||
let user2_free_balance = Balances::free_balance(&2);
|
||||
let user4_free_balance = Balances::free_balance(&4);
|
||||
assert_eq!(user2_free_balance, 256 * 20);
|
||||
assert_eq!(user4_free_balance, 256 * 40);
|
||||
// Account 2 should already have a vesting schedule.
|
||||
let user2_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 20,
|
||||
per_block: 256, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_eq!(Vesting::vesting(&2), Some(user2_vesting_schedule));
|
||||
|
||||
// The vesting schedule we will try to create, fails due to pre-existence of schedule.
|
||||
let new_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 64, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_noop!(
|
||||
Vesting::vested_transfer(Some(4).into(), 2, new_vesting_schedule),
|
||||
Error::<Test>::ExistingVestingSchedule,
|
||||
);
|
||||
// The vesting schedule we will try to create, fails due to pre-existence of schedule.
|
||||
let new_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 64, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_noop!(
|
||||
Vesting::vested_transfer(Some(4).into(), 2, new_vesting_schedule),
|
||||
Error::<Test>::ExistingVestingSchedule,
|
||||
);
|
||||
|
||||
// Fails due to too low transfer amount.
|
||||
let new_vesting_schedule_too_low = VestingInfo {
|
||||
locked: 256 * 1,
|
||||
per_block: 64,
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_noop!(
|
||||
Vesting::vested_transfer(Some(3).into(), 4, new_vesting_schedule_too_low),
|
||||
Error::<Test>::AmountLow,
|
||||
);
|
||||
// Fails due to too low transfer amount.
|
||||
let new_vesting_schedule_too_low =
|
||||
VestingInfo { locked: 256 * 1, per_block: 64, starting_block: 10 };
|
||||
assert_noop!(
|
||||
Vesting::vested_transfer(Some(3).into(), 4, new_vesting_schedule_too_low),
|
||||
Error::<Test>::AmountLow,
|
||||
);
|
||||
|
||||
// Verify no currency transfer happened.
|
||||
assert_eq!(user2_free_balance, 256 * 20);
|
||||
assert_eq!(user4_free_balance, 256 * 40);
|
||||
});
|
||||
// Verify no currency transfer happened.
|
||||
assert_eq!(user2_free_balance, 256 * 20);
|
||||
assert_eq!(user4_free_balance, 256 * 40);
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn force_vested_transfer_works() {
|
||||
ExtBuilder::default()
|
||||
.existential_deposit(256)
|
||||
.build()
|
||||
.execute_with(|| {
|
||||
let user3_free_balance = Balances::free_balance(&3);
|
||||
let user4_free_balance = Balances::free_balance(&4);
|
||||
assert_eq!(user3_free_balance, 256 * 30);
|
||||
assert_eq!(user4_free_balance, 256 * 40);
|
||||
// Account 4 should not have any vesting yet.
|
||||
assert_eq!(Vesting::vesting(&4), None);
|
||||
// Make the schedule for the new transfer.
|
||||
let new_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 64, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_noop!(Vesting::force_vested_transfer(Some(4).into(), 3, 4, new_vesting_schedule), BadOrigin);
|
||||
assert_ok!(Vesting::force_vested_transfer(RawOrigin::Root.into(), 3, 4, new_vesting_schedule));
|
||||
// Now account 4 should have vesting.
|
||||
assert_eq!(Vesting::vesting(&4), Some(new_vesting_schedule));
|
||||
// Ensure the transfer happened correctly.
|
||||
let user3_free_balance_updated = Balances::free_balance(&3);
|
||||
assert_eq!(user3_free_balance_updated, 256 * 25);
|
||||
let user4_free_balance_updated = Balances::free_balance(&4);
|
||||
assert_eq!(user4_free_balance_updated, 256 * 45);
|
||||
// Account 4 has 5 * 256 locked.
|
||||
assert_eq!(Vesting::vesting_balance(&4), Some(256 * 5));
|
||||
ExtBuilder::default().existential_deposit(256).build().execute_with(|| {
|
||||
let user3_free_balance = Balances::free_balance(&3);
|
||||
let user4_free_balance = Balances::free_balance(&4);
|
||||
assert_eq!(user3_free_balance, 256 * 30);
|
||||
assert_eq!(user4_free_balance, 256 * 40);
|
||||
// Account 4 should not have any vesting yet.
|
||||
assert_eq!(Vesting::vesting(&4), None);
|
||||
// Make the schedule for the new transfer.
|
||||
let new_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 64, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_noop!(
|
||||
Vesting::force_vested_transfer(Some(4).into(), 3, 4, new_vesting_schedule),
|
||||
BadOrigin
|
||||
);
|
||||
assert_ok!(Vesting::force_vested_transfer(
|
||||
RawOrigin::Root.into(),
|
||||
3,
|
||||
4,
|
||||
new_vesting_schedule
|
||||
));
|
||||
// Now account 4 should have vesting.
|
||||
assert_eq!(Vesting::vesting(&4), Some(new_vesting_schedule));
|
||||
// Ensure the transfer happened correctly.
|
||||
let user3_free_balance_updated = Balances::free_balance(&3);
|
||||
assert_eq!(user3_free_balance_updated, 256 * 25);
|
||||
let user4_free_balance_updated = Balances::free_balance(&4);
|
||||
assert_eq!(user4_free_balance_updated, 256 * 45);
|
||||
// Account 4 has 5 * 256 locked.
|
||||
assert_eq!(Vesting::vesting_balance(&4), Some(256 * 5));
|
||||
|
||||
System::set_block_number(20);
|
||||
assert_eq!(System::block_number(), 20);
|
||||
System::set_block_number(20);
|
||||
assert_eq!(System::block_number(), 20);
|
||||
|
||||
// Account 4 has 5 * 64 units vested by block 20.
|
||||
assert_eq!(Vesting::vesting_balance(&4), Some(10 * 64));
|
||||
// Account 4 has 5 * 64 units vested by block 20.
|
||||
assert_eq!(Vesting::vesting_balance(&4), Some(10 * 64));
|
||||
|
||||
System::set_block_number(30);
|
||||
assert_eq!(System::block_number(), 30);
|
||||
System::set_block_number(30);
|
||||
assert_eq!(System::block_number(), 30);
|
||||
|
||||
// Account 4 has fully vested.
|
||||
assert_eq!(Vesting::vesting_balance(&4), Some(0));
|
||||
});
|
||||
// Account 4 has fully vested.
|
||||
assert_eq!(Vesting::vesting_balance(&4), Some(0));
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn force_vested_transfer_correctly_fails() {
|
||||
ExtBuilder::default()
|
||||
.existential_deposit(256)
|
||||
.build()
|
||||
.execute_with(|| {
|
||||
let user2_free_balance = Balances::free_balance(&2);
|
||||
let user4_free_balance = Balances::free_balance(&4);
|
||||
assert_eq!(user2_free_balance, 256 * 20);
|
||||
assert_eq!(user4_free_balance, 256 * 40);
|
||||
// Account 2 should already have a vesting schedule.
|
||||
let user2_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 20,
|
||||
per_block: 256, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_eq!(Vesting::vesting(&2), Some(user2_vesting_schedule));
|
||||
ExtBuilder::default().existential_deposit(256).build().execute_with(|| {
|
||||
let user2_free_balance = Balances::free_balance(&2);
|
||||
let user4_free_balance = Balances::free_balance(&4);
|
||||
assert_eq!(user2_free_balance, 256 * 20);
|
||||
assert_eq!(user4_free_balance, 256 * 40);
|
||||
// Account 2 should already have a vesting schedule.
|
||||
let user2_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 20,
|
||||
per_block: 256, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_eq!(Vesting::vesting(&2), Some(user2_vesting_schedule));
|
||||
|
||||
// The vesting schedule we will try to create, fails due to pre-existence of schedule.
|
||||
let new_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 64, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_noop!(
|
||||
Vesting::force_vested_transfer(RawOrigin::Root.into(), 4, 2, new_vesting_schedule),
|
||||
Error::<Test>::ExistingVestingSchedule,
|
||||
);
|
||||
// The vesting schedule we will try to create, fails due to pre-existence of schedule.
|
||||
let new_vesting_schedule = VestingInfo {
|
||||
locked: 256 * 5,
|
||||
per_block: 64, // Vesting over 20 blocks
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_noop!(
|
||||
Vesting::force_vested_transfer(RawOrigin::Root.into(), 4, 2, new_vesting_schedule),
|
||||
Error::<Test>::ExistingVestingSchedule,
|
||||
);
|
||||
|
||||
// Fails due to too low transfer amount.
|
||||
let new_vesting_schedule_too_low = VestingInfo {
|
||||
locked: 256 * 1,
|
||||
per_block: 64,
|
||||
starting_block: 10,
|
||||
};
|
||||
assert_noop!(
|
||||
Vesting::force_vested_transfer(RawOrigin::Root.into(), 3, 4, new_vesting_schedule_too_low),
|
||||
Error::<Test>::AmountLow,
|
||||
);
|
||||
// Fails due to too low transfer amount.
|
||||
let new_vesting_schedule_too_low =
|
||||
VestingInfo { locked: 256 * 1, per_block: 64, starting_block: 10 };
|
||||
assert_noop!(
|
||||
Vesting::force_vested_transfer(
|
||||
RawOrigin::Root.into(),
|
||||
3,
|
||||
4,
|
||||
new_vesting_schedule_too_low
|
||||
),
|
||||
Error::<Test>::AmountLow,
|
||||
);
|
||||
|
||||
// Verify no currency transfer happened.
|
||||
assert_eq!(user2_free_balance, 256 * 20);
|
||||
assert_eq!(user4_free_balance, 256 * 40);
|
||||
});
|
||||
// Verify no currency transfer happened.
|
||||
assert_eq!(user2_free_balance, 256 * 20);
|
||||
assert_eq!(user4_free_balance, 256 * 40);
|
||||
});
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user