def test_encode(self):
"""Test encode of message."""
msg = Message()
cmd = msg.encode()
self.assertEqual(cmd, '0;0;0;0;0;\n')
msg.node_id = 1
msg.child_id = 255
msg.type = MessageType.internal
msg.sub_type = Internal.I_BATTERY_LEVEL
msg.ack = 0
msg.payload = 57
cmd = msg.encode()
self.assertEqual(cmd, '1;255;3;0;0;57\n')
def test_encode_bad_message(self):
"""Test encode of bad message."""
msg = Message()
msg.sub_type = 'bad'
cmd = msg.encode()
self.assertEqual(cmd, None)
