Please add comments (Chia-Network#5565)

* Please add comments * Fix issue * More blocks Co-authored-by: Adam Kelly <[email protected]> Co-authored-by: Mariano <[email protected]>
banwang811 · May 20, 2021 · 15d4938 · 15d4938
1 parent ac59ec5
commit 15d4938
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/wallet/test_wallet.py b/tests/wallet/test_wallet.py
@@ -558,7 +558,6 @@ async def test_wallet_tx_reorg(self, two_wallet_nodes):
         await server_3.start_client(PeerInfo(self_hostname, uint16(fn_server._port)), None)
         for i in range(0, num_blocks):
             await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph))
-        await time_out_assert(10, wallet_node.wallet_state_manager.blockchain.get_peak_height, 5)
 
         funds = sum(
             [calculate_pool_reward(uint32(i)) + calculate_base_farmer_reward(uint32(i)) for i in range(1, num_blocks)]
@@ -569,7 +568,7 @@ async def test_wallet_tx_reorg(self, two_wallet_nodes):
         await time_out_assert(5, wallet.get_confirmed_balance, funds)
         for i in range(0, 2):
             await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(32 * b"0"))
-        await time_out_assert(10, wallet_2.get_confirmed_balance, 1000)
+        await time_out_assert(5, wallet_2.get_confirmed_balance, 1000)
 
         await time_out_assert(5, wallet_node.wallet_state_manager.blockchain.get_peak_height, 7)
 
@@ -584,7 +583,7 @@ async def test_wallet_tx_reorg(self, two_wallet_nodes):
         await time_out_assert(7, full_node_api.full_node.blockchain.get_peak_height, 10)
         await time_out_assert(7, wallet_node.wallet_state_manager.blockchain.get_peak_height, 10)
 
-        for i in range(0, num_blocks):
+        for i in range(0, num_blocks * 3):
             await asyncio.sleep(1)
             await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(32 * b"0"))