Changeset View
Changeset View
Standalone View
Standalone View
qa/rpc-tests/p2p-mempool.py
| #!/usr/bin/env python3 | #!/usr/bin/env python3 | ||||
| # Copyright (c) 2015-2016 The Bitcoin Core developers | # Copyright (c) 2015-2016 The Bitcoin Core 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. | ||||
| from test_framework.mininode import * | from test_framework.mininode import * | ||||
| from test_framework.test_framework import BitcoinTestFramework | from test_framework.test_framework import BitcoinTestFramework | ||||
| from test_framework.util import * | from test_framework.util import * | ||||
| class TestNode(NodeConnCB): | class TestNode(NodeConnCB): | ||||
| def __init__(self): | def __init__(self): | ||||
| NodeConnCB.__init__(self) | NodeConnCB.__init__(self) | ||||
| self.connection = None | self.connection = None | ||||
| self.ping_counter = 1 | self.ping_counter = 1 | ||||
| self.last_pong = msg_pong() | self.last_pong = msg_pong() | ||||
| self.block_receive_map = {} | self.block_receive_map = {} | ||||
| def add_connection(self, conn): | def add_connection(self, conn): | ||||
| ▲ Show 20 Lines • Show All 46 Lines • ▼ Show 20 Lines | def sync_with_ping(self, timeout=30): | ||||
| success = wait_until(received_pong, timeout=timeout) | success = wait_until(received_pong, timeout=timeout) | ||||
| self.ping_counter += 1 | self.ping_counter += 1 | ||||
| return success | return success | ||||
| def send_mempool(self): | def send_mempool(self): | ||||
| self.lastInv = [] | self.lastInv = [] | ||||
| self.send_message(msg_mempool()) | self.send_message(msg_mempool()) | ||||
| class P2PMempoolTests(BitcoinTestFramework): | class P2PMempoolTests(BitcoinTestFramework): | ||||
| def __init__(self): | def __init__(self): | ||||
| super().__init__() | super().__init__() | ||||
| self.setup_clean_chain = True | self.setup_clean_chain = True | ||||
| self.num_nodes = 2 | self.num_nodes = 2 | ||||
| def setup_network(self): | def setup_network(self): | ||||
| # Start a node with maxuploadtarget of 200 MB (/24h) | # Start a node with maxuploadtarget of 200 MB (/24h) | ||||
| self.nodes = [] | self.nodes = [] | ||||
| self.nodes.append(start_node(0, self.options.tmpdir, ["-debug", "-peerbloomfilters=0"])) | self.nodes.append( | ||||
| start_node(0, self.options.tmpdir, ["-debug", "-peerbloomfilters=0"])) | |||||
| def run_test(self): | def run_test(self): | ||||
| #connect a mininode | # connect a mininode | ||||
| aTestNode = TestNode() | aTestNode = TestNode() | ||||
| node = NodeConn('127.0.0.1', p2p_port(0), self.nodes[0], aTestNode) | node = NodeConn('127.0.0.1', p2p_port(0), self.nodes[0], aTestNode) | ||||
| aTestNode.add_connection(node) | aTestNode.add_connection(node) | ||||
| NetworkThread().start() | NetworkThread().start() | ||||
| aTestNode.wait_for_verack() | aTestNode.wait_for_verack() | ||||
| #request mempool | # request mempool | ||||
| aTestNode.send_mempool() | aTestNode.send_mempool() | ||||
| aTestNode.wait_for_disconnect() | aTestNode.wait_for_disconnect() | ||||
| #mininode must be disconnected at this point | # mininode must be disconnected at this point | ||||
| assert_equal(len(self.nodes[0].getpeerinfo()), 0) | assert_equal(len(self.nodes[0].getpeerinfo()), 0) | ||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||
| P2PMempoolTests().main() | P2PMempoolTests().main() | ||||