Differential D13658 Diff 39528 test/functional/abc_rpc_isfinal.py

Changeset View

Standalone View

test/functional/abc_rpc_isfinal.py

#!/usr/bin/env python3		#!/usr/bin/env python3
# Copyright (c) 2022 The Bitcoin developers		# Copyright (c) 2022 The Bitcoin developers
# Distributed under the MIT software license, see the accompanying		# Distributed under the MIT software license, see the accompanying
# file COPYING or http://www.opensource.org/licenses/mit-license.php.		# file COPYING or http://www.opensource.org/licenses/mit-license.php.
"""Test the isfinalxxx RPCS."""		"""Test the isfinalxxx RPCS."""
import random		import random

from test_framework.address import ADDRESS_ECREG_UNSPENDABLE		from test_framework.address import ADDRESS_ECREG_UNSPENDABLE
from test_framework.authproxy import JSONRPCException		from test_framework.authproxy import JSONRPCException
from test_framework.avatools import AvaP2PInterface		from test_framework.avatools import AvaP2PInterface, can_find_inv_in_poll
from test_framework.blocktools import create_block, create_coinbase		from test_framework.blocktools import create_block, create_coinbase
from test_framework.messages import (		from test_framework.messages import CBlockHeader, msg_headers
AvalancheVote,
AvalancheVoteError,
CBlockHeader,
msg_headers,
)
from test_framework.test_framework import BitcoinTestFramework		from test_framework.test_framework import BitcoinTestFramework
from test_framework.util import assert_equal, assert_raises_rpc_error, uint256_hex		from test_framework.util import assert_equal, assert_raises_rpc_error, uint256_hex

QUORUM_NODE_COUNT = 16		QUORUM_NODE_COUNT = 16


class AvalancheIsFinalTest(BitcoinTestFramework):		class AvalancheIsFinalTest(BitcoinTestFramework):
def set_test_params(self):		def set_test_params(self):
Show All 35 Lines	def run_test(self):

# Pick one node from the quorum for polling.		# Pick one node from the quorum for polling.
quorum = get_quorum()		quorum = get_quorum()

def is_quorum_established():		def is_quorum_established():
return node.getavalancheinfo()['ready_to_poll'] is True		return node.getavalancheinfo()['ready_to_poll'] is True
self.wait_until(is_quorum_established)		self.wait_until(is_quorum_established)

def can_find_block_in_poll(
blockhash, resp=AvalancheVoteError.ACCEPTED):
found_hash = False
for n in quorum:
poll = n.get_avapoll_if_available()

# That node has not received a poll
if poll is None:
continue

# We got a poll, check for the hash and repond
votes = []
for inv in poll.invs:
# Vote yes to everything
r = AvalancheVoteError.ACCEPTED

# Look for what we expect
if inv.hash == int(blockhash, 16):
r = resp
found_hash = True

votes.append(AvalancheVote(r, inv.hash))

n.send_avaresponse(poll.round, votes, n.delegated_privkey)

return found_hash

blockhash = self.generate(node, 1, sync_fun=self.no_op)[0]		blockhash = self.generate(node, 1, sync_fun=self.no_op)[0]
cb_txid = node.getblock(blockhash)['tx'][0]		cb_txid = node.getblock(blockhash)['tx'][0]
assert not node.isfinalblock(blockhash)		assert not node.isfinalblock(blockhash)
assert not node.isfinaltransaction(cb_txid, blockhash)		assert not node.isfinaltransaction(cb_txid, blockhash)

def is_finalblock(blockhash):		def is_finalblock(blockhash):
can_find_block_in_poll(blockhash)		can_find_inv_in_poll(quorum, int(blockhash, 16))
return node.isfinalblock(blockhash)		return node.isfinalblock(blockhash)

with node.assert_debug_log([f"Avalanche finalized block {blockhash}"]):		with node.assert_debug_log([f"Avalanche finalized block {blockhash}"]):
self.wait_until(lambda: is_finalblock(blockhash))		self.wait_until(lambda: is_finalblock(blockhash))
assert node.isfinaltransaction(cb_txid, blockhash)		assert node.isfinaltransaction(cb_txid, blockhash)

self.log.info("Check block ancestors are finalized as well")		self.log.info("Check block ancestors are finalized as well")
tip_height = node.getblockheader(blockhash)['height']		tip_height = node.getblockheader(blockhash)['height']
▲ Show 20 Lines • Show All 122 Lines • Show Last 20 Lines