electrum

commit dc55cf4b2d23d84bcde26448c85f0842d611063e
parent 50ab8be7a29d1b1ece777d611a2a77e49402c689
Author: ThomasV <thomasv@electrum.org>
Date:   Thu, 11 Mar 2021 11:01:35 +0100

test_multipart_payment_with_timeout: give extra time for second HTLC to timeout before next test

Diffstat:
M
electrum/tests/test_lnpeer.py
 | 
9
+++++----

1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/electrum/tests/test_lnpeer.py b/electrum/tests/test_lnpeer.py
@@ -785,10 +785,7 @@ class TestPeer(ElectrumTestCase):
                       mpp_invoice=True):
             if mpp_invoice:
                 graph.w_d.features |= LnFeatures.BASIC_MPP_OPT
-            if bob_forwarding:
-                graph.w_b.enable_htlc_forwarding.set()
-            else:
-                graph.w_b.logger.info(f'disabling forwarding')
+            if not bob_forwarding:
                 graph.w_b.enable_htlc_forwarding.clear()
             if alice_uses_trampoline:
                 if graph.w_a.network.channel_db:
@@ -800,6 +797,10 @@ class TestPeer(ElectrumTestCase):
             lnaddr, pay_req = await self.prepare_invoice(graph.w_d, include_routing_hints=True, amount_msat=amount_to_pay)
             self.assertEqual(PR_UNPAID, graph.w_d.get_payment_status(lnaddr.paymenthash))
             result, log = await graph.w_a.pay_invoice(pay_req, attempts=attempts)
+            if not bob_forwarding:
+                # reset to previous state, sleep 2s so that the second htlc can time out
+                graph.w_b.enable_htlc_forwarding.set()
+                await asyncio.sleep(2)
             if result:
                 self.assertEqual(PR_PAID, graph.w_d.get_payment_status(lnaddr.paymenthash))
                 raise PaymentDone()