Fix tests

Author: thewhaleking

bittensor_cli/src/commands/stake/add.py

@@ -88,7 +88,10 @@ async def safe_stake_extrinsic(
             },
         )
         extrinsic = await subtensor.substrate.create_signed_extrinsic(
-            call=call, keypair=wallet.coldkey, nonce=next_nonce, era={"period": era}
+            call=call,
+            keypair=wallet.coldkey,
+            nonce=next_nonce,
+            era={"period": era},
         )
         try:
             response = await subtensor.substrate.submit_extrinsic(

tests/e2e_tests/test_staking_sudo.py

@@ -112,9 +112,11 @@ def test_staking(local_chain, wallet_setup):
             "0.1",
             "--partial",
             "--no-prompt",
+            "--era",
+            "144",
         ],
     )
-    assert "✅ Finalized" in add_stake.stdout
+    assert "✅ Finalized" in add_stake.stdout, add_stake.stderr
 
     # Execute stake show for Alice's wallet
     show_stake = exec_command_alice(
@@ -157,6 +159,8 @@ def test_staking(local_chain, wallet_setup):
             "0.1",
             "--partial",
             "--no-prompt",
+            "--era",
+            "144",
         ],
     )
     assert "✅ Finalized" in remove_stake.stdout

tests/e2e_tests/test_unstaking.py

@@ -139,6 +139,8 @@ def test_unstaking(local_chain, wallet_setup):
                 "--partial",
                 "--tolerance",
                 "0.5",
+                "--era",
+                "144",
             ],
         )
         assert "✅ Finalized" in stake_result.stdout
@@ -184,6 +186,8 @@ def test_unstaking(local_chain, wallet_setup):
             "--partial",
             "--tolerance",
             "0.5",
+            "--era",
+            "144",
         ],
     )
     assert "✅ Finalized" in partial_unstake_netuid_2.stdout
@@ -229,6 +233,8 @@ def test_unstaking(local_chain, wallet_setup):
             "--all-alpha",
             "--no-prompt",
             "--verbose",
+            "--era",
+            "144",
         ],
     )
 
@@ -258,6 +264,8 @@ def test_unstaking(local_chain, wallet_setup):
                 "--partial",
                 "--tolerance",
                 "0.5",
+                "--era",
+                "144",
             ],
         )
         assert "✅ Finalized" in stake_result.stdout
@@ -277,6 +285,8 @@ def test_unstaking(local_chain, wallet_setup):
             "ws://127.0.0.1:9945",
             "--all",
             "--no-prompt",
+            "--era",
+            "144",
         ],
     )
     assert "✅ Finalized: Successfully unstaked all stakes from" in unstake_all.stdout

tests/e2e_tests/test_wallet_interactions.py

@@ -245,6 +245,8 @@ def test_wallet_transfer(local_chain, wallet_setup):
             "--amount",
             "100",
             "--no-prompt",
+            "--era",
+            "144",
         ],
     )