mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2024-11-29 05:18:11 +03:00
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
import asyncio
|
|
|
|
import pytest
|
|
|
|
from src.util.merkle_set import MerkleSet, confirm_included_already_hashed
|
|
from tests.setup_nodes import test_constants, bt
|
|
from src.util.wallet_tools import WalletTool
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def event_loop():
|
|
loop = asyncio.get_event_loop()
|
|
yield loop
|
|
|
|
|
|
class TestMerkleSet:
|
|
@pytest.mark.asyncio
|
|
async def test_basics(self):
|
|
WalletTool()
|
|
|
|
num_blocks = 10
|
|
blocks = bt.get_consecutive_blocks(
|
|
test_constants,
|
|
num_blocks,
|
|
[],
|
|
10,
|
|
)
|
|
|
|
merkle_set = MerkleSet()
|
|
merkle_set_reverse = MerkleSet()
|
|
|
|
for block in reversed(blocks):
|
|
merkle_set_reverse.add_already_hashed(block.get_coinbase().name())
|
|
|
|
for block in blocks:
|
|
merkle_set.add_already_hashed(block.get_coinbase().name())
|
|
|
|
for block in blocks:
|
|
result, proof = merkle_set.is_included_already_hashed(
|
|
block.get_coinbase().name()
|
|
)
|
|
assert result is True
|
|
result_fee, proof_fee = merkle_set.is_included_already_hashed(
|
|
block.get_fees_coin().name()
|
|
)
|
|
assert result_fee is False
|
|
validate_proof = confirm_included_already_hashed(
|
|
merkle_set.get_root(), block.get_coinbase().name(), proof
|
|
)
|
|
validate_proof_fee = confirm_included_already_hashed(
|
|
merkle_set.get_root(), block.get_fees_coin().name(), proof_fee
|
|
)
|
|
assert validate_proof is True
|
|
assert validate_proof_fee is False
|
|
|
|
# Test if order of adding items change the outcome
|
|
assert merkle_set.get_root() == merkle_set_reverse.get_root()
|