test: forward timeouts properly in send_blocks_and_test

ElementsProject · Oct 10, 2018 · 94e21c1 · 94e21c1
1 parent be99270
commit 94e21c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/functional/test_framework/mininode.py b/test/functional/test_framework/mininode.py
@@ -500,9 +500,9 @@ def send_blocks_and_test(self, blocks, node, *, success=True, request_block=True
                 wait_until(lambda: blocks[-1].sha256 in self.getdata_requests, timeout=timeout, lock=mininode_lock)
 
             if expect_disconnect:
-                self.wait_for_disconnect()
+                self.wait_for_disconnect(timeout=timeout)
             else:
-                self.sync_with_ping()
+                self.sync_with_ping(timeout=timeout)
 
             if success:
                 wait_until(lambda: node.getbestblockhash() == blocks[-1].hash, timeout=timeout)