Changeset View
Changeset View
Standalone View
Standalone View
test/functional/abc-p2p-avalanche.py
- This file was added.
| Property | Old Value | New Value |
|---|---|---|
| File Mode | null | 100755 |
| #!/usr/bin/env python3 | |||||
| # Copyright (c) 2018 The Bitcoin developers | |||||
| # Distributed under the MIT software license, see the accompanying | |||||
| # file COPYING or http://www.opensource.org/licenses/mit-license.php. | |||||
| """Test the resolution of forks via avalanche.""" | |||||
| import random | |||||
| from test_framework.test_framework import BitcoinTestFramework | |||||
| from test_framework.util import assert_equal, wait_until | |||||
| from test_framework.mininode import P2PInterface, mininode_lock | |||||
| from test_framework.messages import AvalancheVote, CInv, msg_avapoll | |||||
| BLOCK_ACCEPTED = 0 | |||||
| BLOCK_REJECTED = 1 | |||||
| BLOCK_UNKNOWN = -1 | |||||
| class TestNode(P2PInterface): | |||||
| def __init__(self): | |||||
| self.last_avaresponse = None | |||||
| super().__init__() | |||||
| def on_avaresponse(self, message): | |||||
| self.last_avaresponse = message.response | |||||
| def send_poll(self, hashes): | |||||
| msg = msg_avapoll() | |||||
| for h in hashes: | |||||
| msg.poll.invs.append(CInv(2, h)) | |||||
| self.send_message(msg) | |||||
| def wait_for_avaresponse(self, timeout=10): | |||||
| self.sync_with_ping() | |||||
| def test_function(): | |||||
| m = self.last_message.get("avaresponse") | |||||
| return m is not None and m != self.last_avaresponse | |||||
| wait_until(test_function, timeout=timeout, lock=mininode_lock) | |||||
| class AvalancheTest(BitcoinTestFramework): | |||||
| def set_test_params(self): | |||||
| self.setup_clean_chain = True | |||||
| self.num_nodes = 1 | |||||
| self.extra_args = [['-enableavalanche=1', '-avacooldown=0']] | |||||
| def run_test(self): | |||||
| node = self.nodes[0] | |||||
| # Create a fake node and connect it to our real node. | |||||
| poll_node = TestNode() | |||||
| node.add_p2p_connection(poll_node) | |||||
| poll_node.wait_for_verack() | |||||
| poll_node.sync_with_ping() | |||||
| # Generate many block and poll for them. | |||||
| node.generate(100) | |||||
| self.log.info("Poll for the chain tip...") | |||||
| best_block_hash = int(node.getbestblockhash(), 16) | |||||
| poll_node.send_poll([best_block_hash]) | |||||
| poll_node.wait_for_avaresponse() | |||||
| def assert_response(response, expected): | |||||
| votes = response.votes | |||||
| self.log.info("response: {}".format(repr(response))) | |||||
| assert_equal(len(votes), len(expected)) | |||||
| for i in range(0, len(votes)): | |||||
| assert_equal(repr(votes[i]), repr(expected[i])) | |||||
| assert_response(poll_node.last_avaresponse, [ | |||||
| AvalancheVote(BLOCK_ACCEPTED, best_block_hash)]) | |||||
| self.log.info("Poll for a selection of blocks...") | |||||
| various_block_hashes = [ | |||||
| int(node.getblockhash(0), 16), | |||||
| int(node.getblockhash(1), 16), | |||||
| int(node.getblockhash(10), 16), | |||||
| int(node.getblockhash(25), 16), | |||||
| int(node.getblockhash(42), 16), | |||||
| int(node.getblockhash(96), 16), | |||||
| int(node.getblockhash(99), 16), | |||||
| int(node.getblockhash(100), 16), | |||||
| ] | |||||
| poll_node.send_poll(various_block_hashes) | |||||
| poll_node.wait_for_avaresponse() | |||||
| assert_response(poll_node.last_avaresponse, | |||||
| [AvalancheVote(BLOCK_ACCEPTED, h) for h in various_block_hashes]) | |||||
| self.log.info( | |||||
| "Poll for a selection of blocks, but some are now invalid...") | |||||
| invalidated_block = node.getblockhash(75) | |||||
| node.invalidateblock(invalidated_block) | |||||
| node.generate(30) | |||||
| node.reconsiderblock(invalidated_block) | |||||
| poll_node.send_poll(various_block_hashes) | |||||
| poll_node.wait_for_avaresponse() | |||||
| assert_response(poll_node.last_avaresponse, | |||||
| [AvalancheVote(BLOCK_ACCEPTED, h) for h in various_block_hashes[:5]] + | |||||
| [AvalancheVote(BLOCK_REJECTED, h) for h in various_block_hashes[-3:]]) | |||||
| self.log.info("Poll for unknown blocks...") | |||||
| various_block_hashes = [ | |||||
| int(node.getblockhash(0), 16), | |||||
| int(node.getblockhash(25), 16), | |||||
| int(node.getblockhash(42), 16), | |||||
| various_block_hashes[5], | |||||
| various_block_hashes[6], | |||||
| various_block_hashes[7], | |||||
| random.randrange(1 << 255, (1 << 256) - 1), | |||||
| random.randrange(1 << 255, (1 << 256) - 1), | |||||
| random.randrange(1 << 255, (1 << 256) - 1), | |||||
| ] | |||||
| poll_node.send_poll(various_block_hashes) | |||||
| poll_node.wait_for_avaresponse() | |||||
| assert_response(poll_node.last_avaresponse, | |||||
| [AvalancheVote(BLOCK_ACCEPTED, h) for h in various_block_hashes[:3]] + | |||||
| [AvalancheVote(BLOCK_REJECTED, h) for h in various_block_hashes[3:6]] + | |||||
| [AvalancheVote(BLOCK_UNKNOWN, h) for h in various_block_hashes[-3:]]) | |||||
| if __name__ == '__main__': | |||||
| AvalancheTest().main() | |||||