LBRYFoundation/LBRY-Vault
1
0
Fork 0
mirror of https://github.com/LBRYFoundation/LBRY-Vault.git synced 2025-08-23 17:47:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test_lnpeer: test_reestablish_with_old_state

Browse source
This commit is contained in:
ThomasV 2020-02-12 10:22:22 +01:00
parent 3377627070
commit e3630d87b2
2 changed files with 31 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
electrum/lnpeer.py
View file

@ -1150,6 +1150,8 @@ class Peer(Logger):
def on_commitment_signed(self, payload): def on_commitment_signed(self, payload):
channel_id = payload['channel_id'] channel_id = payload['channel_id']
chan = self.channels[channel_id] chan = self.channels[channel_id]
if chan.peer_state == peer_states.BAD:
return
# make sure there were changes to the ctx, otherwise the remote peer is misbehaving # make sure there were changes to the ctx, otherwise the remote peer is misbehaving
next_htlcs, latest_htlcs = chan.hm.get_htlcs_in_next_ctx(LOCAL), chan.hm.get_htlcs_in_latest_ctx(LOCAL) next_htlcs, latest_htlcs = chan.hm.get_htlcs_in_next_ctx(LOCAL), chan.hm.get_htlcs_in_latest_ctx(LOCAL)
self.logger.info(f'on_commitment_signed. chan {chan.short_channel_id}. ctn: {chan.get_next_ctn(LOCAL)}. ' self.logger.info(f'on_commitment_signed. chan {chan.short_channel_id}. ctn: {chan.get_next_ctn(LOCAL)}. '
@ -1376,6 +1378,8 @@ class Peer(Logger):
def on_revoke_and_ack(self, payload): def on_revoke_and_ack(self, payload):
channel_id = payload["channel_id"] channel_id = payload["channel_id"]
chan = self.channels[channel_id] chan = self.channels[channel_id]
if chan.peer_state == peer_states.BAD:
return
self.logger.info(f'on_revoke_and_ack. chan {chan.short_channel_id}. ctn: {chan.get_oldest_unrevoked_ctn(REMOTE)}') self.logger.info(f'on_revoke_and_ack. chan {chan.short_channel_id}. ctn: {chan.get_oldest_unrevoked_ctn(REMOTE)}')
rev = RevokeAndAck(payload["per_commitment_secret"], payload["next_per_commitment_point"]) rev = RevokeAndAck(payload["per_commitment_secret"], payload["next_per_commitment_point"])
chan.receive_revocation(rev) chan.receive_revocation(rev)

30
electrum/tests/test_lnpeer.py
View file

@ -76,11 +76,10 @@ class MockWallet:
class MockLNWallet: class MockLNWallet:
def __init__(self, remote_keypair, local_keypair, chan, tx_queue): def __init__(self, remote_keypair, local_keypair, chan, tx_queue):
self.chan = chan
self.remote_keypair = remote_keypair self.remote_keypair = remote_keypair
self.node_keypair = local_keypair self.node_keypair = local_keypair
self.network = MockNetwork(tx_queue) self.network = MockNetwork(tx_queue)
self.channels = {self.chan.channel_id: self.chan} self.channels = {chan.channel_id: chan}
self.payments = {} self.payments = {}
self.logs = defaultdict(list) self.logs = defaultdict(list)
self.wallet = MockWallet() self.wallet = MockWallet()
@ -175,7 +174,6 @@ class TestPeer(ElectrumTestCase):
def setUp(self): def setUp(self):
super().setUp() super().setUp()
self.asyncio_loop, self._stop_loop, self._loop_thread = create_and_start_event_loop() self.asyncio_loop, self._stop_loop, self._loop_thread = create_and_start_event_loop()
self.alice_channel, self.bob_channel = create_test_channels()
def tearDown(self): def tearDown(self):
super().tearDown() super().tearDown()
@ -220,6 +218,7 @@ class TestPeer(ElectrumTestCase):
return lnencode(lnaddr, w2.node_keypair.privkey) return lnencode(lnaddr, w2.node_keypair.privkey)
def test_reestablish(self): def test_reestablish(self):
self.alice_channel, self.bob_channel = create_test_channels()
p1, p2, w1, w2, _q1, _q2 = self.prepare_peers() p1, p2, w1, w2, _q1, _q2 = self.prepare_peers()
async def reestablish(): async def reestablish():
await asyncio.gather( await asyncio.gather(
@ -234,7 +233,31 @@ class TestPeer(ElectrumTestCase):
with self.assertRaises(concurrent.futures.CancelledError): with self.assertRaises(concurrent.futures.CancelledError):
run(f()) run(f())
def test_reestablish_with_old_state(self):
self.alice_channel, self.bob_channel = create_test_channels()
self.alice_channel_0, self.bob_channel_0 = create_test_channels() # these are identical
p1, p2, w1, w2, _q1, _q2 = self.prepare_peers()
pay_req = self.prepare_invoice(w2)
async def reestablish():
result = await LNWallet._pay(w1, pay_req)
self.assertEqual(result, True)
w1.channels = {self.alice_channel_0.channel_id: self.alice_channel_0}
await asyncio.gather(
p1.reestablish_channel(self.alice_channel_0),
p2.reestablish_channel(self.bob_channel))
self.assertEqual(self.alice_channel_0.peer_state, peer_states.BAD)
self.assertEqual(self.bob_channel._state, channel_states.FORCE_CLOSING)
# wait so that pending messages are processed
#await asyncio.sleep(1)
gath.cancel()
gath = asyncio.gather(reestablish(), p1._message_loop(), p2._message_loop())
async def f():
await gath
with self.assertRaises(concurrent.futures.CancelledError):
run(f())
def test_payment(self): def test_payment(self):
self.alice_channel, self.bob_channel = create_test_channels()
p1, p2, w1, w2, _q1, _q2 = self.prepare_peers() p1, p2, w1, w2, _q1, _q2 = self.prepare_peers()
pay_req = self.prepare_invoice(w2) pay_req = self.prepare_invoice(w2)
async def pay(): async def pay():
@ -248,6 +271,7 @@ class TestPeer(ElectrumTestCase):
run(f()) run(f())
def test_channel_usage_after_closing(self): def test_channel_usage_after_closing(self):
self.alice_channel, self.bob_channel = create_test_channels()
p1, p2, w1, w2, q1, q2 = self.prepare_peers() p1, p2, w1, w2, q1, q2 = self.prepare_peers()
pay_req = self.prepare_invoice(w2) pay_req = self.prepare_invoice(w2)