In disconnect_p2ps(), wait for p2p connections to disappear from getp…

…eerinfo()
dashpay · Apr 17, 2020 · 3054854 · 3054854
1 parent 76d7b17
commit 3054854
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/test/functional/test_framework/test_node.py b/test/functional/test_framework/test_node.py
@@ -15,6 +15,7 @@
 import time
 
 from .authproxy import JSONRPCException
+from .messages import MY_SUBVERSION
 from .util import (
     assert_equal,
     delete_cookie_file,
@@ -223,6 +224,14 @@ def disconnect_p2ps(self):
             p.peer_disconnect()
         del self.p2ps[:]
 
+        # wait for p2p connections to disappear from getpeerinfo()
+        def check_peers():
+            for p in self.getpeerinfo():
+                if p['subver'] == MY_SUBVERSION.decode():
+                    return False
+            return True
+        wait_until(check_peers, timeout=5)
+
 class TestNodeCLIAttr:
     def __init__(self, cli, command):
         self.cli = cli