-
Notifications
You must be signed in to change notification settings - Fork 2k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
separte BlockCachee into the (simple) production use case and the (co…
…mplex) test use case of mocking a blockchain object
- Loading branch information
Showing
7 changed files
with
190 additions
and
117 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,112 @@ | ||
| from __future__ import annotations | ||
|
|
||
| import logging | ||
| from typing import Dict, List, Optional | ||
|
|
||
| from chia.consensus.block_record import BlockRecord | ||
| from chia.consensus.blockchain_interface import BlockchainInterface | ||
| from chia.types.blockchain_format.sized_bytes import bytes32 | ||
| from chia.types.blockchain_format.sub_epoch_summary import SubEpochSummary | ||
| from chia.types.header_block import HeaderBlock | ||
| from chia.types.weight_proof import SubEpochChallengeSegment, SubEpochSegments | ||
| from chia.util.ints import uint32 | ||
|
|
||
|
|
||
| class BlockchainMock(BlockchainInterface): | ||
| def __init__( | ||
| self, | ||
| blocks: Dict[bytes32, BlockRecord], | ||
| headers: Optional[Dict[bytes32, HeaderBlock]] = None, | ||
| height_to_hash: Optional[Dict[uint32, bytes32]] = None, | ||
| sub_epoch_summaries: Optional[Dict[uint32, SubEpochSummary]] = None, | ||
| ): | ||
| if sub_epoch_summaries is None: | ||
| sub_epoch_summaries = {} | ||
| if height_to_hash is None: | ||
| height_to_hash = {} | ||
| if headers is None: | ||
| headers = {} | ||
| self._block_records = blocks | ||
| self._headers = headers | ||
| self._height_to_hash = height_to_hash | ||
| self._sub_epoch_summaries = sub_epoch_summaries | ||
| self._sub_epoch_segments: Dict[bytes32, SubEpochSegments] = {} | ||
| self.log = logging.getLogger(__name__) | ||
|
|
||
| def block_record(self, header_hash: bytes32) -> BlockRecord: | ||
| return self._block_records[header_hash] | ||
|
|
||
| def height_to_block_record(self, height: uint32, check_db: bool = False) -> BlockRecord: | ||
| # Precondition: height is < peak height | ||
|
|
||
| header_hash: Optional[bytes32] = self.height_to_hash(height) | ||
| assert header_hash is not None | ||
|
|
||
| return self.block_record(header_hash) | ||
|
|
||
| def get_ses_heights(self) -> List[uint32]: | ||
| return sorted(self._sub_epoch_summaries.keys()) | ||
|
|
||
| def get_ses(self, height: uint32) -> SubEpochSummary: | ||
| return self._sub_epoch_summaries[height] | ||
|
|
||
| def height_to_hash(self, height: uint32) -> Optional[bytes32]: | ||
| if height not in self._height_to_hash: | ||
| self.log.warning(f"could not find height in cache {height}") | ||
| return None | ||
| return self._height_to_hash[height] | ||
|
|
||
| def contains_block(self, header_hash: bytes32) -> bool: | ||
| return header_hash in self._block_records | ||
|
|
||
| async def contains_block_from_db(self, header_hash: bytes32) -> bool: | ||
| return header_hash in self._block_records | ||
|
|
||
| def contains_height(self, height: uint32) -> bool: | ||
| return height in self._height_to_hash | ||
|
|
||
| async def get_block_records_in_range(self, start: int, stop: int) -> Dict[bytes32, BlockRecord]: | ||
| return self._block_records | ||
|
|
||
| async def get_block_records_at(self, heights: List[uint32]) -> List[BlockRecord]: | ||
| block_records: List[BlockRecord] = [] | ||
| for height in heights: | ||
| block_records.append(self.height_to_block_record(height)) | ||
| return block_records | ||
|
|
||
| def try_block_record(self, header_hash: bytes32) -> Optional[BlockRecord]: | ||
| return self._block_records.get(header_hash) | ||
|
|
||
| async def get_block_record_from_db(self, header_hash: bytes32) -> Optional[BlockRecord]: | ||
| return self._block_records[header_hash] | ||
|
|
||
| async def prev_block_hash(self, header_hashes: List[bytes32]) -> List[bytes32]: | ||
| ret = [] | ||
| for h in header_hashes: | ||
| ret.append(self._block_records[h].prev_hash) | ||
| return ret | ||
|
|
||
| def remove_block_record(self, header_hash: bytes32) -> None: | ||
| del self._block_records[header_hash] | ||
|
|
||
| def add_block_record(self, block: BlockRecord) -> None: | ||
| self._block_records[block.header_hash] = block | ||
|
|
||
| async def get_header_blocks_in_range( | ||
| self, start: int, stop: int, tx_filter: bool = True | ||
| ) -> Dict[bytes32, HeaderBlock]: | ||
| return self._headers | ||
|
|
||
| async def persist_sub_epoch_challenge_segments( | ||
| self, sub_epoch_summary_hash: bytes32, segments: List[SubEpochChallengeSegment] | ||
| ) -> None: | ||
| self._sub_epoch_segments[sub_epoch_summary_hash] = SubEpochSegments(segments) | ||
|
|
||
| async def get_sub_epoch_challenge_segments( | ||
| self, | ||
| sub_epoch_summary_hash: bytes32, | ||
| ) -> Optional[List[SubEpochChallengeSegment]]: | ||
| segments = self._sub_epoch_segments.get(sub_epoch_summary_hash) | ||
| if segments is None: | ||
| return None | ||
| return segments.challenge_segments |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.