mirror of
https://github.com/bitcoin/bitcoin.git
synced 2024-11-19 18:09:47 +01:00
[test] wait for disconnect_p2ps to be reflected in getpeerinfo
-Waiting is important to avoid race conditions, especially if testing peer info through rpc later. -Wait for mininodes to be disconnected only, even though it's more complex, because we may still want to be connected to test nodes.
This commit is contained in:
parent
e81942d2e1
commit
aeb9fb414e
@ -23,6 +23,7 @@ import sys
|
||||
|
||||
from .authproxy import JSONRPCException
|
||||
from .descriptors import descsum_create
|
||||
from .messages import MY_SUBVERSION
|
||||
from .util import (
|
||||
MAX_NODES,
|
||||
append_config,
|
||||
@ -549,11 +550,16 @@ class TestNode():
|
||||
assert self.p2ps, self._node_msg("No p2p connection")
|
||||
return self.p2ps[0]
|
||||
|
||||
def num_connected_mininodes(self):
|
||||
"""Return number of test framework p2p connections to the node."""
|
||||
return len([peer for peer in self.getpeerinfo() if peer['subver'] == MY_SUBVERSION])
|
||||
|
||||
def disconnect_p2ps(self):
|
||||
"""Close all p2p connections to the node."""
|
||||
for p in self.p2ps:
|
||||
p.peer_disconnect()
|
||||
del self.p2ps[:]
|
||||
wait_until(lambda: self.num_connected_mininodes() == 0)
|
||||
|
||||
|
||||
class TestNodeCLIAttr:
|
||||
|
Loading…
Reference in New Issue
Block a user