tfix tests broken by previous commit - electrum - Electrum Bitcoin wallet HTML git clone https://git.parazyd.org/electrum DIR Log DIR Files DIR Refs DIR Submodules --- DIR commit 2b04cb3bc45f478c254462a9a9fa766668a7035c DIR parent 8d99fe8243d19a85962eff159562d7e7ee17085b HTML Author: ThomasV <thomasv@electrum.org> Date: Thu, 30 May 2019 11:36:51 +0200 fix tests broken by previous commit Diffstat: M electrum/tests/test_lnchannel.py | 46 +++++++++++++++++++++++-------- 1 file changed, 34 insertions(+), 12 deletions(-) --- DIR diff --git a/electrum/tests/test_lnchannel.py b/electrum/tests/test_lnchannel.py t@@ -248,11 +248,18 @@ class TestChannel(unittest.TestCase): self.assertTrue(alice_channel.signature_fits(com())) self.assertNotEqual(alice_channel.included_htlcs(REMOTE, RECEIVED, 1), []) - self.assertEqual({0: [], 1: [htlc]}, alice_channel.included_htlcs_in_their_latest_ctxs(LOCAL)) + + self.assertEqual(alice_channel.included_htlcs(REMOTE, RECEIVED, 0), []) + self.assertEqual(alice_channel.included_htlcs(REMOTE, RECEIVED, 1), [htlc]) + + self.assertEqual(bob_channel.included_htlcs(REMOTE, SENT, 0), []) self.assertEqual(bob_channel.included_htlcs(REMOTE, SENT, 1), []) - self.assertEqual({0: [], 1: []}, bob_channel.included_htlcs_in_their_latest_ctxs(REMOTE)) - self.assertEqual({0: [], 1: []}, alice_channel.included_htlcs_in_their_latest_ctxs(REMOTE)) - self.assertEqual({0: [], 1: []}, bob_channel.included_htlcs_in_their_latest_ctxs(LOCAL)) + + self.assertEqual(alice_channel.included_htlcs(REMOTE, SENT, 0), []) + self.assertEqual(alice_channel.included_htlcs(REMOTE, SENT, 1), []) + + self.assertEqual(bob_channel.included_htlcs(REMOTE, RECEIVED, 0), []) + self.assertEqual(bob_channel.included_htlcs(REMOTE, RECEIVED, 1), []) from electrum.lnutil import extract_ctn_from_tx_and_chan tx0 = str(alice_channel.force_close_tx()) t@@ -284,10 +291,17 @@ class TestChannel(unittest.TestCase): self.assertEqual(bob_channel.config[REMOTE].ctn, 0) self.assertEqual(bob_channel.included_htlcs(REMOTE, SENT, 1), [htlc]) - self.assertEqual({0: [], 1: [htlc]}, alice_channel.included_htlcs_in_their_latest_ctxs(LOCAL)) - self.assertEqual({0: [], 1: [htlc]}, bob_channel.included_htlcs_in_their_latest_ctxs(REMOTE)) - self.assertEqual({0: [], 1: []}, alice_channel.included_htlcs_in_their_latest_ctxs(REMOTE)) - self.assertEqual({0: [], 1: []}, bob_channel.included_htlcs_in_their_latest_ctxs(LOCAL)) + self.assertEqual(alice_channel.included_htlcs(REMOTE, RECEIVED, 0), []) + self.assertEqual(alice_channel.included_htlcs(REMOTE, RECEIVED, 1), [htlc]) + + self.assertEqual(bob_channel.included_htlcs(REMOTE, SENT, 0), []) + self.assertEqual(bob_channel.included_htlcs(REMOTE, SENT, 1), [htlc]) + + self.assertEqual(alice_channel.included_htlcs(REMOTE, SENT, 0), []) + self.assertEqual(alice_channel.included_htlcs(REMOTE, SENT, 1), []) + + self.assertEqual(bob_channel.included_htlcs(REMOTE, RECEIVED, 0), []) + self.assertEqual(bob_channel.included_htlcs(REMOTE, RECEIVED, 1), []) # Bob revokes his prior commitment given to him by Alice, since he now # has a valid signature for a newer commitment. t@@ -406,10 +420,18 @@ class TestChannel(unittest.TestCase): self.assertEqual(alice_channel.hm.htlcs_by_direction(REMOTE, RECEIVED), [htlc]) self.assertEqual(alice_channel.included_htlcs(REMOTE, RECEIVED, alice_channel.config[REMOTE].ctn), [htlc]) - self.assertEqual({1: [htlc], 2: [htlc]}, alice_channel.included_htlcs_in_their_latest_ctxs(LOCAL)) - self.assertEqual({1: [htlc], 2: []}, bob_channel.included_htlcs_in_their_latest_ctxs(REMOTE)) - self.assertEqual({1: [], 2: []}, alice_channel.included_htlcs_in_their_latest_ctxs(REMOTE)) - self.assertEqual({1: [], 2: []}, bob_channel.included_htlcs_in_their_latest_ctxs(LOCAL)) + + self.assertEqual(alice_channel.included_htlcs(REMOTE, RECEIVED, 1), [htlc]) + self.assertEqual(alice_channel.included_htlcs(REMOTE, RECEIVED, 2), [htlc]) + + self.assertEqual(bob_channel.included_htlcs(REMOTE, SENT, 1), [htlc]) + self.assertEqual(bob_channel.included_htlcs(REMOTE, SENT, 2), []) + + self.assertEqual(alice_channel.included_htlcs(REMOTE, SENT, 1), []) + self.assertEqual(alice_channel.included_htlcs(REMOTE, SENT, 2), []) + + self.assertEqual(bob_channel.included_htlcs(REMOTE, RECEIVED, 1), []) + self.assertEqual(bob_channel.included_htlcs(REMOTE, RECEIVED, 2), []) alice_ctx_bob_version = bob_channel.pending_commitment(REMOTE).outputs() alice_ctx_alice_version = alice_channel.pending_commitment(LOCAL).outputs()