test: unsafe stake add

Author: SmartDever02

bittensor_cli/src/commands/stake/add.py

@@ -493,9 +493,19 @@ async def stake_extrinsic(
         price_idx = 0
         for hotkey in hotkeys_to_stake_to:
             for netuid in netuids:
+                # Safety check: if we've processed all items from the first loop, stop
                 if list_idx >= len(amounts_to_stake):
                     break
 
+                # Verify subnet exists (same check as first loop)
+                # If subnet doesn't exist, it was skipped in first loop, so list_idx won't advance
+                # We need to skip it here too to stay in sync
+                subnet_info = all_subnets.get(netuid)
+                if not subnet_info:
+                    # This netuid was skipped in first loop (doesn't exist)
+                    # Don't advance list_idx, just continue to next netuid
+                    continue
+
                 am = amounts_to_stake[list_idx]
                 curr = current_stake_balances[list_idx]
                 staking_address = hotkey[1]

tests/e2e_tests/test_batching.py

@@ -192,6 +192,33 @@ def test_batching(local_chain, wallet_setup):
         register_result.stdout
     )
 
+    # Register Bob in one subnet
+    register_result_subnet3 = exec_command_bob(
+        command="subnets",
+        sub_command="register",
+        extra_args=[
+            "--netuid",
+            "3",
+            "--wallet-path",
+            wallet_path_bob,
+            "--wallet-name",
+            wallet_bob.name,
+            "--hotkey",
+            wallet_bob.hotkey_str,
+            "--chain",
+            "ws://127.0.0.1:9945",
+            "--no-prompt",
+            "--era",
+            "30",
+        ],
+    )
+    assert "✅ Registered" in register_result_subnet3.stdout, (
+        register_result_subnet3.stderr
+    )
+    assert "Your extrinsic has been included" in register_result_subnet3.stdout, (
+        register_result_subnet3.stdout
+    )
+
     # Add stake to subnets
     multiple_netuids = [2, 3]
     stake_result = exec_command_bob(
@@ -211,9 +238,7 @@ def test_batching(local_chain, wallet_setup):
             "--chain",
             "ws://127.0.0.1:9945",
             "--no-prompt",
-            "--partial",
-            "--tolerance",
-            "1.0",  # Increased from 0.5 to 1.0 (100% tolerance) to handle price volatility in pipeline
+            "--unsafe",
             "--era",
             "144",
         ],
@@ -223,6 +248,8 @@ def test_batching(local_chain, wallet_setup):
         stake_result.stdout
     )
 
+    print(f"output: {stake_result.stdout}")
+
     # Verify extrinsic_id is unique (all operations should share the same extrinsic_id when batched)
     # Pattern matches: "Your extrinsic has been included as {block_number}-{extrinsic_index}"
     extrinsic_id_pattern = r"Your extrinsic has been included as (\d+-\d+)"