From 762a9a9e7adb121b60cce92925e9cb043a52d094 Mon Sep 17 00:00:00 2001 From: SomberNight Date: Fri, 13 Jan 2023 10:55:52 +0000 Subject: [PATCH] follow-up prev: fix tests --- electrum/tests/test_lnpeer.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/electrum/tests/test_lnpeer.py b/electrum/tests/test_lnpeer.py index 47df5149c..cbbc4ab76 100644 --- a/electrum/tests/test_lnpeer.py +++ b/electrum/tests/test_lnpeer.py @@ -1532,9 +1532,9 @@ class TestPeer(TestCaseForTestnet): await peer.initialized await group.spawn(send_weird_messages()) - with self.assertRaises(lnmsg.UnknownMandatoryMsgType): + with self.assertRaises(GracefulDisconnect): run(f()) - self.assertTrue(isinstance(failing_task.exception(), lnmsg.UnknownMandatoryMsgType)) + self.assertTrue(isinstance(failing_task.exception().__cause__, lnmsg.UnknownMandatoryMsgType)) @needs_test_with_all_chacha20_implementations def test_sending_weird_messages__known_msg_with_insufficient_length(self): @@ -1562,9 +1562,9 @@ class TestPeer(TestCaseForTestnet): await peer.initialized await group.spawn(send_weird_messages()) - with self.assertRaises(lnmsg.UnexpectedEndOfStream): + with self.assertRaises(GracefulDisconnect): run(f()) - self.assertTrue(isinstance(failing_task.exception(), lnmsg.UnexpectedEndOfStream)) + self.assertTrue(isinstance(failing_task.exception().__cause__, lnmsg.UnexpectedEndOfStream)) def run(coro):