def test_asbstract_bloxroute_message(self):
total_msg_len = 1000
msg_type = b"dummy_msg"
payload_len = total_msg_len - constants.BX_HDR_COMMON_OFF - constants.STARTING_SEQUENCE_BYTES_LEN
buffer = bytearray(total_msg_len)
message = AbstractBloxrouteMessage(msg_type=msg_type, payload_len=payload_len, buf=buffer)
raw_bytes = message.rawbytes()
self.assertEqual(total_msg_len, len(raw_bytes))
self.assertEqual(msg_type, message.msg_type())
self.assertEqual(payload_len, message.payload_len())
self.assertEqual(payload_len, len(message.payload()))
self.assertTrue(message.get_control_flags() & BloxrouteMessageControlFlags.VALID)
message.remove_control_flag(BloxrouteMessageControlFlags.VALID)
self.assertFalse(message.get_control_flags() & BloxrouteMessageControlFlags.VALID)
message.set_control_flag(BloxrouteMessageControlFlags.VALID)
self.assertTrue(message.get_control_flags() & BloxrouteMessageControlFlags.VALID)
# Trying set already set flag
message.set_control_flag(BloxrouteMessageControlFlags.VALID)
self.assertTrue(message.get_control_flags() & BloxrouteMessageControlFlags.VALID)
class MessageTest(AbstractTestCase):
def setUp(self):
self.buf1 = bytearray([i for i in range(40)])
self.payload_len1 = 20
self.msg_type1 = b"example"
self.message1 = AbstractBloxrouteMessage(msg_type=self.msg_type1,
payload_len=self.payload_len1,
buf=self.buf1)
self.buf2 = bytearray([i for i in range(24)])
self.payload_len2 = 50
self.msg_type2 = b"hello"
self.message2 = AbstractBloxrouteMessage(msg_type=self.msg_type2,
payload_len=self.payload_len2,
buf=self.buf2)
def test_init(self):
with self.assertRaises(struct.error):
AbstractBloxrouteMessage(msg_type=None,
payload_len=20,
buf=bytearray([i for i in range(40)]))
with self.assertRaises(ValueError):
AbstractBloxrouteMessage(msg_type=b"hello",
payload_len=-5,
buf=bytearray([i for i in range(40)]))
with self.assertRaises(ValueError):
AbstractBloxrouteMessage(msg_type=b"hello",
payload_len=20,
buf=bytearray([i for i in range(10)]))
self.assertEqual(self.buf1, self.message1.buf)
self.assertEqual(self.buf1, self.message1._memoryview)
self.assertEqual(self.msg_type1, self.message1._msg_type)
self.assertIsNone(self.message1._payload)
self.assertEqual(self.payload_len1, self.message1._payload_len)
def test_rawbytes(self):
self.assertIsInstance(self.message1.rawbytes(), memoryview)
self.assertEqual(self.payload_len1, self.message1._payload_len)
message2_rawbytes = self.message2.rawbytes()
self.assertEqual(self.payload_len2, self.message2._payload_len)
self.assertEqual(message2_rawbytes,
memoryview(self.message2._memoryview))
def test_msg_type(self):
self.assertEqual(self.msg_type1, self.message1.msg_type())
def test_payload_len(self):
self.assertEqual(self.payload_len1, self.message1.payload_len())
def test_payload(self):
self.assertIsNone(self.message1._payload)
self.assertEqual(
self.
buf1[AbstractBloxrouteMessage.HEADER_LENGTH:self.payload_len1 +
AbstractBloxrouteMessage.HEADER_LENGTH],
self.message1.payload())
self.assertEqual(
self.
buf1[AbstractBloxrouteMessage.HEADER_LENGTH:self.payload_len1 +
AbstractBloxrouteMessage.HEADER_LENGTH],
self.message1._payload)
