1
0

follow-up prev: fix tests

This commit is contained in:
SomberNight
2023-01-13 10:55:52 +00:00
parent 95d1064e8b
commit 3e67cba73e

View File

@@ -1532,9 +1532,9 @@ class TestPeer(TestCaseForTestnet):
await peer.initialized await peer.initialized
await group.spawn(send_weird_messages()) await group.spawn(send_weird_messages())
with self.assertRaises(lnmsg.UnknownMandatoryMsgType): with self.assertRaises(GracefulDisconnect):
run(f()) 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 @needs_test_with_all_chacha20_implementations
def test_sending_weird_messages__known_msg_with_insufficient_length(self): def test_sending_weird_messages__known_msg_with_insufficient_length(self):
@@ -1562,9 +1562,9 @@ class TestPeer(TestCaseForTestnet):
await peer.initialized await peer.initialized
await group.spawn(send_weird_messages()) await group.spawn(send_weird_messages())
with self.assertRaises(lnmsg.UnexpectedEndOfStream): with self.assertRaises(GracefulDisconnect):
run(f()) run(f())
self.assertTrue(isinstance(failing_task.exception(), lnmsg.UnexpectedEndOfStream)) self.assertTrue(isinstance(failing_task.exception().__cause__, lnmsg.UnexpectedEndOfStream))
def run(coro): def run(coro):