fix timeout

Author: almogdepaz

chia/_tests/conftest.py

@@ -1274,11 +1274,6 @@ async def farmer_harvester_2_simulators_zero_bits_plot_filter(
     )
 
     async with AsyncExitStack() as async_exit_stack:
-        bt = await create_block_tools_async(
-            zero_bit_plot_filter_consts,
-            keychain=get_temp_keyring,
-        )
-
         config_overrides: dict[str, int] = {"full_node.max_sync_wait": 0}
 
         bts = [
@@ -1309,10 +1304,10 @@ async def farmer_harvester_2_simulators_zero_bits_plot_filter(
         ]
 
         [harvester_service], farmer_service, _ = await async_exit_stack.enter_async_context(
-            setup_farmer_solver_multi_harvester(bt, 1, tmp_path, bt.constants, start_services=True)
+            setup_farmer_solver_multi_harvester(bts[0], 1, tmp_path, bts[0].constants, start_services=True)
         )
 
-        yield farmer_service, harvester_service, simulators[0], simulators[1], bt
+        yield farmer_service, harvester_service, simulators[0], simulators[1], bts[0]
 
 
 @pytest.fixture(name="recording_web_server")

chia/_tests/farmer_harvester/test_third_party_harvesters.py

@@ -92,9 +92,9 @@ async def test_harvester_receive_source_signing_data(
         UnresolvedPeerInfo(str(full_node_service_1.self_hostname), full_node_service_1._server.get_port())
     )
 
-    await wait_until_node_type_connected(farmer.server, NodeType.FULL_NODE)
-    await wait_until_node_type_connected(farmer.server, NodeType.HARVESTER)  # Should already be connected
-    await wait_until_node_type_connected(full_node_1.server, NodeType.FULL_NODE)
+    await time_out_assert(60, node_type_connected, True, farmer.server, NodeType.FULL_NODE)
+    await time_out_assert(60, node_type_connected, True, farmer.server, NodeType.HARVESTER)
+    await time_out_assert(60, node_type_connected, True, full_node_1.server, NodeType.FULL_NODE)
 
     # Prepare test data
     blocks: list[FullBlock]
@@ -429,12 +429,11 @@ def log_has_new_text() -> bool:
     return False
 
 
-async def wait_until_node_type_connected(server: ChiaServer, node_type: NodeType) -> WSChiaConnection:
-    while True:
-        for peer in server.all_connections.values():
-            if peer.connection_type == node_type.value:
-                return peer
-        await asyncio.sleep(1)
+def node_type_connected(server: ChiaServer, node_type: NodeType) -> bool:
+    for peer in server.all_connections.values():
+        if peer.connection_type == node_type.value:
+            return True
+    return False
 
 
 def decode_sp(