Changeset View
Changeset View
Standalone View
Standalone View
test/functional/p2p_timeouts.py
| Show All 22 Lines | |||||
| from time import sleep | from time import sleep | ||||
| from test_framework.messages import msg_ping | from test_framework.messages import msg_ping | ||||
| from test_framework.mininode import network_thread_start, P2PInterface | from test_framework.mininode import network_thread_start, P2PInterface | ||||
| from test_framework.test_framework import BitcoinTestFramework | from test_framework.test_framework import BitcoinTestFramework | ||||
| class TestNode(P2PInterface): | class TestP2PConn(P2PInterface): | ||||
| def on_version(self, message): | def on_version(self, message): | ||||
| # Don't send a verack in response | # Don't send a verack in response | ||||
| pass | pass | ||||
| class TimeoutsTest(BitcoinTestFramework): | class TimeoutsTest(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 | ||||
| def run_test(self): | def run_test(self): | ||||
| # Setup the p2p connections and start up the network thread. | # Setup the p2p connections and start up the network thread. | ||||
| no_verack_node = self.nodes[0].add_p2p_connection(TestNode()) | no_verack_node = self.nodes[0].add_p2p_connection(TestP2PConn()) | ||||
| no_version_node = self.nodes[0].add_p2p_connection( | no_version_node = self.nodes[0].add_p2p_connection( | ||||
| TestNode(), send_version=False) | TestP2PConn(), send_version=False) | ||||
| no_send_node = self.nodes[0].add_p2p_connection( | no_send_node = self.nodes[0].add_p2p_connection( | ||||
| TestNode(), send_version=False) | TestP2PConn(), send_version=False) | ||||
| network_thread_start() | network_thread_start() | ||||
| sleep(1) | sleep(1) | ||||
| assert no_verack_node.connected | assert no_verack_node.connected | ||||
| assert no_version_node.connected | assert no_version_node.connected | ||||
| assert no_send_node.connected | assert no_send_node.connected | ||||
| Show All 24 Lines | |||||