Merge pull request #9796

ef1ff103f tests: Improve p2p tx propagation functional test (iamamyth)

1 files changed, 29 insertions, 10 deletions

diff --git a/tests/functional_tests/p2p.py b/tests/functional_tests/p2p.py
index ec2206aa8..259541be9 100755
--- a/tests/functional_tests/p2p.py
+++ b/tests/functional_tests/p2p.py
@@ -161,12 +161,11 @@ class P2PTest():
 
     def test_p2p_tx_propagation(self):
         print('Testing P2P tx propagation')
-        daemon2 = Daemon(idx = 2)
-        daemon3 = Daemon(idx = 3)
+        daemons = (Daemon(idx=2), Daemon(idx=3))
 
-        for daemon in [daemon2, daemon3]:
+        for daemon in daemons:
             res = daemon.get_transaction_pool_hashes()
-            assert not 'tx_hashes' in res or len(res.tx_hashes) == 0
+            assert len(res.get('tx_hashes', [])) == 0
 
         self.wallet.refresh()
         res = self.wallet.get_balance()
@@ -176,12 +175,32 @@ class P2PTest():
         assert len(res.tx_hash) == 32*2
         txid = res.tx_hash
 
-        time.sleep(5)
-
-        for daemon in [daemon2, daemon3]:
-            res = daemon.get_transaction_pool_hashes()
-            assert len(res.tx_hashes) == 1
-            assert res.tx_hashes[0] == txid
+        # Due to Dandelion++, the network propagates transactions with a
+        # random delay, so poll for the transaction with a timeout. The delay
+        # should almost never exceed a maximum of 13s (~1/billion samples).
+        # Set the timeout slightly higher than the maximum delay to account
+        # for transmission and processing time.
+        timeout = 13.5
+        pending_daemons = set(daemons)
+        expected_hashes = [txid]
+        wait_cutoff = time.monotonic() + timeout
+        while True:
+            done = []
+            for daemon in pending_daemons:
+                res = daemon.get_transaction_pool_hashes()
+                hashes = res.get('tx_hashes')
+                if hashes:
+                    assert hashes == expected_hashes
+                    done.append(daemon)
+            pending_daemons.difference_update(done)
+            if len(pending_daemons) == 0:
+                break
+            max_delay = wait_cutoff - time.monotonic()
+            if max_delay <= 0:
+                break
+            time.sleep(min(.2, max_delay))
+        npending = len(pending_daemons)
+        assert npending == 0, '%d daemons pending' % npending
 
 
 if __name__ == '__main__':