Fix CI: rustfmt wrap and forbid-saturating-math in mock

- cargo fmt wraps the add_stake() call in AddStakeRecycleV1/AddStakeBurnV1
- mock charge_weight replaces saturating_add with checked_add().unwrap()
  to satisfy the ForbidSaturatingMath custom lint

Author: LandynDev

chain-extensions/src/lib.rs

@@ -587,7 +587,8 @@ where
             }
             FunctionId::AddStakeRecycleV1 => {
                 let add_stake_weight =
-                    <<T as pallet_subtensor::Config>::WeightInfo as SubtensorWeightInfo>::add_stake();
+                    <<T as pallet_subtensor::Config>::WeightInfo as SubtensorWeightInfo>::add_stake(
+                    );
                 let recycle_weight =
                     <<T as pallet_subtensor::Config>::WeightInfo as SubtensorWeightInfo>::recycle_alpha();
 
@@ -643,7 +644,8 @@ where
             }
             FunctionId::AddStakeBurnV1 => {
                 let add_stake_weight =
-                    <<T as pallet_subtensor::Config>::WeightInfo as SubtensorWeightInfo>::add_stake();
+                    <<T as pallet_subtensor::Config>::WeightInfo as SubtensorWeightInfo>::add_stake(
+                    );
                 let burn_weight =
                     <<T as pallet_subtensor::Config>::WeightInfo as SubtensorWeightInfo>::burn_alpha();
 

chain-extensions/src/tests.rs

@@ -1286,10 +1286,11 @@ impl SubtensorExtensionEnv<AccountId> for MockEnv {
     }
 
     fn charge_weight(&mut self, weight: Weight) -> Result<(), DispatchError> {
-        let cumulative = self
-            .charged_weight
-            .unwrap_or_default()
-            .saturating_add(weight);
+        let prev = self.charged_weight.unwrap_or_default();
+        let cumulative = Weight::from_parts(
+            prev.ref_time().checked_add(weight.ref_time()).unwrap(),
+            prev.proof_size().checked_add(weight.proof_size()).unwrap(),
+        );
         if let Some(expected) = self.expected_weight
             && (cumulative.ref_time() > expected.ref_time()
                 || cumulative.proof_size() > expected.proof_size())