starred/chia-blockchain
1
0
Fork 0
mirror of https://github.com/Chia-Network/chia-blockchain.git synced 2026-08-24 10:05:29 -05:00
Code Issues Packages Projects Releases Wiki Activity

checkpoint: into main from release/2.1.4 @ f6522969dd (#17280)

Browse Source 
Source hash: f6522969dd
Remaining commits: 1
This commit is contained in:
Chris Marslender
2024-01-10 13:13:40 -07:00
committed by GitHub
parent 0cafc83194 28012ca03b
commit 9d0748fb98
1 changed files with 24 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/wallet/nft_wallet/test_nft_wallet.py
+24 -1
View File
@@ -195,6 +195,10 @@ async def test_nft_wallet_creation_and_transfer(
ph = await wallet_0.get_new_puzzlehash()
ph1 = await wallet_1.get_new_puzzlehash()
async def ensure_wallet_sync() -> None:
await full_node_api.wait_for_wallet_synced(wallet_node=wallet_node_0, timeout=20)
await full_node_api.wait_for_wallet_synced(wallet_node=wallet_node_1, timeout=20)
if trusted:
wallet_node_0.config["trusted_peers"] = {
full_node_api.full_node.server.node_id.hex(): full_node_api.full_node.server.node_id.hex()
@@ -249,7 +253,11 @@ async def test_nft_wallet_creation_and_transfer(
# Test Reorg mint
height = full_node_api.full_node.blockchain.get_peak_height()
assert height is not None
await full_node_api.reorg_from_index_to_new_index(ReorgProtocol(uint32(height - 1), uint32(height + 1), ph1, None))
await time_out_assert(60, full_node_api.full_node.blockchain.get_peak_height, height + 1)
await time_out_assert(30, get_nft_count, 0, nft_wallet_0)
await time_out_assert(30, get_wallet_number, 2, wallet_node_0.wallet_state_manager)
@@ -271,6 +279,7 @@ async def test_nft_wallet_creation_and_transfer(
await time_out_assert(30, wallet_node_0.wallet_state_manager.lock.locked, False)
for _ in range(1, num_blocks * 2):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph1))
await time_out_assert(30, get_nft_count, 2, nft_wallet_0)
coins = await nft_wallet_0.get_current_nfts()
assert len(coins) == 2, "nft not generated"
@@ -292,6 +301,9 @@ async def test_nft_wallet_creation_and_transfer(
for _ in range(1, num_blocks * 2):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph1))
await ensure_wallet_sync()
await time_out_assert(30, get_nft_count, 1, nft_wallet_0)
await time_out_assert(30, get_nft_count, 1, nft_wallet_1)
@@ -312,17 +324,28 @@ async def test_nft_wallet_creation_and_transfer(
)
assert len(compute_memos(txs[0].spend_bundle)) > 0
await ensure_wallet_sync()
# We wish for the wallet to be in a specific state at this point after having
# become synced above.
for _ in range(1, num_blocks):
await full_node_api.farm_new_transaction_block(FarmNewBlockProtocol(ph1))
await time_out_assert(30, wallet_node_0.wallet_state_manager.lock.locked, False)
await ensure_wallet_sync()
# The wallets are locked so we still observe the conditions above.
await time_out_assert(30, get_nft_count, 2, nft_wallet_0)
await time_out_assert(30, get_nft_count, 0, nft_wallet_1)
# Test Reorg
height = full_node_api.full_node.blockchain.get_peak_height()
assert height is not None
await full_node_api.reorg_from_index_to_new_index(ReorgProtocol(uint32(height - 1), uint32(height + 2), ph1, None))
# Wait for wallet sync on all wallets.
await ensure_wallet_sync()
await time_out_assert(30, get_nft_count, 1, nft_wallet_0)
await time_out_assert(30, get_nft_count, 1, nft_wallet_1)