def test_build_msg_choose_db(self):
db_name = 'my_db_name'
data = build_msg_choose_db(db_name)
msg_num, data = dword_unpack(data)
self.assertEqual(msg_num,MsgTypes.CHOOSE_DB)
self.assertEqual(struct.unpack('I',data[0:4])[0], len(db_name))
self.assertEqual(data[4:],db_name)
def test_dword_pack(self):
"""
Test dword_{pack,unpack} functions.
"""
msg = 'This is example msg'
num = 0x1337
data = dword_pack(num,msg)
self.assertEqual(len(data),len(msg) + 4)
num1,msg1 = dword_unpack(data)
self.assertEqual(num1,num)
self.assertEqual(msg1,msg)
