1
0

follow-up prev: fix tests

This commit is contained in:
SomberNight
2023-01-13 10:55:52 +00:00
parent ab953f4a7f
commit 762a9a9e7a

View File

@@ -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):