Changeset View
Changeset View
Standalone View
Standalone View
test/functional/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. | ||||
| """Test p2p mempool message. | """Test p2p mempool message. | ||||
| Test that nodes are disconnected if they send mempool messages when bloom | Test that nodes are disconnected if they send mempool messages when bloom | ||||
| filters are not enabled. | filters are not enabled. | ||||
| """ | """ | ||||
| from test_framework.messages import msg_mempool | from test_framework.messages import msg_mempool | ||||
| from test_framework.mininode import network_thread_start, P2PInterface | from test_framework.mininode import P2PInterface | ||||
| from test_framework.test_framework import BitcoinTestFramework | from test_framework.test_framework import BitcoinTestFramework | ||||
| from test_framework.util import assert_equal | from test_framework.util import assert_equal | ||||
| class P2PMempoolTests(BitcoinTestFramework): | class P2PMempoolTests(BitcoinTestFramework): | ||||
| def set_test_params(self): | def set_test_params(self): | ||||
| self.setup_clean_chain = True | self.setup_clean_chain = True | ||||
| self.num_nodes = 1 | self.num_nodes = 1 | ||||
| self.extra_args = [["-peerbloomfilters=0"]] | self.extra_args = [["-peerbloomfilters=0"]] | ||||
| def run_test(self): | def run_test(self): | ||||
| # Add a p2p connection | # Add a p2p connection | ||||
| self.nodes[0].add_p2p_connection(P2PInterface()) | self.nodes[0].add_p2p_connection(P2PInterface()) | ||||
| network_thread_start() | |||||
| self.nodes[0].p2p.wait_for_verack() | self.nodes[0].p2p.wait_for_verack() | ||||
| # request mempool | # request mempool | ||||
| self.nodes[0].p2p.send_message(msg_mempool()) | self.nodes[0].p2p.send_message(msg_mempool()) | ||||
| self.nodes[0].p2p.wait_for_disconnect() | self.nodes[0].p2p.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() | ||||