Chia-Network/chia-blockchain
1
0
Fork 0
mirror of https://github.com/Chia-Network/chia-blockchain.git synced 2024-09-11 10:56:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

Annotate test_wallet_blockchain.py (#17361)

Browse Source 
Annotate test_wallet_blockchain.py.
This commit is contained in:
Amine Khaldi 2024-01-19 18:00:01 +01:00 committed by GitHub
parent f1d8b77704
commit 4274e89328
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 43 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mypy-exclusions.txt
View File

@ -111,7 +111,6 @@ tests.wallet.test_chialisp
tests.wallet.test_singleton_lifecycle
tests.wallet.test_singleton_lifecycle_fast
tests.wallet.test_taproot
tests.wallet.test_wallet_blockchain
tests.wallet.test_wallet_interested_store
tests.wallet.test_wallet_key_val_store
tools.analyze-chain

66
tests/wallet/test_wallet_blockchain.py
View File

@ -1,49 +1,58 @@
from __future__ import annotations
import dataclasses
from typing import List
import pytest
from chia.consensus.blockchain import AddBlockResult
from chia.protocols import full_node_protocol
from chia.types.blockchain_format.vdf import VDFProof
from chia.types.weight_proof import WeightProof
from chia.types.full_block import FullBlock
from chia.types.header_block import HeaderBlock
from chia.util.generator_tools import get_block_header
from chia.util.ints import uint8, uint32
from chia.wallet.key_val_store import KeyValStore
from chia.wallet.wallet_blockchain import WalletBlockchain
from tests.conftest import ConsensusMode
from tests.util.db_connection import DBConnection
from tests.util.setup_nodes import OldSimulatorsAndWallets
@pytest.mark.limit_consensus_modes(allowed=[ConsensusMode.PLAIN, ConsensusMode.HARD_FORK_2_0], reason="save time")
@pytest.mark.anyio
async def test_wallet_blockchain(simulator_and_wallet, default_1000_blocks):
async def test_wallet_blockchain(
simulator_and_wallet: OldSimulatorsAndWallets, default_1000_blocks: List[FullBlock]
) -> None:
[full_node_api], [(wallet_node, _)], bt = simulator_and_wallet
for block in default_1000_blocks[:600]:
await full_node_api.full_node.add_block(block)
res = await full_node_api.request_proof_of_weight(
resp = await full_node_api.request_proof_of_weight(
full_node_protocol.RequestProofOfWeight(
default_1000_blocks[499].height + 1, default_1000_blocks[499].header_hash
uint32(default_1000_blocks[499].height + 1), default_1000_blocks[499].header_hash
)
)
res_2 = await full_node_api.request_proof_of_weight(
assert resp is not None
resp_2 = await full_node_api.request_proof_of_weight(
full_node_protocol.RequestProofOfWeight(
default_1000_blocks[460].height + 1, default_1000_blocks[460].header_hash
uint32(default_1000_blocks[460].height + 1), default_1000_blocks[460].header_hash
)
)
res_3 = await full_node_api.request_proof_of_weight(
assert resp_2 is not None
resp_3 = await full_node_api.request_proof_of_weight(
full_node_protocol.RequestProofOfWeight(
default_1000_blocks[505].height + 1, default_1000_blocks[505].header_hash
uint32(default_1000_blocks[505].height + 1), default_1000_blocks[505].header_hash
)
)
weight_proof: WeightProof = full_node_protocol.RespondProofOfWeight.from_bytes(res.data).wp
assert resp_3 is not None
weight_proof = full_node_protocol.RespondProofOfWeight.from_bytes(resp.data).wp
assert wallet_node._weight_proof_handler is not None
records = await wallet_node._weight_proof_handler.validate_weight_proof(weight_proof, True)
weight_proof_short: WeightProof = full_node_protocol.RespondProofOfWeight.from_bytes(res_2.data).wp
weight_proof_short = full_node_protocol.RespondProofOfWeight.from_bytes(resp_2.data).wp
records_short = await wallet_node._weight_proof_handler.validate_weight_proof(weight_proof_short, True)
weight_proof_long: WeightProof = full_node_protocol.RespondProofOfWeight.from_bytes(res_3.data).wp
weight_proof_long = full_node_protocol.RespondProofOfWeight.from_bytes(resp_3.data).wp
records_long = await wallet_node._weight_proof_handler.validate_weight_proof(weight_proof_long, True)
async with DBConnection(1) as db_wrapper:
@ -54,17 +63,22 @@ async def test_wallet_blockchain(simulator_and_wallet, default_1000_blocks):
assert chain.get_latest_timestamp() == 0
await chain.new_valid_weight_proof(weight_proof, records)
assert (await chain.get_peak_block()) is not None
assert (await chain.get_peak_block()).height == 499
peak_block = await chain.get_peak_block()
assert peak_block is not None
assert peak_block.height == 499
assert chain.get_latest_timestamp() > 0
await chain.new_valid_weight_proof(weight_proof_short, records_short)
assert (await chain.get_peak_block()).height == 499
peak_block = await chain.get_peak_block()
assert peak_block is not None
assert peak_block.height == 499
await chain.new_valid_weight_proof(weight_proof_long, records_long)
assert (await chain.get_peak_block()).height == 505
peak_block = await chain.get_peak_block()
assert peak_block is not None
assert peak_block.height == 505
header_blocks = []
header_blocks: List[HeaderBlock] = []
for block in default_1000_blocks:
header_block = get_block_header(block, [], [])
header_blocks.append(header_block)
@ -82,15 +96,21 @@ async def test_wallet_blockchain(simulator_and_wallet, default_1000_blocks):
assert res == AddBlockResult.DISCONNECTED_BLOCK
res, err = await chain.add_block(
dataclasses.replace(header_blocks[506], challenge_chain_ip_proof=VDFProof(2, b"123", True))
dataclasses.replace(header_blocks[506], challenge_chain_ip_proof=VDFProof(uint8(2), b"123", True))
)
assert res == AddBlockResult.INVALID_BLOCK
assert (await chain.get_peak_block()).height == 505
peak_block = await chain.get_peak_block()
assert peak_block is not None
assert peak_block.height == 505
for block in header_blocks[506:]:
res, err = await chain.add_block(block)
for header_block in header_blocks[506:]:
res, err = await chain.add_block(header_block)
assert res == AddBlockResult.NEW_PEAK
assert (await chain.get_peak_block()).height == block.height
peak_block = await chain.get_peak_block()
assert peak_block is not None
assert peak_block.height == header_block.height
assert (await chain.get_peak_block()).height == 999
peak_block = await chain.get_peak_block()
assert peak_block is not None
assert peak_block.height == 999