From 6d11a20cc8298577390c45f47d5e36d35931459f Mon Sep 17 00:00:00 2001 From: Isaac Cook Date: Sun, 16 Nov 2014 14:03:44 -0600 Subject: [PATCH] Add a test for proper exception raising on bad start --- bitcoin/tests/test_messages.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/bitcoin/tests/test_messages.py b/bitcoin/tests/test_messages.py index 5b4c7573..e4c39a22 100644 --- a/bitcoin/tests/test_messages.py +++ b/bitcoin/tests/test_messages.py @@ -114,6 +114,12 @@ def test_read_msg_verack(self): m = MsgSerializable.stream_deserialize(f) self.assertEqual(m.command, msg_verack.command) + def test_fail_invalid_message(self): + bad_verack_bytes = b'\xf8' + self.verackbytes[1:] + f = BytesIO(bad_verack_bytes) + with self.assertRaises(ValueError): + MsgSerializable.stream_deserialize(f) + def test_msg_verack_to_bytes(self): m = msg_verack() b = m.to_bytes()