mirror of
https://github.com/dashpay/dash.git
synced 2024-12-27 04:52:59 +01:00
2a52ae63bf
All Node classes in individual test cases subclass from NodeConnCB. Many have duplicate definitions for methods that are defined in the base class. This commit removes those duplicate definitions. This commit removes ~290 lines of duplicate code.
92 lines
3.1 KiB
Python
Executable File
92 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) 2016 The Bitcoin Core developers
|
|
# Distributed under the MIT software license, see the accompanying
|
|
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
"""Test various net timeouts.
|
|
|
|
- Create three bitcoind nodes:
|
|
|
|
no_verack_node - we never send a verack in response to their version
|
|
no_version_node - we never send a version (only a ping)
|
|
no_send_node - we never send any P2P message.
|
|
|
|
- Start all three nodes
|
|
- Wait 1 second
|
|
- Assert that we're connected
|
|
- Send a ping to no_verack_node and no_version_node
|
|
- Wait 30 seconds
|
|
- Assert that we're still connected
|
|
- Send a ping to no_verack_node and no_version_node
|
|
- Wait 31 seconds
|
|
- Assert that we're no longer connected (timeout to receive version/verack is 60 seconds)
|
|
"""
|
|
|
|
from time import sleep
|
|
|
|
from test_framework.mininode import *
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.util import *
|
|
|
|
class TestNode(NodeConnCB):
|
|
def on_version(self, conn, message):
|
|
# Don't send a verack in response
|
|
pass
|
|
|
|
class TimeoutsTest(BitcoinTestFramework):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.setup_clean_chain = True
|
|
self.num_nodes = 1
|
|
|
|
def setup_network(self):
|
|
self.nodes = []
|
|
|
|
# Start up node0 to be a version 1, pre-segwit node.
|
|
self.nodes = start_nodes(self.num_nodes, self.options.tmpdir)
|
|
|
|
def run_test(self):
|
|
# Setup the p2p connections and start up the network thread.
|
|
self.no_verack_node = TestNode() # never send verack
|
|
self.no_version_node = TestNode() # never send version (just ping)
|
|
self.no_send_node = TestNode() # never send anything
|
|
|
|
connections = []
|
|
connections.append(NodeConn('127.0.0.1', p2p_port(0), self.nodes[0], self.no_verack_node))
|
|
connections.append(NodeConn('127.0.0.1', p2p_port(0), self.nodes[0], self.no_version_node, send_version=False))
|
|
connections.append(NodeConn('127.0.0.1', p2p_port(0), self.nodes[0], self.no_send_node, send_version=False))
|
|
self.no_verack_node.add_connection(connections[0])
|
|
self.no_version_node.add_connection(connections[1])
|
|
self.no_send_node.add_connection(connections[2])
|
|
|
|
NetworkThread().start() # Start up network handling in another thread
|
|
|
|
sleep(1)
|
|
|
|
assert(self.no_verack_node.connected)
|
|
assert(self.no_version_node.connected)
|
|
assert(self.no_send_node.connected)
|
|
|
|
ping_msg = msg_ping()
|
|
connections[0].send_message(ping_msg)
|
|
connections[1].send_message(ping_msg)
|
|
|
|
sleep(30)
|
|
|
|
assert "version" in self.no_verack_node.last_message
|
|
|
|
assert(self.no_verack_node.connected)
|
|
assert(self.no_version_node.connected)
|
|
assert(self.no_send_node.connected)
|
|
|
|
connections[0].send_message(ping_msg)
|
|
connections[1].send_message(ping_msg)
|
|
|
|
sleep(31)
|
|
|
|
assert(not self.no_verack_node.connected)
|
|
assert(not self.no_version_node.connected)
|
|
assert(not self.no_send_node.connected)
|
|
|
|
if __name__ == '__main__':
|
|
TimeoutsTest().main()
|