fix unit tests

Author: open-junius

pallets/subtensor/src/staking/remove_stake.rs

@@ -490,7 +490,10 @@ impl<T: Config> Pallet<T> {
             .collect::<Vec<_>>();
 
         for hot in hotkeys_in_subnet.iter() {
-            for ((cold, _), share_u64f64) in Alpha::<T>::iter_prefix((hot,)) {
+            for ((cold, this_netuid), share_u64f64) in Alpha::<T>::iter_prefix((hot,)) {
+                if this_netuid != netuid {
+                    continue;
+                }
                 keys_to_remove.push((hot.clone(), cold.clone()));
                 if !hotkeys_seen.contains(hot) {
                     hotkeys_seen.push(hot.clone());

pallets/subtensor/src/tests/networks.rs

@@ -106,6 +106,7 @@ fn dissolve_single_alpha_out_staker_gets_all_tao() {
         let pot: u64 = 99_999;
         SubnetTAO::<Test>::insert(net, TaoCurrency::from(pot));
         SubtensorModule::set_subnet_locked_balance(net, 0.into());
+        TotalHotkeyAlpha::<Test>::insert(s_hot, net, AlphaCurrency::from(5_000u64));
 
         // Cold-key balance before
         let before = SubtensorModule::get_coldkey_balance(&s_cold);
@@ -142,6 +143,9 @@ fn dissolve_two_stakers_pro_rata_distribution() {
         Alpha::<Test>::insert((s1_hot, s1_cold, net), U64F64::from_num(a1));
         Alpha::<Test>::insert((s2_hot, s2_cold, net), U64F64::from_num(a2));
 
+        TotalHotkeyAlpha::<Test>::insert(s1_hot, net, AlphaCurrency::from(a1 as u64));
+        TotalHotkeyAlpha::<Test>::insert(s2_hot, net, AlphaCurrency::from(a2 as u64));
+
         let pot: u64 = 10_000;
         SubnetTAO::<Test>::insert(net, TaoCurrency::from(pot));
         SubtensorModule::set_subnet_locked_balance(net, 5_000.into()); // owner refund path present; emission = 0
@@ -627,6 +631,7 @@ fn dissolve_alpha_out_but_zero_tao_no_rewards() {
         SubnetTAO::<Test>::insert(net, TaoCurrency::from(0)); // zero TAO
         SubtensorModule::set_subnet_locked_balance(net, TaoCurrency::from(0));
         Emission::<Test>::insert(net, Vec::<AlphaCurrency>::new());
+        TotalHotkeyAlpha::<Test>::insert(sh, net, AlphaCurrency::from(1_000u64));
 
         let before = SubtensorModule::get_coldkey_balance(&sc);
         assert_ok!(SubtensorModule::do_dissolve_network(net));
@@ -672,6 +677,9 @@ fn dissolve_rounding_remainder_distribution() {
         SubnetTAO::<Test>::insert(net, TaoCurrency::from(1)); // TAO pot = 1
         SubtensorModule::set_subnet_locked_balance(net, TaoCurrency::from(0));
 
+        TotalHotkeyAlpha::<Test>::insert(s1h, net, AlphaCurrency::from(3u64));
+        TotalHotkeyAlpha::<Test>::insert(s2h, net, AlphaCurrency::from(2u64));
+
         // Cold-key balances before
         let c1_before = SubtensorModule::get_coldkey_balance(&s1c);
         let c2_before = SubtensorModule::get_coldkey_balance(&s2c);
@@ -1874,6 +1882,7 @@ fn massive_dissolve_refund_and_reregistration_flow_is_lossless_and_cleans_state(
         // Ordered α snapshot per net at **pair granularity** (pre‑LP):
         let mut alpha_pairs_per_net: BTreeMap<NetUid, Vec<((U256, U256), u128)>> = BTreeMap::new();
 
+        
         // Register both hotkeys for each participating cold on each net and stake τ→α.
         for (ni, &net) in nets.iter().enumerate() {
             let participants = lp_sets_per_net[ni];