AaronFiore/electrum
1
0
Fork 0
Code Activity

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

Browse Source
This commit is contained in:
ThomasV
2021-03-11 11:01:35 +01:00
parent 50ab8be7a2
commit dc55cf4b2d
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
electrum/tests/test_lnpeer.py
View File

@@ -785,10 +785,7 @@ class TestPeer(ElectrumTestCase):
mpp_invoice=True): mpp_invoice=True):
if mpp_invoice: if mpp_invoice:
graph.w_d.features |= LnFeatures.BASIC_MPP_OPT graph.w_d.features |= LnFeatures.BASIC_MPP_OPT
if bob_forwarding: if not bob_forwarding:
graph.w_b.enable_htlc_forwarding.set()
else:
graph.w_b.logger.info(f'disabling forwarding')
graph.w_b.enable_htlc_forwarding.clear() graph.w_b.enable_htlc_forwarding.clear()
if alice_uses_trampoline: if alice_uses_trampoline:
if graph.w_a.network.channel_db: 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) 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)) self.assertEqual(PR_UNPAID, graph.w_d.get_payment_status(lnaddr.paymenthash))
result, log = await graph.w_a.pay_invoice(pay_req, attempts=attempts) 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: if result:
self.assertEqual(PR_PAID, graph.w_d.get_payment_status(lnaddr.paymenthash)) self.assertEqual(PR_PAID, graph.w_d.get_payment_status(lnaddr.paymenthash))
raise PaymentDone() raise PaymentDone()