Changeset View
Changeset View
Standalone View
Standalone View
test/functional/abc_rpc_avalancheproof.py
| #!/usr/bin/env python3 | #!/usr/bin/env python3 | ||||
| # Copyright (c) 2021 The Bitcoin developers | # Copyright (c) 2021 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 building avalanche proofs and using them to add avalanche peers.""" | """Test building avalanche proofs and using them to add avalanche peers.""" | ||||
| import base64 | import base64 | ||||
| from decimal import Decimal | from decimal import Decimal | ||||
| from test_framework.address import ADDRESS_BCHREG_UNSPENDABLE | |||||
| from test_framework.avatools import ( | from test_framework.avatools import ( | ||||
| create_coinbase_stakes, | create_coinbase_stakes, | ||||
| create_stakes, | create_stakes, | ||||
| get_proof_ids, | get_proof_ids, | ||||
| ) | ) | ||||
| from test_framework.key import ECKey, bytes_to_wif | from test_framework.key import ECKey, bytes_to_wif | ||||
| from test_framework.messages import ( | from test_framework.messages import ( | ||||
| AvalancheDelegation, | AvalancheDelegation, | ||||
| ▲ Show 20 Lines • Show All 263 Lines • ▼ Show 20 Lines | def run_test(self): | ||||
| # sending of an inv message with our proof: | # sending of an inv message with our proof: | ||||
| # | # | ||||
| # def inv_found(): | # def inv_found(): | ||||
| # with p2p_lock: | # with p2p_lock: | ||||
| # return peer.last_message.get( | # return peer.last_message.get( | ||||
| # "inv") and peer.last_message["inv"].inv[-1].hash == proofid | # "inv") and peer.last_message["inv"].inv[-1].hash == proofid | ||||
| # wait_until(inv_found) | # wait_until(inv_found) | ||||
| self.log.info("Check the getrawproof RPC") | |||||
| raw_proof = node.getrawavalancheproof("{:064x}".format(proofid)) | |||||
| assert_equal(raw_proof['proof'], proof) | |||||
| assert_equal(raw_proof['orphan'], False) | |||||
| assert_raises_rpc_error(-8, "Proof not found", | |||||
| node.getrawavalancheproof, '0' * 64) | |||||
| # Orphan the proof by sending the stake | |||||
| raw_tx = node.createrawtransaction( | |||||
| [{"txid": stakes[-1]["txid"], "vout": 0}], | |||||
| {ADDRESS_BCHREG_UNSPENDABLE: stakes[-1] | |||||
| ["amount"] - Decimal('0.01')} | |||||
| ) | |||||
| signed_tx = node.signrawtransactionwithkey(raw_tx, [addrkey0.key]) | |||||
| node.sendrawtransaction(signed_tx["hex"]) | |||||
| node.generate(1) | |||||
| wait_until(lambda: proofid not in get_proof_ids(node)) | |||||
| raw_proof = node.getrawavalancheproof("{:064x}".format(proofid)) | |||||
| assert_equal(raw_proof['proof'], proof) | |||||
| assert_equal(raw_proof['orphan'], True) | |||||
| self.log.info("Bad proof should be rejected at startup") | self.log.info("Bad proof should be rejected at startup") | ||||
| self.stop_node(0) | self.stop_node(0) | ||||
| node.assert_start_raises_init_error( | node.assert_start_raises_init_error( | ||||
| self.extra_args[0] + [ | self.extra_args[0] + [ | ||||
| "-avasessionkey=0", | "-avasessionkey=0", | ||||
| ], | ], | ||||
| ▲ Show 20 Lines • Show All 52 Lines • Show Last 20 Lines | |||||