feat: solving staking bug

Author: functor-flow

pallets/torus0/src/lib.rs

@@ -229,6 +229,10 @@ pub mod pallet {
             amount: BalanceOf<T>,
         ) -> DispatchResult {
             let key = ensure_signed(origin)?;
+            ensure!(
+                amount >= crate::MinAllowedStake::<T>::get(),
+                crate::Error::<T>::StakeTooSmall
+            );
             stake::add_stake::<T>(key, agent_key, amount)
         }
 

pallets/torus0/src/stake.rs

@@ -11,11 +11,6 @@ pub fn add_stake<T: crate::Config>(
     staked: AccountIdOf<T>,
     amount: BalanceOf<T>,
 ) -> DispatchResult {
-    ensure!(
-        amount >= crate::MinAllowedStake::<T>::get(),
-        crate::Error::<T>::StakeTooSmall
-    );
-
     ensure!(
         agent::exists::<T>(&staked),
         crate::Error::<T>::AgentDoesNotExist
@@ -76,6 +71,8 @@ pub fn remove_stake<T: crate::Config>(
 
     let _ = <T as crate::Config>::Currency::deposit_creating(&key, amount);
 
+    crate::Pallet::<T>::deposit_event(crate::Event::<T>::StakeRemoved(key, agent_key, amount));
+
     Ok(())
 }
 

pallets/torus0/tests/stake.rs

@@ -1,6 +1,6 @@
 use pallet_torus0::{Error, MinAllowedStake, StakedBy, StakingTo, TotalStake};
 use polkadot_sdk::frame_support::{assert_err, traits::Currency};
-use test_utils::{assert_ok, pallet_governance, Balances, Test};
+use test_utils::{assert_ok, get_origin, pallet_governance, Balances, Test};
 
 #[test]
 fn add_stake_correctly() {
@@ -78,7 +78,7 @@ fn add_stake_without_the_minimum_stake() {
         assert_eq!(Balances::total_balance(&to), 0);
 
         assert_err!(
-            pallet_torus0::stake::add_stake::<Test>(from, to, stake_to_add - 1),
+            pallet_torus0::Pallet::<Test>::add_stake(get_origin(from), to, stake_to_add - 1),
             Error::<Test>::StakeTooSmall
         );