mirror of
https://github.com/dashpay/dash.git
synced 2024-12-26 20:42:59 +01:00
[tests] Add NetworkThread assertions
Check that P2PConnections aren't created after the NetworkThread has started and that at any time only one NetworkThread is running.
This commit is contained in:
parent
34e08b3510
commit
5c8ff26b2f
@ -69,6 +69,10 @@ class P2PConnection(asyncore.dispatcher):
|
|||||||
sub-classed and the on_message() callback overridden."""
|
sub-classed and the on_message() callback overridden."""
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
|
# All P2PConnections must be created before starting the NetworkThread.
|
||||||
|
# assert that the network thread is not running.
|
||||||
|
assert not network_thread_running()
|
||||||
|
|
||||||
super().__init__(map=mininode_socket_map)
|
super().__init__(map=mininode_socket_map)
|
||||||
|
|
||||||
def peer_connect(self, dstaddr, dstport, net="regtest"):
|
def peer_connect(self, dstaddr, dstport, net="regtest"):
|
||||||
@ -418,6 +422,9 @@ class NetworkThread(threading.Thread):
|
|||||||
|
|
||||||
def network_thread_start():
|
def network_thread_start():
|
||||||
"""Start the network thread."""
|
"""Start the network thread."""
|
||||||
|
# Only one network thread may run at a time
|
||||||
|
assert not network_thread_running()
|
||||||
|
|
||||||
NetworkThread().start()
|
NetworkThread().start()
|
||||||
|
|
||||||
def network_thread_running():
|
def network_thread_running():
|
||||||
|
Loading…
Reference in New Issue
Block a user