def test_parse_keep_alive(self):
(messages, buf) = Msg.get_messages_from_buffer('\x00\x00\x00\x00')
self.assertEqual(messages[0].msg_name, 'keep_alive')
def test_parse_choke_from_buffer(self):
(msgs, buf) = Msg.get_messages_from_buffer('\x00\x00\x00\x01\x00')
self.assertEqual(msgs[0].msg_name, 'choke')
def test_choke_to_buffer_and_back(self):
choke = ChokeMsg()
choke_buf = choke.get_buffer_from_message()
self.assertEqual(choke_buf,
Msg.get_messages_from_buffer(choke_buf)[0][0].get_buffer_from_message())
def test_parse_keep_alive_extra_bytes(self):
(messages, buf) = Msg.get_messages_from_buffer('\x00\x00\x00\x00\x01')
self.assertEqual(buf, '\x01')
