difftreelog
refac(app-promo): impl for `unstake_all` & `unstake_partial` extrinsics
in: master
Added benchmark for `unstake_partial` extrinsic
7 files changed
Cargo.lockdiffbeforeafterboth--- a/Cargo.lock
+++ b/Cargo.lock
@@ -5782,7 +5782,7 @@
[[package]]
name = "pallet-app-promotion"
-version = "0.1.4"
+version = "0.1.5"
dependencies = [
"frame-benchmarking",
"frame-support",
pallets/app-promotion/CHANGELOG.mddiffbeforeafterboth--- a/pallets/app-promotion/CHANGELOG.md
+++ b/pallets/app-promotion/CHANGELOG.md
@@ -4,6 +4,12 @@
<!-- bureaucrate goes here -->
+## [0.1.5] - 2023-02-14
+
+### Added
+
+- `unstake_partial` extrinsic.
+
## [0.1.4] - 2023-01-31
### Changed
pallets/app-promotion/Cargo.tomldiffbeforeafterboth--- a/pallets/app-promotion/Cargo.toml
+++ b/pallets/app-promotion/Cargo.toml
@@ -9,7 +9,7 @@
license = 'GPLv3'
name = 'pallet-app-promotion'
repository = 'https://github.com/UniqueNetwork/unique-chain'
-version = '0.1.4'
+version = '0.1.5'
[package.metadata.docs.rs]
targets = ['x86_64-unknown-linux-gnu']
pallets/app-promotion/src/benchmarking.rsdiffbeforeafterboth--- a/pallets/app-promotion/src/benchmarking.rs
+++ b/pallets/app-promotion/src/benchmarking.rs
@@ -65,7 +65,7 @@
let staker = account::<T::AccountId>("staker", index, SEED);
<T as Config>::Currency::make_free_balance_be(&staker, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());
PromototionPallet::<T>::stake(RawOrigin::Signed(staker.clone()).into(), Into::<BalanceOf<T>>::into(100u128) * T::Nominal::get())?;
- PromototionPallet::<T>::unstake(RawOrigin::Signed(staker.clone()).into()).map_err(|e| e.error)?;
+ PromototionPallet::<T>::unstake_all(RawOrigin::Signed(staker.clone()).into()).map_err(|e| e.error)?;
Result::<(), sp_runtime::DispatchError>::Ok(())
})?;
let block_number = <frame_system::Pallet<T>>::current_block_number() + T::PendingInterval::get();
@@ -115,7 +115,7 @@
let _ = <T as Config>::Currency::make_free_balance_be(&caller, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());
} : _(RawOrigin::Signed(caller.clone()), share * <T as Config>::Currency::total_balance(&caller))
- unstake {
+ unstake_all {
let caller = account::<T::AccountId>("caller", 0, SEED);
let share = Perbill::from_rational(1u32, 20);
let _ = <T as Config>::Currency::make_free_balance_be(&caller, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());
@@ -130,6 +130,21 @@
} : _(RawOrigin::Signed(caller.clone()))
+ unstake_partial {
+ let caller = account::<T::AccountId>("caller", 0, SEED);
+ let share = Perbill::from_rational(1u32, 20);
+ let _ = <T as Config>::Currency::make_free_balance_be(&caller, Perbill::from_rational(1u32, 2) * BalanceOf::<T>::max_value());
+ (1..11).map(|i| {
+ // used to change block number
+ <frame_system::Pallet<T>>::set_block_number(i.into());
+ T::RelayBlockNumberProvider::set_block_number((2*i).into());
+ assert_eq!(<frame_system::Pallet<T>>::block_number(), i.into());
+ assert_eq!(T::RelayBlockNumberProvider::current_block_number(), (2*i).into());
+ PromototionPallet::<T>::stake(RawOrigin::Signed(caller.clone()).into(), Into::<BalanceOf<T>>::into(100u128) * T::Nominal::get())
+ }).collect::<Result<Vec<_>, _>>()?;
+
+ } : _(RawOrigin::Signed(caller.clone()), Into::<BalanceOf<T>>::into(1000u128) * T::Nominal::get())
+
sponsor_collection {
let pallet_admin = account::<T::AccountId>("admin", 0, SEED);
PromototionPallet::<T>::set_admin_address(RawOrigin::Root.into(), T::CrossAccountId::from_sub(pallet_admin.clone()))?;
pallets/app-promotion/src/lib.rsdiffbeforeafterboth--- a/pallets/app-promotion/src/lib.rs
+++ b/pallets/app-promotion/src/lib.rs
@@ -208,6 +208,8 @@
SponsorNotSet,
/// Errors caused by incorrect actions with a locked balance.
IncorrectLockedBalanceOperation,
+ /// Errors caused by insufficient staked balance.
+ InsufficientStakedBalance,
}
/// Stores the total staked amount.
@@ -489,65 +491,30 @@
}
/// Unstakes all stakes.
- /// Moves the sum of all stakes to the `reserved` state.
/// After the end of `PendingInterval` this sum becomes completely
/// free for further use.
#[pallet::call_index(2)]
- #[pallet::weight(<T as Config>::WeightInfo::unstake())]
- pub fn unstake_all(staker: OriginFor<T>) -> DispatchResultWithPostInfo {
+ #[pallet::weight(<T as Config>::WeightInfo::unstake_all())]
+ pub fn unstake_all(staker: OriginFor<T>) -> DispatchResult {
let staker_id = ensure_signed(staker)?;
- let config = <PalletConfiguration<T>>::get();
-
- // calculate block number where the sum would be free
- let block = <frame_system::Pallet<T>>::block_number() + config.pending_interval;
-
- let mut pendings = <PendingUnstake<T>>::get(block);
-
- // checks that we can do unreserve stakes in the block
- ensure!(!pendings.is_full(), Error::<T>::PendingForBlockOverflow);
-
- let mut total_stakes = 0u64;
-
- let total_staked: BalanceOf<T> = Staked::<T>::drain_prefix((&staker_id,))
- .map(|(_, (amount, _))| {
- total_stakes += 1;
- amount
- })
- .sum();
-
- if total_staked.is_zero() {
- return Ok(None::<Weight>.into()); // TO-DO
- }
-
- pendings
- .try_push((staker_id.clone(), total_staked))
- .map_err(|_| Error::<T>::PendingForBlockOverflow)?;
- <PendingUnstake<T>>::insert(block, pendings);
-
- TotalStaked::<T>::set(
- TotalStaked::<T>::get()
- .checked_sub(&total_staked)
- .ok_or(ArithmeticError::Underflow)?,
- );
-
- StakesPerAccount::<T>::remove(&staker_id);
-
- Self::deposit_event(Event::Unstake(staker_id, total_staked));
-
- Ok(None::<Weight>.into())
+ Self::unstake_all_internal(staker_id)
}
- /// Unstakes all stakes.
- /// Moves the sum of all stakes to the `reserved` state.
+ /// Unstakes the amount of balance for the staker.
/// After the end of `PendingInterval` this sum becomes completely
/// free for further use.
+ ///
+ /// # Arguments
+ ///
+ /// * `staker`: staker account.
+ /// * `amount`: amount of unstaked funds.
#[pallet::call_index(8)]
- #[pallet::weight(<T as Config>::WeightInfo::unstake())]
+ #[pallet::weight(<T as Config>::WeightInfo::unstake_partial())]
pub fn unstake_partial(staker: OriginFor<T>, amount: BalanceOf<T>) -> DispatchResult {
let staker_id = ensure_signed(staker)?;
- Self::partial_unstake(&staker_id, amount)
+ Self::unstake_partial_internal(staker_id, amount)
}
/// Sets the pallet to be the sponsor for the collection.
@@ -821,12 +788,18 @@
T::PalletId::get().into_account_truncating()
}
- fn partial_unstake(staker_id: &T::AccountId, unstaked_balance: BalanceOf<T>) -> DispatchResult {
-
+ /// Unstakes the balance for the staker.
+ ///
+ /// - `staker`: staker account.
+ /// - `amount`: amount of unstaked funds.
+ fn unstake_partial_internal(
+ staker_id: T::AccountId,
+ unstaked_balance: BalanceOf<T>,
+ ) -> DispatchResult {
if unstaked_balance == Default::default() {
return Ok(());
}
-
+
let config = <PalletConfiguration<T>>::get();
// calculate block number where the sum would be free
@@ -837,7 +810,7 @@
// checks that we can do unreserve stakes in the block
ensure!(!pendings.is_full(), Error::<T>::PendingForBlockOverflow);
- let mut stakes = Staked::<T>::iter_prefix((staker_id,)).collect::<Vec<_>>();
+ let mut stakes = Staked::<T>::iter_prefix((&staker_id,)).collect::<Vec<_>>();
let total_staked = stakes
.iter()
@@ -845,7 +818,10 @@
acc + *balance
});
- ensure!(total_staked >= unstaked_balance, ArithmeticError::Underflow);
+ ensure!(
+ total_staked >= unstaked_balance,
+ <Error<T>>::InsufficientStakedBalance
+ );
<TotalStaked<T>>::set(
<TotalStaked<T>>::get()
@@ -880,7 +856,7 @@
.try_push((staker_id.clone(), unstaked_balance))
.map_err(|_| Error::<T>::PendingForBlockOverflow)?;
- StakesPerAccount::<T>::try_mutate(staker_id, |stakes| -> DispatchResult {
+ StakesPerAccount::<T>::try_mutate(&staker_id, |stakes| -> DispatchResult {
*stakes = stakes
.checked_sub(will_deleted_stakes_count)
.ok_or(ArithmeticError::Underflow)?;
@@ -891,9 +867,9 @@
.iter()
.for_each(|(staked_block, (current_stake_state, _))| {
if current_stake_state == &Default::default() {
- <Staked<T>>::remove((staker_id, staked_block));
+ <Staked<T>>::remove((&staker_id, staked_block));
} else {
- <Staked<T>>::mutate((staker_id, staked_block), |(old_stake_state, _)| {
+ <Staked<T>>::mutate((&staker_id, staked_block), |(old_stake_state, _)| {
*old_stake_state = *current_stake_state
});
}
@@ -901,7 +877,7 @@
<PendingUnstake<T>>::insert(unpending_block, pendings);
- Self::deposit_event(Event::Unstake(staker_id.clone(), total_staked));
+ Self::deposit_event(Event::Unstake(staker_id, total_staked));
Ok(())
}
@@ -1102,4 +1078,47 @@
unsorted_res.sort_by_key(|(block, _)| *block);
unsorted_res
}
+
+ fn unstake_all_internal(staker_id: T::AccountId) -> DispatchResult {
+ let config = <PalletConfiguration<T>>::get();
+
+ // calculate block number where the sum would be free
+ let block = <frame_system::Pallet<T>>::block_number() + config.pending_interval;
+
+ let mut pendings = <PendingUnstake<T>>::get(block);
+
+ // checks that we can do unreserve stakes in the block
+ ensure!(!pendings.is_full(), Error::<T>::PendingForBlockOverflow);
+
+ let mut total_stakes = 0u64;
+
+ let total_staked: BalanceOf<T> = Staked::<T>::drain_prefix((&staker_id,))
+ .map(|(_, (amount, _))| {
+ total_stakes += 1;
+ amount
+ })
+ .sum();
+
+ if total_staked.is_zero() {
+ return Ok(());
+ }
+
+ pendings
+ .try_push((staker_id.clone(), total_staked))
+ .map_err(|_| Error::<T>::PendingForBlockOverflow)?;
+
+ <PendingUnstake<T>>::insert(block, pendings);
+
+ TotalStaked::<T>::set(
+ TotalStaked::<T>::get()
+ .checked_sub(&total_staked)
+ .ok_or(ArithmeticError::Underflow)?,
+ );
+
+ StakesPerAccount::<T>::remove(&staker_id);
+
+ Self::deposit_event(Event::Unstake(staker_id, total_staked));
+
+ Ok(())
+ }
}
pallets/app-promotion/src/weights.rsdiffbeforeafterboth--- a/pallets/app-promotion/src/weights.rs
+++ b/pallets/app-promotion/src/weights.rs
@@ -3,7 +3,7 @@
//! Autogenerated weights for pallet_app_promotion
//!
//! THIS FILE WAS AUTO-GENERATED USING THE SUBSTRATE BENCHMARK CLI VERSION 4.0.0-dev
-//! DATE: 2022-12-25, STEPS: `50`, REPEAT: 80, LOW RANGE: `[]`, HIGH RANGE: `[]`
+//! DATE: 2023-02-14, STEPS: `50`, REPEAT: 80, LOW RANGE: `[]`, HIGH RANGE: `[]`
//! EXECUTION: None, WASM-EXECUTION: Compiled, CHAIN: None, DB CACHE: 1024
// Executed Command:
@@ -38,7 +38,8 @@
fn set_admin_address() -> Weight;
fn payout_stakers(b: u32, ) -> Weight;
fn stake() -> Weight;
- fn unstake() -> Weight;
+ fn unstake_all() -> Weight;
+ fn unstake_partial() -> Weight;
fn sponsor_collection() -> Weight;
fn stop_sponsoring_collection() -> Weight;
fn sponsor_contract() -> Weight;
@@ -49,18 +50,19 @@
pub struct SubstrateWeight<T>(PhantomData<T>);
impl<T: frame_system::Config> WeightInfo for SubstrateWeight<T> {
// Storage: AppPromotion PendingUnstake (r:1 w:0)
+ // Storage: Balances Locks (r:1 w:1)
// Storage: System Account (r:1 w:1)
fn on_initialize(b: u32, ) -> Weight {
- Weight::from_ref_time(3_079_948 as u64)
- // Standard Error: 30_376
- .saturating_add(Weight::from_ref_time(6_343_630 as u64).saturating_mul(b as u64))
+ Weight::from_ref_time(2_562_275 as u64)
+ // Standard Error: 21_950
+ .saturating_add(Weight::from_ref_time(7_177_129 as u64).saturating_mul(b as u64))
.saturating_add(T::DbWeight::get().reads(1 as u64))
- .saturating_add(T::DbWeight::get().reads((1 as u64).saturating_mul(b as u64)))
- .saturating_add(T::DbWeight::get().writes((1 as u64).saturating_mul(b as u64)))
+ .saturating_add(T::DbWeight::get().reads((2 as u64).saturating_mul(b as u64)))
+ .saturating_add(T::DbWeight::get().writes((2 as u64).saturating_mul(b as u64)))
}
// Storage: AppPromotion Admin (r:0 w:1)
fn set_admin_address() -> Weight {
- Weight::from_ref_time(6_653_000 as u64)
+ Weight::from_ref_time(6_146_000 as u64)
.saturating_add(T::DbWeight::get().writes(1 as u64))
}
// Storage: AppPromotion Admin (r:1 w:0)
@@ -72,9 +74,9 @@
// Storage: Balances Locks (r:1 w:1)
// Storage: AppPromotion TotalStaked (r:1 w:1)
fn payout_stakers(b: u32, ) -> Weight {
- Weight::from_ref_time(74_048_000 as u64)
- // Standard Error: 33_223
- .saturating_add(Weight::from_ref_time(57_702_092 as u64).saturating_mul(b as u64))
+ Weight::from_ref_time(64_035_000 as u64)
+ // Standard Error: 19_434
+ .saturating_add(Weight::from_ref_time(47_251_111 as u64).saturating_mul(b as u64))
.saturating_add(T::DbWeight::get().reads(7 as u64))
.saturating_add(T::DbWeight::get().reads((12 as u64).saturating_mul(b as u64)))
.saturating_add(T::DbWeight::get().writes(3 as u64))
@@ -88,47 +90,55 @@
// Storage: AppPromotion Staked (r:1 w:1)
// Storage: AppPromotion TotalStaked (r:1 w:1)
fn stake() -> Weight {
- Weight::from_ref_time(20_314_000 as u64)
+ Weight::from_ref_time(18_078_000 as u64)
.saturating_add(T::DbWeight::get().reads(7 as u64))
.saturating_add(T::DbWeight::get().writes(5 as u64))
}
// Storage: Configuration AppPromomotionConfigurationOverride (r:1 w:0)
// Storage: AppPromotion PendingUnstake (r:1 w:1)
// Storage: AppPromotion Staked (r:11 w:10)
- // Storage: Balances Locks (r:1 w:1)
- // Storage: System Account (r:1 w:1)
// Storage: AppPromotion TotalStaked (r:1 w:1)
// Storage: AppPromotion StakesPerAccount (r:0 w:1)
- fn unstake() -> Weight {
- Weight::from_ref_time(64_582_000 as u64)
- .saturating_add(T::DbWeight::get().reads(16 as u64))
- .saturating_add(T::DbWeight::get().writes(15 as u64))
+ fn unstake_all() -> Weight {
+ Weight::from_ref_time(45_038_000 as u64)
+ .saturating_add(T::DbWeight::get().reads(14 as u64))
+ .saturating_add(T::DbWeight::get().writes(13 as u64))
+ }
+ // Storage: Configuration AppPromomotionConfigurationOverride (r:1 w:0)
+ // Storage: AppPromotion PendingUnstake (r:1 w:1)
+ // Storage: AppPromotion Staked (r:11 w:10)
+ // Storage: AppPromotion TotalStaked (r:1 w:1)
+ // Storage: AppPromotion StakesPerAccount (r:1 w:1)
+ fn unstake_partial() -> Weight {
+ Weight::from_ref_time(48_863_000 as u64)
+ .saturating_add(T::DbWeight::get().reads(15 as u64))
+ .saturating_add(T::DbWeight::get().writes(13 as u64))
}
// Storage: AppPromotion Admin (r:1 w:0)
// Storage: Common CollectionById (r:1 w:1)
fn sponsor_collection() -> Weight {
- Weight::from_ref_time(16_364_000 as u64)
+ Weight::from_ref_time(14_808_000 as u64)
.saturating_add(T::DbWeight::get().reads(2 as u64))
.saturating_add(T::DbWeight::get().writes(1 as u64))
}
// Storage: AppPromotion Admin (r:1 w:0)
// Storage: Common CollectionById (r:1 w:1)
fn stop_sponsoring_collection() -> Weight {
- Weight::from_ref_time(15_710_000 as u64)
+ Weight::from_ref_time(14_587_000 as u64)
.saturating_add(T::DbWeight::get().reads(2 as u64))
.saturating_add(T::DbWeight::get().writes(1 as u64))
}
// Storage: AppPromotion Admin (r:1 w:0)
// Storage: EvmContractHelpers Sponsoring (r:0 w:1)
fn sponsor_contract() -> Weight {
- Weight::from_ref_time(12_669_000 as u64)
+ Weight::from_ref_time(11_791_000 as u64)
.saturating_add(T::DbWeight::get().reads(1 as u64))
.saturating_add(T::DbWeight::get().writes(1 as u64))
}
// Storage: AppPromotion Admin (r:1 w:0)
// Storage: EvmContractHelpers Sponsoring (r:1 w:1)
fn stop_sponsoring_contract() -> Weight {
- Weight::from_ref_time(14_406_000 as u64)
+ Weight::from_ref_time(13_576_000 as u64)
.saturating_add(T::DbWeight::get().reads(2 as u64))
.saturating_add(T::DbWeight::get().writes(1 as u64))
}
@@ -137,18 +147,19 @@
// For backwards compatibility and tests
impl WeightInfo for () {
// Storage: AppPromotion PendingUnstake (r:1 w:0)
+ // Storage: Balances Locks (r:1 w:1)
// Storage: System Account (r:1 w:1)
fn on_initialize(b: u32, ) -> Weight {
- Weight::from_ref_time(3_079_948 as u64)
- // Standard Error: 30_376
- .saturating_add(Weight::from_ref_time(6_343_630 as u64).saturating_mul(b as u64))
+ Weight::from_ref_time(2_562_275 as u64)
+ // Standard Error: 21_950
+ .saturating_add(Weight::from_ref_time(7_177_129 as u64).saturating_mul(b as u64))
.saturating_add(RocksDbWeight::get().reads(1 as u64))
- .saturating_add(RocksDbWeight::get().reads((1 as u64).saturating_mul(b as u64)))
- .saturating_add(RocksDbWeight::get().writes((1 as u64).saturating_mul(b as u64)))
+ .saturating_add(RocksDbWeight::get().reads((2 as u64).saturating_mul(b as u64)))
+ .saturating_add(RocksDbWeight::get().writes((2 as u64).saturating_mul(b as u64)))
}
// Storage: AppPromotion Admin (r:0 w:1)
fn set_admin_address() -> Weight {
- Weight::from_ref_time(6_653_000 as u64)
+ Weight::from_ref_time(6_146_000 as u64)
.saturating_add(RocksDbWeight::get().writes(1 as u64))
}
// Storage: AppPromotion Admin (r:1 w:0)
@@ -160,9 +171,9 @@
// Storage: Balances Locks (r:1 w:1)
// Storage: AppPromotion TotalStaked (r:1 w:1)
fn payout_stakers(b: u32, ) -> Weight {
- Weight::from_ref_time(74_048_000 as u64)
- // Standard Error: 33_223
- .saturating_add(Weight::from_ref_time(57_702_092 as u64).saturating_mul(b as u64))
+ Weight::from_ref_time(64_035_000 as u64)
+ // Standard Error: 19_434
+ .saturating_add(Weight::from_ref_time(47_251_111 as u64).saturating_mul(b as u64))
.saturating_add(RocksDbWeight::get().reads(7 as u64))
.saturating_add(RocksDbWeight::get().reads((12 as u64).saturating_mul(b as u64)))
.saturating_add(RocksDbWeight::get().writes(3 as u64))
@@ -176,47 +187,55 @@
// Storage: AppPromotion Staked (r:1 w:1)
// Storage: AppPromotion TotalStaked (r:1 w:1)
fn stake() -> Weight {
- Weight::from_ref_time(20_314_000 as u64)
+ Weight::from_ref_time(18_078_000 as u64)
.saturating_add(RocksDbWeight::get().reads(7 as u64))
.saturating_add(RocksDbWeight::get().writes(5 as u64))
}
// Storage: Configuration AppPromomotionConfigurationOverride (r:1 w:0)
// Storage: AppPromotion PendingUnstake (r:1 w:1)
// Storage: AppPromotion Staked (r:11 w:10)
- // Storage: Balances Locks (r:1 w:1)
- // Storage: System Account (r:1 w:1)
// Storage: AppPromotion TotalStaked (r:1 w:1)
// Storage: AppPromotion StakesPerAccount (r:0 w:1)
- fn unstake() -> Weight {
- Weight::from_ref_time(64_582_000 as u64)
- .saturating_add(RocksDbWeight::get().reads(16 as u64))
- .saturating_add(RocksDbWeight::get().writes(15 as u64))
+ fn unstake_all() -> Weight {
+ Weight::from_ref_time(45_038_000 as u64)
+ .saturating_add(RocksDbWeight::get().reads(14 as u64))
+ .saturating_add(RocksDbWeight::get().writes(13 as u64))
}
+ // Storage: Configuration AppPromomotionConfigurationOverride (r:1 w:0)
+ // Storage: AppPromotion PendingUnstake (r:1 w:1)
+ // Storage: AppPromotion Staked (r:11 w:10)
+ // Storage: AppPromotion TotalStaked (r:1 w:1)
+ // Storage: AppPromotion StakesPerAccount (r:1 w:1)
+ fn unstake_partial() -> Weight {
+ Weight::from_ref_time(48_863_000 as u64)
+ .saturating_add(RocksDbWeight::get().reads(15 as u64))
+ .saturating_add(RocksDbWeight::get().writes(13 as u64))
+ }
// Storage: AppPromotion Admin (r:1 w:0)
// Storage: Common CollectionById (r:1 w:1)
fn sponsor_collection() -> Weight {
- Weight::from_ref_time(16_364_000 as u64)
+ Weight::from_ref_time(14_808_000 as u64)
.saturating_add(RocksDbWeight::get().reads(2 as u64))
.saturating_add(RocksDbWeight::get().writes(1 as u64))
}
// Storage: AppPromotion Admin (r:1 w:0)
// Storage: Common CollectionById (r:1 w:1)
fn stop_sponsoring_collection() -> Weight {
- Weight::from_ref_time(15_710_000 as u64)
+ Weight::from_ref_time(14_587_000 as u64)
.saturating_add(RocksDbWeight::get().reads(2 as u64))
.saturating_add(RocksDbWeight::get().writes(1 as u64))
}
// Storage: AppPromotion Admin (r:1 w:0)
// Storage: EvmContractHelpers Sponsoring (r:0 w:1)
fn sponsor_contract() -> Weight {
- Weight::from_ref_time(12_669_000 as u64)
+ Weight::from_ref_time(11_791_000 as u64)
.saturating_add(RocksDbWeight::get().reads(1 as u64))
.saturating_add(RocksDbWeight::get().writes(1 as u64))
}
// Storage: AppPromotion Admin (r:1 w:0)
// Storage: EvmContractHelpers Sponsoring (r:1 w:1)
fn stop_sponsoring_contract() -> Weight {
- Weight::from_ref_time(14_406_000 as u64)
+ Weight::from_ref_time(13_576_000 as u64)
.saturating_add(RocksDbWeight::get().reads(2 as u64))
.saturating_add(RocksDbWeight::get().writes(1 as u64))
}
tests/src/sub/appPromotion/appPromotion.test.tsdiffbeforeafterboth1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617import {IKeyringPair} from '@polkadot/types/types';18import {19 itSub, usingPlaygrounds, Pallets, requirePalletsOrSkip, LOCKING_PERIOD, UNLOCKING_PERIOD,20} from '../../util';21import {DevUniqueHelper} from '../../util/playgrounds/unique.dev';22import {itEth, expect, SponsoringMode} from '../../eth/util';2324let donor: IKeyringPair;25let palletAdmin: IKeyringPair;26let nominal: bigint;27let palletAddress: string;28let accounts: IKeyringPair[];29// App promotion periods:30// LOCKING_PERIOD = 12 blocks of relay31// UNLOCKING_PERIOD = 6 blocks of parachain3233describe('App promotion', () => {34 before(async function () {35 await usingPlaygrounds(async (helper, privateKey) => {36 requirePalletsOrSkip(this, helper, [Pallets.AppPromotion]);37 donor = await privateKey({filename: __filename});38 palletAddress = helper.arrange.calculatePalletAddress('appstake');39 palletAdmin = await privateKey('//PromotionAdmin');40 nominal = helper.balance.getOneTokenNominal();4142 const accountBalances = new Array(100);43 accountBalances.fill(1000n);44 accounts = await helper.arrange.createAccounts(accountBalances, donor); // create accounts-pool to speed up tests45 });46 });4748 describe('stake extrinsic', () => {49 itSub('should "lock" staking balance, add it to "staked" map, and increase "totalStaked" amount', async ({helper}) => {50 const [staker, recepient] = [accounts.pop()!, accounts.pop()!];51 const totalStakedBefore = await helper.staking.getTotalStaked();5253 // Minimum stake amount is 100:54 await expect(helper.staking.stake(staker, 100n * nominal - 1n)).to.be.rejected;55 await helper.staking.stake(staker, 100n * nominal);5657 // Staker balance is: miscFrozen: 100, feeFrozen: 100, reserved: 0n...58 // ...so he can not transfer 90059 expect(await helper.balance.getSubstrateFull(staker.address)).to.contain({miscFrozen: 100n * nominal, feeFrozen: 100n * nominal, reserved: 0n});60 expect(await helper.balance.getLocked(staker.address)).to.deep.eq([{id: 'appstake', amount: 100n * nominal, reasons: 'All'}]);61 await expect(helper.balance.transferToSubstrate(staker, recepient.address, 900n * nominal)).to.be.rejectedWith('balances.LiquidityRestrictions');6263 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(100n * nominal);64 expect(await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n);65 // it is potentially flaky test. Promotion can credited some tokens. Maybe we need to use closeTo?66 expect(await helper.staking.getTotalStaked()).to.be.equal(totalStakedBefore + 100n * nominal); // total tokens amount staked in app-promotion increased676869 await helper.staking.stake(staker, 200n * nominal);70 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(300n * nominal);71 const totalStakedPerBlock = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});72 expect(totalStakedPerBlock[0].amount).to.equal(100n * nominal);73 expect(totalStakedPerBlock[1].amount).to.equal(200n * nominal);74 });7576 itSub('should allow to create maximum 10 stakes for account', async ({helper}) => {77 const [staker] = await helper.arrange.createAccounts([2000n], donor);78 for (let i = 0; i < 10; i++) {79 await helper.staking.stake(staker, 100n * nominal);80 }8182 // can have 10 stakes83 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(1000n * nominal);84 expect(await helper.staking.getTotalStakedPerBlock({Substrate: staker.address})).to.have.length(10);8586 await expect(helper.staking.stake(staker, 100n * nominal)).to.be.rejectedWith('appPromotion.NoPermission');8788 // After unstake can stake again89 await helper.staking.unstakeAll(staker);90 await helper.staking.stake(staker, 100n * nominal);91 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.equal(100n * nominal);92 });9394 itSub('should allow to stake() if balance is locked with different id', async ({helper}) => {95 const staker = accounts.pop()!;9697 // staker has tokens locked with vesting id:98 await helper.balance.vestedTransfer(donor, staker.address, {start: 0n, period: 1n, periodCount: 1n, perPeriod: 200n * nominal});99 expect(await helper.balance.getSubstrateFull(staker.address))100 .to.deep.contain({free: 1200n * nominal, miscFrozen: 200n * nominal, feeFrozen: 200n * nominal, reserved: 0n});101102 // Locked balance can be staked. staker can stake 1200 tokens (minus fee):103 await helper.staking.stake(staker, 1000n * nominal);104 await helper.staking.stake(staker, 199n * nominal);105 // check balances106 expect(await helper.balance.getLocked(staker.address)).to.deep.eq([{id: 'ormlvest', amount: 200n * nominal, reasons: 'All'}, {id: 'appstake', amount: 1199n * nominal, reasons: 'All'}]);107 expect(await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, miscFrozen: 1199n * nominal, feeFrozen: 1199n * nominal});108 expect(await helper.balance.getSubstrate(staker.address) / nominal).to.eq(1199n);109 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.eq(1199n * nominal);110111 // staker can unstake112 await helper.staking.unstakeAll(staker);113 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.eq(1199n * nominal);114 const [pendingUnstake] = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});115 await helper.wait.forParachainBlockNumber(pendingUnstake.block);116117 // check balances118 expect(await helper.balance.getLocked(staker.address)).to.deep.eq([{id: 'ormlvest', amount: 200n * nominal, reasons: 'All'}]);119 expect(await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, miscFrozen: 200n * nominal, feeFrozen: 200n * nominal});120 expect(await helper.balance.getSubstrate(staker.address) / nominal).to.eq(1199n);121 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.eq(0n);122123 // staker can transfer balances now124 await helper.balance.transferToSubstrate(staker, donor.address, 900n * nominal);125 });126127 itSub('should not allow to stake(), if stake amount is more than total free balance minus locked by staking', async ({helper}) => {128 const staker = accounts.pop()!;129130 // Can't stake full balance because Alice needs to pay some fee131 await expect(helper.staking.stake(staker, 1000n * nominal)).to.be.rejected; // With('Arithmetic')132 await helper.staking.stake(staker, 500n * nominal);133134 // Can't stake 500 tkn because Alice has Less than 500 transferable;135 await expect(helper.staking.stake(staker, 500n * nominal)).to.be.rejected; // With('Arithmetic');136 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(500n * nominal);137 });138139 itSub('for different accounts in one block is possible', async ({helper}) => {140 const crowd = [accounts.pop()!, accounts.pop()!, accounts.pop()!, accounts.pop()!];141142 const crowdStartsToStake = crowd.map(user => helper.staking.stake(user, 100n * nominal));143 await expect(Promise.all(crowdStartsToStake)).to.be.fulfilled;144145 const crowdStakes = await Promise.all(crowd.map(address => helper.staking.getTotalStaked({Substrate: address.address})));146 expect(crowdStakes).to.deep.equal([100n * nominal, 100n * nominal, 100n * nominal, 100n * nominal]);147 });148 });149150 describe('Unstaking', () => {151 [152 {method: 'unstakeAll' as const},153 {method: 'unstakePartial' as const},154 ].map(testCase => {155 itSub(`[${testCase.method}] should move tokens to "pendingUnstake" and subtract it from totalStaked`, async ({helper}) => {156 const [staker, recepient] = [accounts.pop()!, accounts.pop()!];157 const totalStakedBefore = await helper.staking.getTotalStaked();158 const STAKE_AMOUNT = 900n * nominal;159160 await helper.staking.stake(staker, STAKE_AMOUNT);161 testCase.method === 'unstakeAll'162 ? await helper.staking.unstakeAll(staker)163 : await helper.staking.unstakePartial(staker, STAKE_AMOUNT);164165 // Right after unstake tokens are still locked166 expect(await helper.staking.getStakesNumber({Substrate: staker.address})).to.eq(0);167 expect(await helper.balance.getLocked(staker.address)).to.deep.eq([{id: 'appstake', amount: STAKE_AMOUNT, reasons: 'All'}]);168 expect(await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, miscFrozen: STAKE_AMOUNT, feeFrozen: STAKE_AMOUNT});169 // Staker can not transfer170 await expect(helper.balance.transferToSubstrate(staker, recepient.address, 100n * nominal)).to.be.rejectedWith('balances.LiquidityRestrictions');171 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(STAKE_AMOUNT);172 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(0n);173 expect(await helper.staking.getTotalStaked()).to.be.equal(totalStakedBefore);174 });175 });176177 [178 {method: 'unstakeAll' as const},179 {method: 'unstakePartial' as const},180 ].map(testCase => {181 itSub(`[${testCase.method}] should unlock balance after unlocking period ends and remove it from "pendingUnstake"`, async ({helper}) => {182 const staker = accounts.pop()!;183 await helper.staking.stake(staker, 100n * nominal);184 testCase.method === 'unstakeAll'185 ? await helper.staking.unstakeAll(staker)186 : await helper.staking.unstakePartial(staker, 100n * nominal);187 const [pendingUnstake] = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});188189 // Wait for unstaking period. Balance now free ~1000; reserved, frozen, miscFrozeb: 0n190 await helper.wait.forParachainBlockNumber(pendingUnstake.block);191 expect(await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, miscFrozen: 0n, feeFrozen: 0n});192 expect(await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n);193194 // staker can transfer:195 await helper.balance.transferToSubstrate(staker, donor.address, 998n * nominal);196 expect(await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(1n);197 });198 });199200 [201 {method: 'unstakeAll' as const},202 {method: 'unstakePartial' as const},203 ].map(testCase => {204 itSub(`[${testCase.method}] should successfully unstake multiple stakes`, async ({helper}) => {205 const staker = accounts.pop()!;206 await helper.staking.stake(staker, 100n * nominal);207 await helper.staking.stake(staker, 200n * nominal);208 await helper.staking.stake(staker, 300n * nominal);209210 // staked: [100, 200, 300]; unstaked: 0211 let totalPendingUnstake = await helper.staking.getPendingUnstake({Substrate: staker.address});212 let pendingUnstake = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});213 let stakes = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});214 expect(totalPendingUnstake).to.be.deep.equal(0n);215 expect(pendingUnstake).to.be.deep.equal([]);216 expect(stakes[0].amount).to.equal(100n * nominal);217 expect(stakes[1].amount).to.equal(200n * nominal);218 expect(stakes[2].amount).to.equal(300n * nominal);219220 // Can unstake multiple stakes221 testCase.method === 'unstakeAll'222 ? await helper.staking.unstakeAll(staker)223 : await helper.staking.unstakePartial(staker, 600n * nominal);224225 pendingUnstake = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});226 totalPendingUnstake = await helper.staking.getPendingUnstake({Substrate: staker.address});227 stakes = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});228 expect(totalPendingUnstake).to.be.equal(600n * nominal);229 expect(stakes).to.be.deep.equal([]);230 expect(pendingUnstake[0].amount).to.equal(600n * nominal);231232 expect (await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, feeFrozen: 600n * nominal, miscFrozen: 600n * nominal});233 expect (await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n);234 await helper.wait.forParachainBlockNumber(pendingUnstake[0].block);235 expect (await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, feeFrozen: 0n, miscFrozen: 0n});236 expect (await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n);237 });238 });239240 [241 {method: 'unstakeAll' as const},242 {method: 'unstakePartial' as const},243 ].map(testCase => {244 itSub(`[${testCase.method}] should not have any effects if no active stakes`, async ({helper}) => {245 const staker = accounts.pop()!;246247 // unstake has no effect if no stakes at all248 testCase.method === 'unstakeAll'249 ? await helper.staking.unstakeAll(staker)250 : await expect(helper.staking.unstakePartial(staker, 100n * nominal)).to.be.rejectedWith('Arithmetic: Underflow');251252 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(0n);253 expect(await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n); // TODO bigint closeTo helper254255 // TODO stake() unstake() waitUnstaked() unstake();256257 // can't unstake if there are only pendingUnstakes258 await helper.staking.stake(staker, 100n * nominal);259260 if (testCase.method === 'unstakeAll') {261 await helper.staking.unstakeAll(staker);262 await helper.staking.unstakeAll(staker);263 } else {264 await helper.staking.unstakePartial(staker, 100n * nominal);265 await expect(helper.staking.unstakePartial(staker, 100n * nominal)).to.be.rejectedWith('Arithmetic: Underflow');266 }267268 expect(await helper.staking.getStakesNumber({Substrate: staker.address})).to.eq(0);269 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(100n * nominal);270 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(0n);271 });272 });273274 [275 {method: 'unstakeAll' as const},276 {method: 'unstakePartial' as const},277 ].map(testCase => {278 itSub(`[${testCase.method}] should create different pending-unlock for each unlocking stake`, async ({helper}) => {279 const staker = accounts.pop()!;280 await helper.staking.stake(staker, 100n * nominal);281 testCase.method === 'unstakeAll'282 ? await helper.staking.unstakeAll(staker)283 : await helper.staking.unstakePartial(staker, 100n * nominal);284 await helper.staking.stake(staker, 120n * nominal);285 testCase.method === 'unstakeAll'286 ? await helper.staking.unstakeAll(staker)287 : await helper.staking.unstakePartial(staker, 120n * nominal);288289 const unstakingPerBlock = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});290 expect(unstakingPerBlock).has.length(2);291 expect(unstakingPerBlock[0].amount).to.equal(100n * nominal);292 expect(unstakingPerBlock[1].amount).to.equal(120n * nominal);293 expect(await helper.staking.getStakesNumber({Substrate: staker.address})).to.equal(0);294 });295 });296297 [298 {method: 'unstakeAll' as const},299 {method: 'unstakePartial' as const},300 ].map(testCase => {301 itSub(`[${testCase.method}] should be possible for 3 accounts in one block`, async ({helper}) => {302 const stakers = [accounts.pop()!, accounts.pop()!, accounts.pop()!];303304 await Promise.all(stakers.map(staker => helper.staking.stake(staker, 100n * nominal)));305 await Promise.all(stakers.map(staker => {306 return testCase.method === 'unstakeAll'307 ? helper.staking.unstakeAll(staker)308 : helper.staking.unstakePartial(staker, 100n * nominal);309 }));310311 await Promise.all(stakers.map(async (staker) => {312 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(100n * nominal);313 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(0n);314 }));315 });316 });317318 itSub('should not be possible for more than 3 accounts in one block', async ({helper}) => {319 if (!await helper.arrange.isDevNode()) {320 const stakers = await helper.arrange.createAccounts([200n,200n,200n,200n,200n,200n,200n,200n,200n,200n], donor);321322 await Promise.all(stakers.map(staker => helper.staking.stake(staker, 100n * nominal)));323 const unstakingResults = await Promise.allSettled(stakers.map((staker, i) => {324 return i % 2 === 0325 ? helper.staking.unstakeAll(staker)326 : helper.staking.unstakePartial(staker, 100n * nominal);327 }));328329 const successfulUnstakes = unstakingResults.filter(result => result.status === 'fulfilled');330 expect(successfulUnstakes).to.have.length(3);331 }332 });333 });334335 describe('collection sponsoring', () => {336 itSub('should actually sponsor transactions', async ({helper}) => {337 const api = helper.getApi();338 const [collectionOwner, tokenSender, receiver] = [accounts.pop()!, accounts.pop()!, accounts.pop()!];339 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'Name', description: 'Description', tokenPrefix: 'Prefix', limits: {sponsorTransferTimeout: 0}});340 const token = await collection.mintToken(collectionOwner, {Substrate: tokenSender.address});341 await helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId));342 const palletBalanceBefore = await helper.balance.getSubstrate(palletAddress);343344 await token.transfer(tokenSender, {Substrate: receiver.address});345 expect (await token.getOwner()).to.be.deep.equal({Substrate: receiver.address});346 const palletBalanceAfter = await helper.balance.getSubstrate(palletAddress);347348 // senders balance the same, transaction has sponsored349 expect (await helper.balance.getSubstrate(tokenSender.address)).to.be.equal(1000n * nominal);350 expect (palletBalanceBefore > palletBalanceAfter).to.be.true;351 });352353 itSub('can not be set by non admin', async ({helper}) => {354 const api = helper.getApi();355 const [collectionOwner, nonAdmin] = [accounts.pop()!, accounts.pop()!];356357 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});358359 await expect(helper.signTransaction(nonAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId))).to.be.rejected;360 expect((await collection.getData())?.raw.sponsorship).to.equal('Disabled');361 });362363 itSub('should set pallet address as confirmed admin', async ({helper}) => {364 const api = helper.getApi();365 const [collectionOwner, oldSponsor] = [accounts.pop()!, accounts.pop()!];366367 // Can set sponsoring for collection without sponsor368 const collectionWithoutSponsor = await helper.nft.mintCollection(collectionOwner, {name: 'No-sponsor', description: 'New Collection', tokenPrefix: 'Promotion'});369 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionWithoutSponsor.collectionId))).to.be.fulfilled;370 expect((await collectionWithoutSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});371372 // Can set sponsoring for collection with unconfirmed sponsor373 const collectionWithUnconfirmedSponsor = await helper.nft.mintCollection(collectionOwner, {name: 'Unconfirmed', description: 'New Collection', tokenPrefix: 'Promotion', pendingSponsor: oldSponsor.address});374 expect((await collectionWithUnconfirmedSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Unconfirmed: oldSponsor.address});375 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionWithUnconfirmedSponsor.collectionId))).to.be.fulfilled;376 expect((await collectionWithUnconfirmedSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});377378 // Can set sponsoring for collection with confirmed sponsor379 const collectionWithConfirmedSponsor = await helper.nft.mintCollection(collectionOwner, {name: 'Confirmed', description: 'New Collection', tokenPrefix: 'Promotion', pendingSponsor: oldSponsor.address});380 await collectionWithConfirmedSponsor.confirmSponsorship(oldSponsor);381 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionWithConfirmedSponsor.collectionId))).to.be.fulfilled;382 expect((await collectionWithConfirmedSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});383 });384385 itSub('can be overwritten by collection owner', async ({helper}) => {386 const api = helper.getApi();387 const [collectionOwner, newSponsor] = [accounts.pop()!, accounts.pop()!];388 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});389 const collectionId = collection.collectionId;390391 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionId))).to.be.fulfilled;392393 // Collection limits still can be changed by the owner394 expect(await collection.setLimits(collectionOwner, {sponsorTransferTimeout: 0})).to.be.true;395 expect((await collection.getData())?.raw.limits.sponsorTransferTimeout).to.be.equal(0);396 expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});397398 // Collection sponsor can be changed too399 expect((await collection.setSponsor(collectionOwner, newSponsor.address))).to.be.true;400 expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Unconfirmed: newSponsor.address});401 });402403 itSub('should not overwrite collection limits set by the owner earlier', async ({helper}) => {404 const api = helper.getApi();405 const limits = {ownerCanDestroy: true, ownerCanTransfer: true, sponsorTransferTimeout: 0};406 const collectionWithLimits = await helper.nft.mintCollection(accounts.pop()!, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion', limits});407408 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionWithLimits.collectionId))).to.be.fulfilled;409 expect((await collectionWithLimits.getData())?.raw.limits).to.be.deep.contain(limits);410 });411412 itSub('should reject transaction if collection doesn\'t exist', async ({helper}) => {413 const api = helper.getApi();414 const collectionOwner = accounts.pop()!;415416 // collection has never existed417 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(999999999))).to.be.rejected;418 // collection has been burned419 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});420 await collection.burn(collectionOwner);421422 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId))).to.be.rejected;423 });424 });425426 describe('stopSponsoringCollection', () => {427 itSub('can not be called by non-admin', async ({helper}) => {428 const api = helper.getApi();429 const [collectionOwner, nonAdmin] = [accounts.pop()!, accounts.pop()!];430 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});431432 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId))).to.be.fulfilled;433434 await expect(helper.signTransaction(nonAdmin, api.tx.appPromotion.stopSponsoringCollection(collection.collectionId))).to.be.rejected;435 expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});436 });437438 itSub('should set sponsoring as disabled', async ({helper}) => {439 const api = helper.getApi();440 const [collectionOwner, recepient] = [accounts.pop()!, accounts.pop()!];441 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion', limits: {sponsorTransferTimeout: 0}});442 const token = await collection.mintToken(collectionOwner, {Substrate: collectionOwner.address});443444 await helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId));445 await helper.signTransaction(palletAdmin, api.tx.appPromotion.stopSponsoringCollection(collection.collectionId));446447 expect((await collection.getData())?.raw.sponsorship).to.be.equal('Disabled');448449 // Transactions are not sponsored anymore:450 const ownerBalanceBefore = await helper.balance.getSubstrate(collectionOwner.address);451 await token.transfer(collectionOwner, {Substrate: recepient.address});452 const ownerBalanceAfter = await helper.balance.getSubstrate(collectionOwner.address);453 expect(ownerBalanceAfter < ownerBalanceBefore).to.be.equal(true);454 });455456 itSub('should not affect collection which is not sponsored by pallete', async ({helper}) => {457 const api = helper.getApi();458 const collectionOwner = accounts.pop()!;459 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion', pendingSponsor: collectionOwner.address});460 await collection.confirmSponsorship(collectionOwner);461462 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.stopSponsoringCollection(collection.collectionId))).to.be.rejected;463464 expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: collectionOwner.address});465 });466467 itSub('should reject transaction if collection does not exist', async ({helper}) => {468 const collectionOwner = accounts.pop()!;469 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});470471 await collection.burn(collectionOwner);472 await expect(helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.stopSponsoringCollection', [collection.collectionId], true)).to.be.rejectedWith('common.CollectionNotFound');473 await expect(helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.stopSponsoringCollection', [999_999_999], true)).to.be.rejectedWith('common.CollectionNotFound');474 });475 });476477 describe('contract sponsoring', () => {478 itEth('should set palletes address as a sponsor', async ({helper}) => {479 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();480 const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);481 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);482483 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address]);484485 expect(await contractHelper.methods.hasSponsor(flipper.options.address).call()).to.be.true;486 expect((await helper.callRpc('api.query.evmContractHelpers.owner', [flipper.options.address])).toJSON()).to.be.equal(contractOwner);487 expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({488 confirmed: {489 substrate: palletAddress,490 },491 });492 });493494 itEth('should overwrite sponsoring mode and existed sponsor', async ({helper}) => {495 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();496 const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);497 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);498499 await expect(contractHelper.methods.selfSponsoredEnable(flipper.options.address).send()).to.be.fulfilled;500501 // Contract is self sponsored502 expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.be.deep.equal({503 confirmed: {504 ethereum: flipper.options.address.toLowerCase(),505 },506 });507508 // set promotion sponsoring509 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true);510511 // new sponsor is pallet address512 expect(await contractHelper.methods.hasSponsor(flipper.options.address).call()).to.be.true;513 expect((await helper.callRpc('api.query.evmContractHelpers.owner', [flipper.options.address])).toJSON()).to.be.equal(contractOwner);514 expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({515 confirmed: {516 substrate: palletAddress,517 },518 });519 });520521 itEth('can be overwritten by contract owner', async ({helper}) => {522 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();523 const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);524 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);525526 // contract sponsored by pallet527 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true);528529 // owner sets self sponsoring530 await expect(contractHelper.methods.selfSponsoredEnable(flipper.options.address).send()).to.be.not.rejected;531532 expect(await contractHelper.methods.hasSponsor(flipper.options.address).call()).to.be.true;533 expect((await helper.callRpc('api.query.evmContractHelpers.owner', [flipper.options.address])).toJSON()).to.be.equal(contractOwner);534 expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({535 confirmed: {536 ethereum: flipper.options.address.toLowerCase(),537 },538 });539 });540541 itEth('can not be set by non admin', async ({helper}) => {542 const nonAdmin = accounts.pop()!;543 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();544 const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);545 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);546547 await expect(contractHelper.methods.selfSponsoredEnable(flipper.options.address).send()).to.be.fulfilled;548549 // nonAdmin calls sponsorContract550 await expect(helper.executeExtrinsic(nonAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true)).to.be.rejectedWith('appPromotion.NoPermission');551552 // contract still self-sponsored553 expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({554 confirmed: {555 ethereum: flipper.options.address.toLowerCase(),556 },557 });558 });559560 itEth('should actually sponsor transactions', async ({helper}) => {561 // Contract caller562 const caller = await helper.eth.createAccountWithBalance(donor, 1000n);563 const palletBalanceBefore = await helper.balance.getSubstrate(palletAddress);564565 // Deploy flipper566 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();567 const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);568 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);569570 // Owner sets to sponsor every tx571 await contractHelper.methods.setSponsoringRateLimit(flipper.options.address, 0).send({from: contractOwner});572 await contractHelper.methods.setSponsoringMode(flipper.options.address, SponsoringMode.Generous).send({from: contractOwner});573 await helper.eth.transferBalanceFromSubstrate(donor, flipper.options.address, 1000n); // transferBalanceToEth(api, alice, flipper.options.address, 1000n);574575 // Set promotion to the Flipper576 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true);577578 // Caller calls Flipper579 await flipper.methods.flip().send({from: caller});580 expect(await flipper.methods.getValue().call()).to.be.true;581582 // The contracts and caller balances have not changed583 const callerBalance = await helper.balance.getEthereum(caller);584 const contractBalanceAfter = await helper.balance.getEthereum(flipper.options.address);585 expect(callerBalance).to.be.equal(1000n * nominal);586 expect(1000n * nominal === contractBalanceAfter).to.be.true;587588 // The pallet balance has decreased589 const palletBalanceAfter = await helper.balance.getSubstrate(palletAddress);590 expect(palletBalanceAfter < palletBalanceBefore).to.be.true;591 });592 });593594 describe('stopSponsoringContract', () => {595 itEth('should remove pallet address from contract sponsors', async ({helper}) => {596 const caller = await helper.eth.createAccountWithBalance(donor, 1000n);597 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();598 const flipper = await helper.eth.deployFlipper(contractOwner);599 await helper.eth.transferBalanceFromSubstrate(donor, flipper.options.address);600 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);601602 await contractHelper.methods.setSponsoringMode(flipper.options.address, SponsoringMode.Generous).send({from: contractOwner});603 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true);604 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.stopSponsoringContract', [flipper.options.address], true);605606 expect(await contractHelper.methods.hasSponsor(flipper.options.address).call()).to.be.false;607 expect((await helper.callRpc('api.query.evmContractHelpers.owner', [flipper.options.address])).toJSON()).to.be.equal(contractOwner);608 expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({609 disabled: null,610 });611612 await flipper.methods.flip().send({from: caller});613 expect(await flipper.methods.getValue().call()).to.be.true;614615 const callerBalance = await helper.balance.getEthereum(caller);616 const contractBalanceAfter = await helper.balance.getEthereum(flipper.options.address);617618 // caller payed for call619 expect(1000n * nominal > callerBalance).to.be.true;620 expect(contractBalanceAfter).to.be.equal(100n * nominal);621 });622623 itEth('can not be called by non-admin', async ({helper}) => {624 const nonAdmin = accounts.pop()!;625 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();626 const flipper = await helper.eth.deployFlipper(contractOwner);627628 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address]);629 await expect(helper.executeExtrinsic(nonAdmin, 'api.tx.appPromotion.stopSponsoringContract', [flipper.options.address]))630 .to.be.rejectedWith(/appPromotion\.NoPermission/);631 });632633 itEth('should not affect a contract which is not sponsored by pallete', async ({helper}) => {634 const nonAdmin = accounts.pop()!;635 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();636 const flipper = await helper.eth.deployFlipper(contractOwner);637 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);638 await expect(contractHelper.methods.selfSponsoredEnable(flipper.options.address).send()).to.be.fulfilled;639640 await expect(helper.executeExtrinsic(nonAdmin, 'api.tx.appPromotion.stopSponsoringContract', [flipper.options.address], true)).to.be.rejectedWith('appPromotion.NoPermission');641 });642 });643644 describe('payoutStakers', () => {645 itSub('can not be called by non admin', async ({helper}) => {646 const nonAdmin = accounts.pop()!;647 await expect(helper.admin.payoutStakers(nonAdmin, 100)).to.be.rejectedWith('appPromotion.NoPermission');648 });649650 itSub('should increase total staked', async ({helper}) => {651 const staker = accounts.pop()!;652 const totalStakedBefore = await helper.staking.getTotalStaked();653 await helper.staking.stake(staker, 100n * nominal);654655 // Wait for rewards and pay656 const [stakedInBlock] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});657 await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stakedInBlock.block));658 const totalPayout = (await helper.admin.payoutStakers(palletAdmin, 100)).reduce((prev, payout) => prev + payout.payout, 0n);659660 const totalStakedAfter = await helper.staking.getTotalStaked();661 expect(totalStakedAfter).to.equal(totalStakedBefore + (100n * nominal) + totalPayout);662 // staker can unstake663 await helper.staking.unstakeAll(staker);664 expect(await helper.staking.getTotalStaked()).to.be.equal(totalStakedAfter - calculateIncome(100n * nominal));665 });666667 itSub('should credit 0.05% for staking period', async ({helper}) => {668 const staker = accounts.pop()!;669670 await waitPromotionPeriodDoesntEnd(helper);671672 await helper.staking.stake(staker, 100n * nominal);673 await helper.staking.stake(staker, 200n * nominal);674675 // wait rewards are available:676 const [_stake1, stake2] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});677 await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake2.block));678679 const payoutToStaker = (await helper.admin.payoutStakers(palletAdmin, 100)).find((payout) => payout.staker === staker.address)?.payout;680 expect(payoutToStaker + 300n * nominal).to.equal(calculateIncome(300n * nominal));681682 const totalStakedPerBlock = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});683 const income1 = calculateIncome(100n * nominal);684 const income2 = calculateIncome(200n * nominal);685 expect(totalStakedPerBlock[0].amount).to.equal(income1);686 expect(totalStakedPerBlock[1].amount).to.equal(income2);687688 const stakerBalance = await helper.balance.getSubstrateFull(staker.address);689 expect(stakerBalance).to.contain({miscFrozen: income1 + income2, feeFrozen: income1 + income2, reserved: 0n});690 expect(stakerBalance.free / nominal).to.eq(999n);691 });692693 itSub('shoud be paid for more than one period if payments was missed', async ({helper}) => {694 const staker = accounts.pop()!;695696 await helper.staking.stake(staker, 100n * nominal);697 // wait for two rewards are available:698 let [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});699 await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake.block) + LOCKING_PERIOD);700701 await helper.admin.payoutStakers(palletAdmin, 100);702 [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});703 const frozenBalanceShouldBe = calculateIncome(100n * nominal, 2);704 expect(stake.amount).to.be.equal(frozenBalanceShouldBe);705706 const stakerFullBalance = await helper.balance.getSubstrateFull(staker.address);707708 expect(stakerFullBalance).to.contain({reserved: 0n, feeFrozen: frozenBalanceShouldBe, miscFrozen: frozenBalanceShouldBe});709 });710711 itSub('should not be credited for pending-unstaked tokens', async ({helper}) => {712 // staker unstakes before rewards been payed713 const staker = accounts.pop()!;714 await helper.staking.stake(staker, 100n * nominal);715 const [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});716 await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake.block) + LOCKING_PERIOD);717 await helper.staking.unstakeAll(staker);718719 // so he did not receive any rewards720 const totalBalanceBefore = await helper.balance.getSubstrate(staker.address);721 await helper.admin.payoutStakers(palletAdmin, 100);722 const totalBalanceAfter = await helper.balance.getSubstrate(staker.address);723724 expect(totalBalanceBefore).to.be.equal(totalBalanceAfter);725 });726727 itSub('should bring compound interest', async ({helper}) => {728 const staker = accounts.pop()!;729730 await helper.staking.stake(staker, 100n * nominal);731732 let [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});733 await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake.block));734735 await helper.admin.payoutStakers(palletAdmin, 100);736 [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});737 expect(stake.amount).to.equal(calculateIncome(100n * nominal));738739 await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake.block) + LOCKING_PERIOD);740 await helper.admin.payoutStakers(palletAdmin, 100);741 [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});742 expect(stake.amount).to.equal(calculateIncome(100n * nominal, 2));743 });744745 itSub.skip('can be paid 1000 rewards in a time', async ({helper}) => {746 // all other stakes should be unstaked747 const oneHundredStakers = await helper.arrange.createCrowd(100, 1050n, donor);748749 // stakers stakes 10 times each750 for (let i = 0; i < 10; i++) {751 await Promise.all(oneHundredStakers.map(staker => helper.staking.stake(staker, 100n * nominal)));752 }753 await helper.wait.newBlocks(40);754 await helper.admin.payoutStakers(palletAdmin, 100);755 });756757 itSub.skip('can handle 40.000 rewards', async ({helper}) => {758 const crowdStakes = async () => {759 // each account in the crowd stakes 2 times760 const crowd = await helper.arrange.createCrowd(500, 300n, donor);761 await Promise.all(crowd.map(account => helper.staking.stake(account, 100n * nominal)));762 await Promise.all(crowd.map(account => helper.staking.stake(account, 100n * nominal)));763 //764 };765766 for (let i = 0; i < 40; i++) {767 await crowdStakes();768 }769770 // TODO pay rewards for some period771 });772 });773});774775function calculateIncome(base: bigint, iter = 0, calcPeriod: bigint = UNLOCKING_PERIOD): bigint {776 const DAY = 7200n;777 const ACCURACY = 1_000_000_000n;778 // 5n / 10_000n = 0.05% p/day779 const income = base + base * (ACCURACY * (calcPeriod * 5n) / (10_000n * DAY)) / ACCURACY ;780781 if (iter > 1) {782 return calculateIncome(income, iter - 1, calcPeriod);783 } else return income;784}785786function rewardAvailableInBlock(stakedInBlock: bigint) {787 if (stakedInBlock % LOCKING_PERIOD === 0n) return stakedInBlock + LOCKING_PERIOD;788 return (stakedInBlock - stakedInBlock % LOCKING_PERIOD) + (LOCKING_PERIOD * 2n);789}790791// Wait while promotion period less than specified block, to avoid boundary cases792// 0 if this should be the beginning of the period.793async function waitPromotionPeriodDoesntEnd(helper: DevUniqueHelper, waitBlockLessThan = LOCKING_PERIOD / 3n) {794 const relayBlockNumber = (await helper.callRpc('api.query.parachainSystem.validationData', [])).value.relayParentNumber.toNumber(); // await helper.chain.getLatestBlockNumber();795 const currentPeriodBlock = BigInt(relayBlockNumber) % LOCKING_PERIOD;796797 if (currentPeriodBlock > waitBlockLessThan) {798 await helper.wait.forRelayBlockNumber(BigInt(relayBlockNumber) + LOCKING_PERIOD - currentPeriodBlock);799 }800}1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617import {IKeyringPair} from '@polkadot/types/types';18import {19 itSub, usingPlaygrounds, Pallets, requirePalletsOrSkip, LOCKING_PERIOD, UNLOCKING_PERIOD,20} from '../../util';21import {DevUniqueHelper} from '../../util/playgrounds/unique.dev';22import {itEth, expect, SponsoringMode} from '../../eth/util';2324let donor: IKeyringPair;25let palletAdmin: IKeyringPair;26let nominal: bigint;27let palletAddress: string;28let accounts: IKeyringPair[];29// App promotion periods:30// LOCKING_PERIOD = 12 blocks of relay31// UNLOCKING_PERIOD = 6 blocks of parachain3233describe('App promotion', () => {34 before(async function () {35 await usingPlaygrounds(async (helper, privateKey) => {36 requirePalletsOrSkip(this, helper, [Pallets.AppPromotion]);37 donor = await privateKey({filename: __filename});38 palletAddress = helper.arrange.calculatePalletAddress('appstake');39 palletAdmin = await privateKey('//PromotionAdmin');40 nominal = helper.balance.getOneTokenNominal();4142 const accountBalances = new Array(100);43 accountBalances.fill(1000n);44 accounts = await helper.arrange.createAccounts(accountBalances, donor); // create accounts-pool to speed up tests45 });46 });4748 describe('stake extrinsic', () => {49 itSub('should "lock" staking balance, add it to "staked" map, and increase "totalStaked" amount', async ({helper}) => {50 const [staker, recepient] = [accounts.pop()!, accounts.pop()!];51 const totalStakedBefore = await helper.staking.getTotalStaked();5253 // Minimum stake amount is 100:54 await expect(helper.staking.stake(staker, 100n * nominal - 1n)).to.be.rejected;55 await helper.staking.stake(staker, 100n * nominal);5657 // Staker balance is: miscFrozen: 100, feeFrozen: 100, reserved: 0n...58 // ...so he can not transfer 90059 expect(await helper.balance.getSubstrateFull(staker.address)).to.contain({miscFrozen: 100n * nominal, feeFrozen: 100n * nominal, reserved: 0n});60 expect(await helper.balance.getLocked(staker.address)).to.deep.eq([{id: 'appstake', amount: 100n * nominal, reasons: 'All'}]);61 await expect(helper.balance.transferToSubstrate(staker, recepient.address, 900n * nominal)).to.be.rejectedWith('balances.LiquidityRestrictions');6263 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(100n * nominal);64 expect(await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n);65 // it is potentially flaky test. Promotion can credited some tokens. Maybe we need to use closeTo?66 expect(await helper.staking.getTotalStaked()).to.be.equal(totalStakedBefore + 100n * nominal); // total tokens amount staked in app-promotion increased676869 await helper.staking.stake(staker, 200n * nominal);70 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(300n * nominal);71 const totalStakedPerBlock = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});72 expect(totalStakedPerBlock[0].amount).to.equal(100n * nominal);73 expect(totalStakedPerBlock[1].amount).to.equal(200n * nominal);74 });7576 itSub('should allow to create maximum 10 stakes for account', async ({helper}) => {77 const [staker] = await helper.arrange.createAccounts([2000n], donor);78 for (let i = 0; i < 10; i++) {79 await helper.staking.stake(staker, 100n * nominal);80 }8182 // can have 10 stakes83 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(1000n * nominal);84 expect(await helper.staking.getTotalStakedPerBlock({Substrate: staker.address})).to.have.length(10);8586 await expect(helper.staking.stake(staker, 100n * nominal)).to.be.rejectedWith('appPromotion.NoPermission');8788 // After unstake can stake again89 await helper.staking.unstakeAll(staker);90 await helper.staking.stake(staker, 100n * nominal);91 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.equal(100n * nominal);92 });9394 itSub('should allow to stake() if balance is locked with different id', async ({helper}) => {95 const staker = accounts.pop()!;9697 // staker has tokens locked with vesting id:98 await helper.balance.vestedTransfer(donor, staker.address, {start: 0n, period: 1n, periodCount: 1n, perPeriod: 200n * nominal});99 expect(await helper.balance.getSubstrateFull(staker.address))100 .to.deep.contain({free: 1200n * nominal, miscFrozen: 200n * nominal, feeFrozen: 200n * nominal, reserved: 0n});101102 // Locked balance can be staked. staker can stake 1200 tokens (minus fee):103 await helper.staking.stake(staker, 1000n * nominal);104 await helper.staking.stake(staker, 199n * nominal);105 // check balances106 expect(await helper.balance.getLocked(staker.address)).to.deep.eq([{id: 'ormlvest', amount: 200n * nominal, reasons: 'All'}, {id: 'appstake', amount: 1199n * nominal, reasons: 'All'}]);107 expect(await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, miscFrozen: 1199n * nominal, feeFrozen: 1199n * nominal});108 expect(await helper.balance.getSubstrate(staker.address) / nominal).to.eq(1199n);109 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.eq(1199n * nominal);110111 // staker can unstake112 await helper.staking.unstakeAll(staker);113 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.eq(1199n * nominal);114 const [pendingUnstake] = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});115 await helper.wait.forParachainBlockNumber(pendingUnstake.block);116117 // check balances118 expect(await helper.balance.getLocked(staker.address)).to.deep.eq([{id: 'ormlvest', amount: 200n * nominal, reasons: 'All'}]);119 expect(await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, miscFrozen: 200n * nominal, feeFrozen: 200n * nominal});120 expect(await helper.balance.getSubstrate(staker.address) / nominal).to.eq(1199n);121 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.eq(0n);122123 // staker can transfer balances now124 await helper.balance.transferToSubstrate(staker, donor.address, 900n * nominal);125 });126127 itSub('should not allow to stake(), if stake amount is more than total free balance minus locked by staking', async ({helper}) => {128 const staker = accounts.pop()!;129130 // Can't stake full balance because Alice needs to pay some fee131 await expect(helper.staking.stake(staker, 1000n * nominal)).to.be.rejected; // With('Arithmetic')132 await helper.staking.stake(staker, 500n * nominal);133134 // Can't stake 500 tkn because Alice has Less than 500 transferable;135 await expect(helper.staking.stake(staker, 500n * nominal)).to.be.rejected; // With('Arithmetic');136 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(500n * nominal);137 });138139 itSub('for different accounts in one block is possible', async ({helper}) => {140 const crowd = [accounts.pop()!, accounts.pop()!, accounts.pop()!, accounts.pop()!];141142 const crowdStartsToStake = crowd.map(user => helper.staking.stake(user, 100n * nominal));143 await expect(Promise.all(crowdStartsToStake)).to.be.fulfilled;144145 const crowdStakes = await Promise.all(crowd.map(address => helper.staking.getTotalStaked({Substrate: address.address})));146 expect(crowdStakes).to.deep.equal([100n * nominal, 100n * nominal, 100n * nominal, 100n * nominal]);147 });148 });149150 describe('Unstaking', () => {151 [152 {method: 'unstakeAll' as const},153 {method: 'unstakePartial' as const},154 ].map(testCase => {155 itSub(`[${testCase.method}] should move tokens to "pendingUnstake" and subtract it from totalStaked`, async ({helper}) => {156 const [staker, recepient] = [accounts.pop()!, accounts.pop()!];157 const totalStakedBefore = await helper.staking.getTotalStaked();158 const STAKE_AMOUNT = 900n * nominal;159160 await helper.staking.stake(staker, STAKE_AMOUNT);161 testCase.method === 'unstakeAll'162 ? await helper.staking.unstakeAll(staker)163 : await helper.staking.unstakePartial(staker, STAKE_AMOUNT);164165 // Right after unstake tokens are still locked166 expect(await helper.staking.getStakesNumber({Substrate: staker.address})).to.eq(0);167 expect(await helper.balance.getLocked(staker.address)).to.deep.eq([{id: 'appstake', amount: STAKE_AMOUNT, reasons: 'All'}]);168 expect(await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, miscFrozen: STAKE_AMOUNT, feeFrozen: STAKE_AMOUNT});169 // Staker can not transfer170 await expect(helper.balance.transferToSubstrate(staker, recepient.address, 100n * nominal)).to.be.rejectedWith('balances.LiquidityRestrictions');171 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(STAKE_AMOUNT);172 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(0n);173 expect(await helper.staking.getTotalStaked()).to.be.equal(totalStakedBefore);174 });175 });176177 [178 {method: 'unstakeAll' as const},179 {method: 'unstakePartial' as const},180 ].map(testCase => {181 itSub(`[${testCase.method}] should unlock balance after unlocking period ends and remove it from "pendingUnstake"`, async ({helper}) => {182 const staker = accounts.pop()!;183 await helper.staking.stake(staker, 100n * nominal);184 testCase.method === 'unstakeAll'185 ? await helper.staking.unstakeAll(staker)186 : await helper.staking.unstakePartial(staker, 100n * nominal);187 const [pendingUnstake] = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});188189 // Wait for unstaking period. Balance now free ~1000; reserved, frozen, miscFrozeb: 0n190 await helper.wait.forParachainBlockNumber(pendingUnstake.block);191 expect(await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, miscFrozen: 0n, feeFrozen: 0n});192 expect(await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n);193194 // staker can transfer:195 await helper.balance.transferToSubstrate(staker, donor.address, 998n * nominal);196 expect(await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(1n);197 });198 });199200 [201 {method: 'unstakeAll' as const},202 {method: 'unstakePartial' as const},203 ].map(testCase => {204 itSub(`[${testCase.method}] should successfully unstake multiple stakes`, async ({helper}) => {205 const staker = accounts.pop()!;206 await helper.staking.stake(staker, 100n * nominal);207 await helper.staking.stake(staker, 200n * nominal);208 await helper.staking.stake(staker, 300n * nominal);209210 // staked: [100, 200, 300]; unstaked: 0211 let totalPendingUnstake = await helper.staking.getPendingUnstake({Substrate: staker.address});212 let pendingUnstake = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});213 let stakes = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});214 expect(totalPendingUnstake).to.be.deep.equal(0n);215 expect(pendingUnstake).to.be.deep.equal([]);216 expect(stakes[0].amount).to.equal(100n * nominal);217 expect(stakes[1].amount).to.equal(200n * nominal);218 expect(stakes[2].amount).to.equal(300n * nominal);219220 // Can unstake multiple stakes221 testCase.method === 'unstakeAll'222 ? await helper.staking.unstakeAll(staker)223 : await helper.staking.unstakePartial(staker, 600n * nominal);224225 pendingUnstake = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});226 totalPendingUnstake = await helper.staking.getPendingUnstake({Substrate: staker.address});227 stakes = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});228 expect(totalPendingUnstake).to.be.equal(600n * nominal);229 expect(stakes).to.be.deep.equal([]);230 expect(pendingUnstake[0].amount).to.equal(600n * nominal);231232 expect (await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, feeFrozen: 600n * nominal, miscFrozen: 600n * nominal});233 expect (await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n);234 await helper.wait.forParachainBlockNumber(pendingUnstake[0].block);235 expect (await helper.balance.getSubstrateFull(staker.address)).to.deep.contain({reserved: 0n, feeFrozen: 0n, miscFrozen: 0n});236 expect (await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n);237 });238 });239240 [241 {method: 'unstakeAll' as const},242 {method: 'unstakePartial' as const},243 ].map(testCase => {244 itSub(`[${testCase.method}] should not have any effects if no active stakes`, async ({helper}) => {245 const staker = accounts.pop()!;246247 // unstake has no effect if no stakes at all248 testCase.method === 'unstakeAll'249 ? await helper.staking.unstakeAll(staker)250 : await expect(helper.staking.unstakePartial(staker, 100n * nominal)).to.be.rejectedWith('appPromotion.InsufficientStakedBalance');251252 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(0n);253 expect(await helper.balance.getSubstrate(staker.address) / nominal).to.be.equal(999n); // TODO bigint closeTo helper254255 // TODO stake() unstake() waitUnstaked() unstake();256257 // can't unstake if there are only pendingUnstakes258 await helper.staking.stake(staker, 100n * nominal);259260 if (testCase.method === 'unstakeAll') {261 await helper.staking.unstakeAll(staker);262 await helper.staking.unstakeAll(staker);263 } else {264 await helper.staking.unstakePartial(staker, 100n * nominal);265 await expect(helper.staking.unstakePartial(staker, 100n * nominal)).to.be.rejectedWith('appPromotion.InsufficientStakedBalance');266 }267268 expect(await helper.staking.getStakesNumber({Substrate: staker.address})).to.eq(0);269 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(100n * nominal);270 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(0n);271 });272 });273274 [275 {method: 'unstakeAll' as const},276 {method: 'unstakePartial' as const},277 ].map(testCase => {278 itSub(`[${testCase.method}] should create different pending-unlock for each unlocking stake`, async ({helper}) => {279 const staker = accounts.pop()!;280 await helper.staking.stake(staker, 100n * nominal);281 testCase.method === 'unstakeAll'282 ? await helper.staking.unstakeAll(staker)283 : await helper.staking.unstakePartial(staker, 100n * nominal);284 await helper.staking.stake(staker, 120n * nominal);285 testCase.method === 'unstakeAll'286 ? await helper.staking.unstakeAll(staker)287 : await helper.staking.unstakePartial(staker, 120n * nominal);288289 const unstakingPerBlock = await helper.staking.getPendingUnstakePerBlock({Substrate: staker.address});290 expect(unstakingPerBlock).has.length(2);291 expect(unstakingPerBlock[0].amount).to.equal(100n * nominal);292 expect(unstakingPerBlock[1].amount).to.equal(120n * nominal);293 expect(await helper.staking.getStakesNumber({Substrate: staker.address})).to.equal(0);294 });295 });296297 [298 {method: 'unstakeAll' as const},299 {method: 'unstakePartial' as const},300 ].map(testCase => {301 itSub(`[${testCase.method}] should be possible for 3 accounts in one block`, async ({helper}) => {302 const stakers = [accounts.pop()!, accounts.pop()!, accounts.pop()!];303304 await Promise.all(stakers.map(staker => helper.staking.stake(staker, 100n * nominal)));305 await Promise.all(stakers.map(staker => {306 return testCase.method === 'unstakeAll'307 ? helper.staking.unstakeAll(staker)308 : helper.staking.unstakePartial(staker, 100n * nominal);309 }));310311 await Promise.all(stakers.map(async (staker) => {312 expect(await helper.staking.getPendingUnstake({Substrate: staker.address})).to.be.equal(100n * nominal);313 expect(await helper.staking.getTotalStaked({Substrate: staker.address})).to.be.equal(0n);314 }));315 });316 });317318 itSub('should not be possible for more than 3 accounts in one block', async ({helper}) => {319 if (!await helper.arrange.isDevNode()) {320 const stakers = await helper.arrange.createAccounts([200n,200n,200n,200n,200n,200n,200n,200n,200n,200n], donor);321322 await Promise.all(stakers.map(staker => helper.staking.stake(staker, 100n * nominal)));323 const unstakingResults = await Promise.allSettled(stakers.map((staker, i) => {324 return i % 2 === 0325 ? helper.staking.unstakeAll(staker)326 : helper.staking.unstakePartial(staker, 100n * nominal);327 }));328329 const successfulUnstakes = unstakingResults.filter(result => result.status === 'fulfilled');330 expect(successfulUnstakes).to.have.length(3);331 }332 });333 });334335 describe('collection sponsoring', () => {336 itSub('should actually sponsor transactions', async ({helper}) => {337 const api = helper.getApi();338 const [collectionOwner, tokenSender, receiver] = [accounts.pop()!, accounts.pop()!, accounts.pop()!];339 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'Name', description: 'Description', tokenPrefix: 'Prefix', limits: {sponsorTransferTimeout: 0}});340 const token = await collection.mintToken(collectionOwner, {Substrate: tokenSender.address});341 await helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId));342 const palletBalanceBefore = await helper.balance.getSubstrate(palletAddress);343344 await token.transfer(tokenSender, {Substrate: receiver.address});345 expect (await token.getOwner()).to.be.deep.equal({Substrate: receiver.address});346 const palletBalanceAfter = await helper.balance.getSubstrate(palletAddress);347348 // senders balance the same, transaction has sponsored349 expect (await helper.balance.getSubstrate(tokenSender.address)).to.be.equal(1000n * nominal);350 expect (palletBalanceBefore > palletBalanceAfter).to.be.true;351 });352353 itSub('can not be set by non admin', async ({helper}) => {354 const api = helper.getApi();355 const [collectionOwner, nonAdmin] = [accounts.pop()!, accounts.pop()!];356357 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});358359 await expect(helper.signTransaction(nonAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId))).to.be.rejected;360 expect((await collection.getData())?.raw.sponsorship).to.equal('Disabled');361 });362363 itSub('should set pallet address as confirmed admin', async ({helper}) => {364 const api = helper.getApi();365 const [collectionOwner, oldSponsor] = [accounts.pop()!, accounts.pop()!];366367 // Can set sponsoring for collection without sponsor368 const collectionWithoutSponsor = await helper.nft.mintCollection(collectionOwner, {name: 'No-sponsor', description: 'New Collection', tokenPrefix: 'Promotion'});369 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionWithoutSponsor.collectionId))).to.be.fulfilled;370 expect((await collectionWithoutSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});371372 // Can set sponsoring for collection with unconfirmed sponsor373 const collectionWithUnconfirmedSponsor = await helper.nft.mintCollection(collectionOwner, {name: 'Unconfirmed', description: 'New Collection', tokenPrefix: 'Promotion', pendingSponsor: oldSponsor.address});374 expect((await collectionWithUnconfirmedSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Unconfirmed: oldSponsor.address});375 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionWithUnconfirmedSponsor.collectionId))).to.be.fulfilled;376 expect((await collectionWithUnconfirmedSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});377378 // Can set sponsoring for collection with confirmed sponsor379 const collectionWithConfirmedSponsor = await helper.nft.mintCollection(collectionOwner, {name: 'Confirmed', description: 'New Collection', tokenPrefix: 'Promotion', pendingSponsor: oldSponsor.address});380 await collectionWithConfirmedSponsor.confirmSponsorship(oldSponsor);381 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionWithConfirmedSponsor.collectionId))).to.be.fulfilled;382 expect((await collectionWithConfirmedSponsor.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});383 });384385 itSub('can be overwritten by collection owner', async ({helper}) => {386 const api = helper.getApi();387 const [collectionOwner, newSponsor] = [accounts.pop()!, accounts.pop()!];388 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});389 const collectionId = collection.collectionId;390391 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionId))).to.be.fulfilled;392393 // Collection limits still can be changed by the owner394 expect(await collection.setLimits(collectionOwner, {sponsorTransferTimeout: 0})).to.be.true;395 expect((await collection.getData())?.raw.limits.sponsorTransferTimeout).to.be.equal(0);396 expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});397398 // Collection sponsor can be changed too399 expect((await collection.setSponsor(collectionOwner, newSponsor.address))).to.be.true;400 expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Unconfirmed: newSponsor.address});401 });402403 itSub('should not overwrite collection limits set by the owner earlier', async ({helper}) => {404 const api = helper.getApi();405 const limits = {ownerCanDestroy: true, ownerCanTransfer: true, sponsorTransferTimeout: 0};406 const collectionWithLimits = await helper.nft.mintCollection(accounts.pop()!, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion', limits});407408 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collectionWithLimits.collectionId))).to.be.fulfilled;409 expect((await collectionWithLimits.getData())?.raw.limits).to.be.deep.contain(limits);410 });411412 itSub('should reject transaction if collection doesn\'t exist', async ({helper}) => {413 const api = helper.getApi();414 const collectionOwner = accounts.pop()!;415416 // collection has never existed417 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(999999999))).to.be.rejected;418 // collection has been burned419 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});420 await collection.burn(collectionOwner);421422 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId))).to.be.rejected;423 });424 });425426 describe('stopSponsoringCollection', () => {427 itSub('can not be called by non-admin', async ({helper}) => {428 const api = helper.getApi();429 const [collectionOwner, nonAdmin] = [accounts.pop()!, accounts.pop()!];430 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});431432 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId))).to.be.fulfilled;433434 await expect(helper.signTransaction(nonAdmin, api.tx.appPromotion.stopSponsoringCollection(collection.collectionId))).to.be.rejected;435 expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: palletAddress});436 });437438 itSub('should set sponsoring as disabled', async ({helper}) => {439 const api = helper.getApi();440 const [collectionOwner, recepient] = [accounts.pop()!, accounts.pop()!];441 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion', limits: {sponsorTransferTimeout: 0}});442 const token = await collection.mintToken(collectionOwner, {Substrate: collectionOwner.address});443444 await helper.signTransaction(palletAdmin, api.tx.appPromotion.sponsorCollection(collection.collectionId));445 await helper.signTransaction(palletAdmin, api.tx.appPromotion.stopSponsoringCollection(collection.collectionId));446447 expect((await collection.getData())?.raw.sponsorship).to.be.equal('Disabled');448449 // Transactions are not sponsored anymore:450 const ownerBalanceBefore = await helper.balance.getSubstrate(collectionOwner.address);451 await token.transfer(collectionOwner, {Substrate: recepient.address});452 const ownerBalanceAfter = await helper.balance.getSubstrate(collectionOwner.address);453 expect(ownerBalanceAfter < ownerBalanceBefore).to.be.equal(true);454 });455456 itSub('should not affect collection which is not sponsored by pallete', async ({helper}) => {457 const api = helper.getApi();458 const collectionOwner = accounts.pop()!;459 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion', pendingSponsor: collectionOwner.address});460 await collection.confirmSponsorship(collectionOwner);461462 await expect(helper.signTransaction(palletAdmin, api.tx.appPromotion.stopSponsoringCollection(collection.collectionId))).to.be.rejected;463464 expect((await collection.getData())?.raw.sponsorship).to.be.deep.equal({Confirmed: collectionOwner.address});465 });466467 itSub('should reject transaction if collection does not exist', async ({helper}) => {468 const collectionOwner = accounts.pop()!;469 const collection = await helper.nft.mintCollection(collectionOwner, {name: 'New', description: 'New Collection', tokenPrefix: 'Promotion'});470471 await collection.burn(collectionOwner);472 await expect(helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.stopSponsoringCollection', [collection.collectionId], true)).to.be.rejectedWith('common.CollectionNotFound');473 await expect(helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.stopSponsoringCollection', [999_999_999], true)).to.be.rejectedWith('common.CollectionNotFound');474 });475 });476477 describe('contract sponsoring', () => {478 itEth('should set palletes address as a sponsor', async ({helper}) => {479 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();480 const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);481 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);482483 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address]);484485 expect(await contractHelper.methods.hasSponsor(flipper.options.address).call()).to.be.true;486 expect((await helper.callRpc('api.query.evmContractHelpers.owner', [flipper.options.address])).toJSON()).to.be.equal(contractOwner);487 expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({488 confirmed: {489 substrate: palletAddress,490 },491 });492 });493494 itEth('should overwrite sponsoring mode and existed sponsor', async ({helper}) => {495 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();496 const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);497 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);498499 await expect(contractHelper.methods.selfSponsoredEnable(flipper.options.address).send()).to.be.fulfilled;500501 // Contract is self sponsored502 expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.be.deep.equal({503 confirmed: {504 ethereum: flipper.options.address.toLowerCase(),505 },506 });507508 // set promotion sponsoring509 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true);510511 // new sponsor is pallet address512 expect(await contractHelper.methods.hasSponsor(flipper.options.address).call()).to.be.true;513 expect((await helper.callRpc('api.query.evmContractHelpers.owner', [flipper.options.address])).toJSON()).to.be.equal(contractOwner);514 expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({515 confirmed: {516 substrate: palletAddress,517 },518 });519 });520521 itEth('can be overwritten by contract owner', async ({helper}) => {522 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();523 const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);524 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);525526 // contract sponsored by pallet527 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true);528529 // owner sets self sponsoring530 await expect(contractHelper.methods.selfSponsoredEnable(flipper.options.address).send()).to.be.not.rejected;531532 expect(await contractHelper.methods.hasSponsor(flipper.options.address).call()).to.be.true;533 expect((await helper.callRpc('api.query.evmContractHelpers.owner', [flipper.options.address])).toJSON()).to.be.equal(contractOwner);534 expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({535 confirmed: {536 ethereum: flipper.options.address.toLowerCase(),537 },538 });539 });540541 itEth('can not be set by non admin', async ({helper}) => {542 const nonAdmin = accounts.pop()!;543 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();544 const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);545 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);546547 await expect(contractHelper.methods.selfSponsoredEnable(flipper.options.address).send()).to.be.fulfilled;548549 // nonAdmin calls sponsorContract550 await expect(helper.executeExtrinsic(nonAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true)).to.be.rejectedWith('appPromotion.NoPermission');551552 // contract still self-sponsored553 expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({554 confirmed: {555 ethereum: flipper.options.address.toLowerCase(),556 },557 });558 });559560 itEth('should actually sponsor transactions', async ({helper}) => {561 // Contract caller562 const caller = await helper.eth.createAccountWithBalance(donor, 1000n);563 const palletBalanceBefore = await helper.balance.getSubstrate(palletAddress);564565 // Deploy flipper566 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();567 const flipper = await helper.eth.deployFlipper(contractOwner); // await deployFlipper(web3, contractOwner);568 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);569570 // Owner sets to sponsor every tx571 await contractHelper.methods.setSponsoringRateLimit(flipper.options.address, 0).send({from: contractOwner});572 await contractHelper.methods.setSponsoringMode(flipper.options.address, SponsoringMode.Generous).send({from: contractOwner});573 await helper.eth.transferBalanceFromSubstrate(donor, flipper.options.address, 1000n); // transferBalanceToEth(api, alice, flipper.options.address, 1000n);574575 // Set promotion to the Flipper576 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true);577578 // Caller calls Flipper579 await flipper.methods.flip().send({from: caller});580 expect(await flipper.methods.getValue().call()).to.be.true;581582 // The contracts and caller balances have not changed583 const callerBalance = await helper.balance.getEthereum(caller);584 const contractBalanceAfter = await helper.balance.getEthereum(flipper.options.address);585 expect(callerBalance).to.be.equal(1000n * nominal);586 expect(1000n * nominal === contractBalanceAfter).to.be.true;587588 // The pallet balance has decreased589 const palletBalanceAfter = await helper.balance.getSubstrate(palletAddress);590 expect(palletBalanceAfter < palletBalanceBefore).to.be.true;591 });592 });593594 describe('stopSponsoringContract', () => {595 itEth('should remove pallet address from contract sponsors', async ({helper}) => {596 const caller = await helper.eth.createAccountWithBalance(donor, 1000n);597 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();598 const flipper = await helper.eth.deployFlipper(contractOwner);599 await helper.eth.transferBalanceFromSubstrate(donor, flipper.options.address);600 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);601602 await contractHelper.methods.setSponsoringMode(flipper.options.address, SponsoringMode.Generous).send({from: contractOwner});603 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address], true);604 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.stopSponsoringContract', [flipper.options.address], true);605606 expect(await contractHelper.methods.hasSponsor(flipper.options.address).call()).to.be.false;607 expect((await helper.callRpc('api.query.evmContractHelpers.owner', [flipper.options.address])).toJSON()).to.be.equal(contractOwner);608 expect((await helper.callRpc('api.query.evmContractHelpers.sponsoring', [flipper.options.address])).toJSON()).to.deep.equal({609 disabled: null,610 });611612 await flipper.methods.flip().send({from: caller});613 expect(await flipper.methods.getValue().call()).to.be.true;614615 const callerBalance = await helper.balance.getEthereum(caller);616 const contractBalanceAfter = await helper.balance.getEthereum(flipper.options.address);617618 // caller payed for call619 expect(1000n * nominal > callerBalance).to.be.true;620 expect(contractBalanceAfter).to.be.equal(100n * nominal);621 });622623 itEth('can not be called by non-admin', async ({helper}) => {624 const nonAdmin = accounts.pop()!;625 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();626 const flipper = await helper.eth.deployFlipper(contractOwner);627628 await helper.executeExtrinsic(palletAdmin, 'api.tx.appPromotion.sponsorContract', [flipper.options.address]);629 await expect(helper.executeExtrinsic(nonAdmin, 'api.tx.appPromotion.stopSponsoringContract', [flipper.options.address]))630 .to.be.rejectedWith(/appPromotion\.NoPermission/);631 });632633 itEth('should not affect a contract which is not sponsored by pallete', async ({helper}) => {634 const nonAdmin = accounts.pop()!;635 const contractOwner = (await helper.eth.createAccountWithBalance(donor, 1000n)).toLowerCase();636 const flipper = await helper.eth.deployFlipper(contractOwner);637 const contractHelper = await helper.ethNativeContract.contractHelpers(contractOwner);638 await expect(contractHelper.methods.selfSponsoredEnable(flipper.options.address).send()).to.be.fulfilled;639640 await expect(helper.executeExtrinsic(nonAdmin, 'api.tx.appPromotion.stopSponsoringContract', [flipper.options.address], true)).to.be.rejectedWith('appPromotion.NoPermission');641 });642 });643644 describe('payoutStakers', () => {645 itSub('can not be called by non admin', async ({helper}) => {646 const nonAdmin = accounts.pop()!;647 await expect(helper.admin.payoutStakers(nonAdmin, 100)).to.be.rejectedWith('appPromotion.NoPermission');648 });649650 itSub('should increase total staked', async ({helper}) => {651 const staker = accounts.pop()!;652 const totalStakedBefore = await helper.staking.getTotalStaked();653 await helper.staking.stake(staker, 100n * nominal);654655 // Wait for rewards and pay656 const [stakedInBlock] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});657 await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stakedInBlock.block));658 const totalPayout = (await helper.admin.payoutStakers(palletAdmin, 100)).reduce((prev, payout) => prev + payout.payout, 0n);659660 const totalStakedAfter = await helper.staking.getTotalStaked();661 expect(totalStakedAfter).to.equal(totalStakedBefore + (100n * nominal) + totalPayout);662 // staker can unstake663 await helper.staking.unstakeAll(staker);664 expect(await helper.staking.getTotalStaked()).to.be.equal(totalStakedAfter - calculateIncome(100n * nominal));665 });666667 itSub('should credit 0.05% for staking period', async ({helper}) => {668 const staker = accounts.pop()!;669670 await waitPromotionPeriodDoesntEnd(helper);671672 await helper.staking.stake(staker, 100n * nominal);673 await helper.staking.stake(staker, 200n * nominal);674675 // wait rewards are available:676 const [_stake1, stake2] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});677 await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake2.block));678679 const payoutToStaker = (await helper.admin.payoutStakers(palletAdmin, 100)).find((payout) => payout.staker === staker.address)?.payout;680 expect(payoutToStaker + 300n * nominal).to.equal(calculateIncome(300n * nominal));681682 const totalStakedPerBlock = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});683 const income1 = calculateIncome(100n * nominal);684 const income2 = calculateIncome(200n * nominal);685 expect(totalStakedPerBlock[0].amount).to.equal(income1);686 expect(totalStakedPerBlock[1].amount).to.equal(income2);687688 const stakerBalance = await helper.balance.getSubstrateFull(staker.address);689 expect(stakerBalance).to.contain({miscFrozen: income1 + income2, feeFrozen: income1 + income2, reserved: 0n});690 expect(stakerBalance.free / nominal).to.eq(999n);691 });692693 itSub('shoud be paid for more than one period if payments was missed', async ({helper}) => {694 const staker = accounts.pop()!;695696 await helper.staking.stake(staker, 100n * nominal);697 // wait for two rewards are available:698 let [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});699 await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake.block) + LOCKING_PERIOD);700701 await helper.admin.payoutStakers(palletAdmin, 100);702 [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});703 const frozenBalanceShouldBe = calculateIncome(100n * nominal, 2);704 expect(stake.amount).to.be.equal(frozenBalanceShouldBe);705706 const stakerFullBalance = await helper.balance.getSubstrateFull(staker.address);707708 expect(stakerFullBalance).to.contain({reserved: 0n, feeFrozen: frozenBalanceShouldBe, miscFrozen: frozenBalanceShouldBe});709 });710711 itSub('should not be credited for pending-unstaked tokens', async ({helper}) => {712 // staker unstakes before rewards been payed713 const staker = accounts.pop()!;714 await helper.staking.stake(staker, 100n * nominal);715 const [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});716 await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake.block) + LOCKING_PERIOD);717 await helper.staking.unstakeAll(staker);718719 // so he did not receive any rewards720 const totalBalanceBefore = await helper.balance.getSubstrate(staker.address);721 await helper.admin.payoutStakers(palletAdmin, 100);722 const totalBalanceAfter = await helper.balance.getSubstrate(staker.address);723724 expect(totalBalanceBefore).to.be.equal(totalBalanceAfter);725 });726727 itSub('should bring compound interest', async ({helper}) => {728 const staker = accounts.pop()!;729730 await helper.staking.stake(staker, 100n * nominal);731732 let [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});733 await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake.block));734735 await helper.admin.payoutStakers(palletAdmin, 100);736 [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});737 expect(stake.amount).to.equal(calculateIncome(100n * nominal));738739 await helper.wait.forRelayBlockNumber(rewardAvailableInBlock(stake.block) + LOCKING_PERIOD);740 await helper.admin.payoutStakers(palletAdmin, 100);741 [stake] = await helper.staking.getTotalStakedPerBlock({Substrate: staker.address});742 expect(stake.amount).to.equal(calculateIncome(100n * nominal, 2));743 });744745 itSub.skip('can be paid 1000 rewards in a time', async ({helper}) => {746 // all other stakes should be unstaked747 const oneHundredStakers = await helper.arrange.createCrowd(100, 1050n, donor);748749 // stakers stakes 10 times each750 for (let i = 0; i < 10; i++) {751 await Promise.all(oneHundredStakers.map(staker => helper.staking.stake(staker, 100n * nominal)));752 }753 await helper.wait.newBlocks(40);754 await helper.admin.payoutStakers(palletAdmin, 100);755 });756757 itSub.skip('can handle 40.000 rewards', async ({helper}) => {758 const crowdStakes = async () => {759 // each account in the crowd stakes 2 times760 const crowd = await helper.arrange.createCrowd(500, 300n, donor);761 await Promise.all(crowd.map(account => helper.staking.stake(account, 100n * nominal)));762 await Promise.all(crowd.map(account => helper.staking.stake(account, 100n * nominal)));763 //764 };765766 for (let i = 0; i < 40; i++) {767 await crowdStakes();768 }769770 // TODO pay rewards for some period771 });772 });773});774775function calculateIncome(base: bigint, iter = 0, calcPeriod: bigint = UNLOCKING_PERIOD): bigint {776 const DAY = 7200n;777 const ACCURACY = 1_000_000_000n;778 // 5n / 10_000n = 0.05% p/day779 const income = base + base * (ACCURACY * (calcPeriod * 5n) / (10_000n * DAY)) / ACCURACY ;780781 if (iter > 1) {782 return calculateIncome(income, iter - 1, calcPeriod);783 } else return income;784}785786function rewardAvailableInBlock(stakedInBlock: bigint) {787 if (stakedInBlock % LOCKING_PERIOD === 0n) return stakedInBlock + LOCKING_PERIOD;788 return (stakedInBlock - stakedInBlock % LOCKING_PERIOD) + (LOCKING_PERIOD * 2n);789}790791// Wait while promotion period less than specified block, to avoid boundary cases792// 0 if this should be the beginning of the period.793async function waitPromotionPeriodDoesntEnd(helper: DevUniqueHelper, waitBlockLessThan = LOCKING_PERIOD / 3n) {794 const relayBlockNumber = (await helper.callRpc('api.query.parachainSystem.validationData', [])).value.relayParentNumber.toNumber(); // await helper.chain.getLatestBlockNumber();795 const currentPeriodBlock = BigInt(relayBlockNumber) % LOCKING_PERIOD;796797 if (currentPeriodBlock > waitBlockLessThan) {798 await helper.wait.forRelayBlockNumber(BigInt(relayBlockNumber) + LOCKING_PERIOD - currentPeriodBlock);799 }800}