def test_create_message_failure(self):
message = HelloMessage(protocol_version=1, network_num=2)
with self.assertRaises(PayloadLenError):
bloxroute_message_factory.create_message_from_buffer(
message.rawbytes()[:-1])
bloxroute_message_factory.create_message_from_buffer(
message.rawbytes()[:-1])
def test_msg_ping(self):
self.connection.msg_ping(PingMessage())
self.assertTrue(self.connection.outputbuf.length > 0)
self.connection.outputbuf.flush()
output_buf_msg = self.connection.outputbuf.get_buffer()
pong_reply_msg = bloxroute_message_factory.create_message_from_buffer(
output_buf_msg)
self.assertTrue(pong_reply_msg)
self.assertTrue(isinstance(pong_reply_msg, PongMessage))
def test_pong_response_msg(self):
pong = PongMessage(nonce=50)
self.assertEqual(50, pong.nonce())
msg = bloxroute_message_factory.create_message_from_buffer(pong.buf)