mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-09-05 02:24:21 -05:00
ded9f68583
* protocols|server: Define new harvester plot refreshing protocol messages * protocols: Bump `protocol_version` to `0.0.34` * tests: Introduce `setup_farmer_multi_harvester` Allows to run a test setup with 1 farmer and mutiple harvesters. * plotting: Add an initial plot loading indication to `PlotManager` * plotting|tests: Don't add removed duplicates to `total_result.removed` `PlotRefreshResult.removed` should only contain plots that were loaded properly before they were removed. It shouldn't contain e.g. removed duplicates or invalid plots since those are synced in an extra sync step and not as diff but as whole list every time. * harvester: Reset `PlotManager` on shutdown * plot_sync: Implement plot sync protocol * farmer|harvester: Integrate and enable plot sync * tests: Implement tests for the plot sync protocol * farmer|tests: Drop obsolete harvester caching code * setup: Add `chia.plot_sync` to packages * plot_sync: Type hints in `DeltaType` * plot_sync: Drop parameters in `super()` calls * plot_sync: Introduce `send_response` helper in `Receiver._process` * plot_sync: Add some parentheses Co-authored-by: Kyle Altendorf <sda@fstab.net> * plot_sync: Additional hint for a `Receiver.process_path_list` parameter * plot_sync: Force named parameters in `Receiver.process_path_list` * test: Fix fixtures after rebase * tests: Fix sorting after rebase * tests: Return type hint for `plot_sync_setup` * tests: Rename `WSChiaConnection` and move it in the outer scope * tests|plot_sync: More type hints * tests: Rework some delta tests * tests: Drop a `range` and iterate over the list directly * tests: Use the proper flags to overwrite * test: More missing duplicates tests * tests: Drop `ExpectedResult.reset` * tests: Reduce some asserts * tests: Add messages to some `assert False` statements * tests: Introduce `ErrorSimulation` enum in `test_sync_simulated.py` * tests: Use `secrects` instead of `Crypto.Random` * Fixes after rebase * Import from `typing_extensions` to support python 3.7 * Drop task name to support python 3.7 * Introduce `Sender.syncing`, `Sender.connected` and a log about the task * Add `tests/plot_sync/config.py` * Align the multi harvester fixture with what we do in other places * Update the workflows Co-authored-by: Kyle Altendorf <sda@fstab.net>
54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
import time
|
|
from dataclasses import dataclass
|
|
from secrets import token_bytes
|
|
from typing import Optional
|
|
|
|
from chia.harvester.harvester_api import Harvester
|
|
from chia.plot_sync.sender import Sender
|
|
from chia.protocols.harvester_protocol import PlotSyncIdentifier
|
|
from chia.server.start_service import Service
|
|
from chia.server.ws_connection import Message, NodeType
|
|
from chia.types.blockchain_format.sized_bytes import bytes32
|
|
from chia.util.ints import uint64
|
|
from tests.time_out_assert import time_out_assert
|
|
|
|
|
|
@dataclass
|
|
class WSChiaConnectionDummy:
|
|
connection_type: NodeType
|
|
peer_node_id: bytes32
|
|
peer_host: str = "localhost"
|
|
peer_port: int = 0
|
|
last_sent_message: Optional[Message] = None
|
|
|
|
async def send_message(self, message: Message) -> None:
|
|
self.last_sent_message = message
|
|
|
|
|
|
def get_dummy_connection(node_type: NodeType, peer_id: Optional[bytes32] = None) -> WSChiaConnectionDummy:
|
|
return WSChiaConnectionDummy(node_type, bytes32(token_bytes(32)) if peer_id is None else peer_id)
|
|
|
|
|
|
def plot_sync_identifier(current_sync_id: uint64, message_id: uint64) -> PlotSyncIdentifier:
|
|
return PlotSyncIdentifier(uint64(int(time.time())), current_sync_id, message_id)
|
|
|
|
|
|
async def start_harvester_service(harvester_service: Service) -> Harvester:
|
|
# Set the `last_refresh_time` of the plot manager to avoid initial plot loading
|
|
harvester: Harvester = harvester_service._node
|
|
harvester.plot_manager.last_refresh_time = time.time()
|
|
await harvester_service.start()
|
|
harvester.plot_manager.stop_refreshing() # type: ignore[no-untyped-call] # TODO, Add typing in PlotManager
|
|
|
|
assert harvester.plot_sync_sender._sync_id == 0
|
|
assert harvester.plot_sync_sender._next_message_id == 0
|
|
assert harvester.plot_sync_sender._last_sync_id == 0
|
|
assert harvester.plot_sync_sender._messages == []
|
|
|
|
def wait_for_farmer_connection(plot_sync_sender: Sender) -> bool:
|
|
return plot_sync_sender._connection is not None
|
|
|
|
await time_out_assert(10, wait_for_farmer_connection, True, harvester.plot_sync_sender)
|
|
|
|
return harvester
|