class MessageCenterTestCase(TestCase):
def setUp(self):
self.message = Message(max_num=5)
self.assertEqual(
self.message.getTime()['Recv_Readed_Time'] < time.strftime(
'%Y-%m-%d %H:%M:%S', time.localtime(time.time())), True)
self.assertEqual(
self.message.getTime()['Send_Readed_Time'] < time.strftime(
'%Y-%m-%d %H:%M:%S', time.localtime(time.time())), True)
def test_init(self):
self.assertEqual(self.message.root.get('max_num'), '5')
self.assertEqual(len(self.message.root), 0)
self.assertRaises(Exception, Message, 'test')
def test_append(self):
self.message.append({
'direction': 'Recv',
'info_type': 'test',
'info_data': 'TEST',
'user_name': '1',
'text': 'text1'
})
self.message.append({
'direction': 'Send',
'info_type': 'aaa',
'info_data': '111',
'user_name': '2',
'text': 'text2'
})
self.message.append({
'direction': 'Send',
'info_type': 'bbb',
'info_data': '222',
'user_name': '3',
'text': 'text3'
})
self.message.append({
'direction': 'Send',
'info_type': 'bbb',
'info_data': '222',
'user_name': '4',
'text': 'text4'
})
self.message.append({
'direction': 'Send',
'info_type': '',
'user_name': '5',
'text': 'text5'
})
self.assertEqual(len(self.message.root), 5)
self.assertEqual(self.message.root[0].tag, 'message')
self.assertEqual(self.message.root[0].get('direction'), 'Recv')
self.assertEqual(self.message.root[0].get('info_type'), 'test')
self.assertEqual(self.message.root[0].find('user_name').text, '1')
self.assertEqual(self.message.root[0].find('text').text, 'text1')
self.assertEqual(self.message.root[0].find('info_data').text, 'TEST')
def test_append_KeyError(self):
arg = {
'direction': '123',
'info_type': 'test',
'info_data': 'TEST',
'user_name': '1',
'text': 'text1'
}
self.assertRaises(KeyError, self.message.append, arg)
def test_append_Exception(self):
arg = {
'direction1': 'Recv',
'info_type': 'test',
'info_data': 'TEST',
'user_name': '1',
'text': 'text1'
}
self.assertRaises(Exception, self.message.append, arg)
arg = {
'direction': 'Recv',
'info_type1': 'test',
'info_data': 'TEST',
'user_name': '1',
'text': 'text1'
}
self.assertRaises(Exception, self.message.append, arg)
arg = {
'direction': 'Recv',
'info_type': 'test',
'info_data1': 'TEST',
'user_name': '1',
'text': 'text1'
}
self.assertRaises(Exception, self.message.append, arg)
arg = {
'direction': 'Recv',
'info_type': 'test',
'info_data': 'TEST',
'user_name1': '1',
'text': 'text1'
}
self.assertRaises(Exception, self.message.append, arg)
arg = {
'direction': 'Recv',
'info_type': 'test',
'info_data': 'TEST',
'user_name': '1',
'text1': 'text1'
}
self.assertRaises(Exception, self.message.append, arg)
arg = {
'direction': 'Recv',
'info_type': 'test',
'user_name': '1',
'text': 'text1'
}
self.assertRaises(Exception, self.message.append, arg)
def test_append_max_num(self):
self.message.append({
'direction': 'Recv',
'info_type': 'test',
'info_data': 'TEST',
'user_name': '1',
'text': 'text1'
})
self.message.append({
'direction': 'Send',
'info_type': 'aaa',
'info_data': '111',
'user_name': '2',
'text': 'text2'
})
self.message.append({
'direction': 'Send',
'info_type': 'bbb',
'info_data': '222',
'user_name': '3',
'text': 'text3'
})
self.message.append({
'direction': 'Send',
'info_type': 'bbb',
'info_data': '222',
'user_name': '4',
'text': 'text4'
})
self.message.append({
'direction': 'Send',
'info_type': '',
'user_name': '5',
'text': 'text5'
})
self.message.append({
'direction': 'Send',
'info_type': '',
'user_name': '6',
'text': 'text6'
})
self.assertEqual(len(self.message.root), 5)
self.assertEqual(self.message.root[0].tag, 'message')
self.assertEqual(self.message.root[0].get('direction'), 'Send')
self.assertEqual(self.message.root[0].get('info_type'), 'aaa')
self.assertEqual(self.message.root[0].find('user_name').text, '2')
self.assertEqual(self.message.root[0].find('text').text, 'text2')
self.assertEqual(self.message.root[0].find('info_data').text, '111')
self.assertRaises(Exception, self.message.index, 6)
def test_index(self):
self.message.append({
'direction': 'Recv',
'info_type': 'test',
'info_data': 'TEST',
'user_name': '1',
'text': 'text1'
})
self.message.append({
'direction': 'Send',
'info_type': 'aaa',
'info_data': '111',
'user_name': '2',
'text': 'text2'
})
self.message.append({
'direction': 'Send',
'info_type': 'bbb',
'info_data': '222',
'user_name': '3',
'text': 'text3'
})
self.message.append({
'direction': 'Send',
'info_type': 'bbb',
'info_data': '222',
'user_name': '4',
'text': 'text4'
})
self.message.append({
'direction': 'Send',
'info_type': '',
'user_name': '5',
'text': 'text5'
})
self.assertEqual(self.message.index(3)['direction'], 'Send')
self.assertEqual(self.message.index(3)['info_type'], 'bbb')
self.assertEqual(self.message.index(3)['info_data'], '222')
self.assertEqual(self.message.index(3)['user_name'], '4')
self.assertEqual(self.message.index(3)['text'], 'text4')
def test_last(self):
self.message.append({
'direction': 'Recv',
'info_type': 'test',
'info_data': 'TEST',
'user_name': '1',
'text': 'text1'
})
self.message.append({
'direction': 'Send',
'info_type': 'aaa',
'info_data': '111',
'user_name': '2',
'text': 'text2'
})
self.message.append({
'direction': 'Send',
'info_type': 'bbb',
'info_data': '222',
'user_name': '3',
'text': 'text3'
})
last = self.message.last()
self.assertEqual(self.message.last()['direction'], 'Send')
self.assertEqual(self.message.last()['info_type'], 'bbb')
self.assertEqual(self.message.last()['info_data'], '222')
self.assertEqual(self.message.last()['user_name'], '3')
self.assertEqual(self.message.last()['text'], 'text3')
def test_tostring_origin(self):
self.message.append({
'direction': 'Recv',
'info_type': 'test',
'info_data': 'TEST',
'user_name': '1',
'text': 'text1'
})
self.message.append({
'direction': 'Send',
'info_type': 'aaa',
'info_data': '111',
'user_name': '2',
'text': 'text2'
})
self.message.append({
'direction': 'Send',
'info_type': 'bbb',
'info_data': '222',
'user_name': '3',
'text': 'text3'
})
self.message.append({
'direction': 'Send',
'info_type': 'bbb',
'info_data': '222',
'user_name': '4',
'text': 'text4'
})
self.message.append({
'direction': 'Send',
'info_type': '',
'user_name': '5',
'text': 'text5'
})
tmp = Message(origin=self.message.tostring().decode())
self.assertEqual(self.message.tostring(), tmp.tostring())
def test_pretty_print(self):
self.message.append({
'direction': 'Recv',
'info_type': 'test',
'info_data': 'TEST',
'user_name': '1',
'text': 'text1'
})
self.message.append({
'direction': 'Send',
'info_type': 'aaa',
'info_data': '111',
'user_name': '2',
'text': 'text2'
})
self.message.append({
'direction': 'Send',
'info_type': 'bbb',
'info_data': '222',
'user_name': '3',
'text': 'text3'
})
self.message.append({
'direction': 'Send',
'info_type': 'bbb',
'info_data': '222',
'user_name': '4',
'text': 'text4'
})
self.message.append({
'direction': 'Send',
'info_type': '',
'user_name': '5',
'text': 'text5'
})
tmp = Message(origin=self.message.pretty_print().decode())
self.assertEqual(tmp.index(3)['direction'], 'Send')
self.assertEqual(tmp.index(3)['info_type'], 'bbb')
self.assertEqual(tmp.index(3)['info_data'], '222')
self.assertEqual(tmp.index(3)['user_name'], '4')
self.assertEqual(tmp.index(3)['text'], 'text4')
def test_sizeof(self):
self.message.append({
'direction': 'Recv',
'info_type': 'test',
'info_data': 'TEST',
'user_name': '1',
'text': 'text1'
})
self.message.append({
'direction': 'Send',
'info_type': 'aaa',
'info_data': '111',
'user_name': '2',
'text': 'text2'
})
self.message.append({
'direction': 'Send',
'info_type': 'bbb',
'info_data': '222',
'user_name': '3',
'text': 'text3'
})
self.message.append({
'direction': 'Send',
'info_type': 'bbb',
'info_data': '222',
'user_name': '4',
'text': 'text4'
})
self.message.append({
'direction': 'Send',
'info_type': '',
'user_name': '5',
'text': 'text5'
})
self.assertEqual(self.message.__sizeof__(), 5)
class MessageCenterTestCase(TestCase):
def setUp(self):
self.message = Message(max_num=5)
self.assertEqual(
self.message.getTime()['Recv_Readed_Time'] < time.strftime(
'%Y-%m-%d %H:%M:%S',
time.localtime(
time.time())),
True)
self.assertEqual(
self.message.getTime()['Send_Readed_Time'] < time.strftime(
'%Y-%m-%d %H:%M:%S',
time.localtime(
time.time())),
True)
def test_init(self):
self.assertEqual(self.message.root.get('max_num'), '5')
self.assertEqual(len(self.message.root), 0)
self.assertRaises(Exception, Message, 'test')
def test_append(self):
self.message.append(
{'direction': 'Recv', 'info_type': 'test', 'info_data': 'TEST',
'user_name': '1', 'text': 'text1'})
self.message.append(
{'direction': 'Send', 'info_type': 'aaa', 'info_data': '111',
'user_name': '2', 'text': 'text2'})
self.message.append(
{'direction': 'Send', 'info_type': 'bbb', 'info_data': '222',
'user_name': '3', 'text': 'text3'})
self.message.append(
{'direction': 'Send', 'info_type': 'bbb', 'info_data': '222',
'user_name': '4', 'text': 'text4'})
self.message.append(
{'direction': 'Send', 'info_type': '', 'user_name': '5',
'text': 'text5'})
self.assertEqual(len(self.message.root), 5)
self.assertEqual(self.message.root[0].tag, 'message')
self.assertEqual(self.message.root[0].get('direction'), 'Recv')
self.assertEqual(self.message.root[0].get('info_type'), 'test')
self.assertEqual(self.message.root[0].find('user_name').text, '1')
self.assertEqual(self.message.root[0].find('text').text, 'text1')
self.assertEqual(self.message.root[0].find('info_data').text, 'TEST')
def test_append_KeyError(self):
arg = {
'direction': '123',
'info_type': 'test',
'info_data': 'TEST',
'user_name': '1',
'text': 'text1'}
self.assertRaises(KeyError, self.message.append, arg)
def test_append_Exception(self):
arg = {
'direction1': 'Recv',
'info_type': 'test',
'info_data': 'TEST',
'user_name': '1',
'text': 'text1'}
self.assertRaises(Exception, self.message.append, arg)
arg = {
'direction': 'Recv',
'info_type1': 'test',
'info_data': 'TEST',
'user_name': '1',
'text': 'text1'}
self.assertRaises(Exception, self.message.append, arg)
arg = {
'direction': 'Recv',
'info_type': 'test',
'info_data1': 'TEST',
'user_name': '1',
'text': 'text1'}
self.assertRaises(Exception, self.message.append, arg)
arg = {
'direction': 'Recv',
'info_type': 'test',
'info_data': 'TEST',
'user_name1': '1',
'text': 'text1'}
self.assertRaises(Exception, self.message.append, arg)
arg = {
'direction': 'Recv',
'info_type': 'test',
'info_data': 'TEST',
'user_name': '1',
'text1': 'text1'}
self.assertRaises(Exception, self.message.append, arg)
arg = {
'direction': 'Recv',
'info_type': 'test',
'user_name': '1',
'text': 'text1'}
self.assertRaises(Exception, self.message.append, arg)
def test_append_max_num(self):
self.message.append(
{'direction': 'Recv', 'info_type': 'test', 'info_data': 'TEST',
'user_name': '1', 'text': 'text1'})
self.message.append(
{'direction': 'Send', 'info_type': 'aaa', 'info_data': '111',
'user_name': '2', 'text': 'text2'})
self.message.append(
{'direction': 'Send', 'info_type': 'bbb', 'info_data': '222',
'user_name': '3', 'text': 'text3'})
self.message.append(
{'direction': 'Send', 'info_type': 'bbb', 'info_data': '222',
'user_name': '4', 'text': 'text4'})
self.message.append(
{'direction': 'Send', 'info_type': '', 'user_name': '5',
'text': 'text5'})
self.message.append(
{'direction': 'Send', 'info_type': '', 'user_name': '6',
'text': 'text6'})
self.assertEqual(len(self.message.root), 5)
self.assertEqual(self.message.root[0].tag, 'message')
self.assertEqual(self.message.root[0].get('direction'), 'Send')
self.assertEqual(self.message.root[0].get('info_type'), 'aaa')
self.assertEqual(self.message.root[0].find('user_name').text, '2')
self.assertEqual(self.message.root[0].find('text').text, 'text2')
self.assertEqual(self.message.root[0].find('info_data').text, '111')
self.assertRaises(Exception, self.message.index, 6)
def test_index(self):
self.message.append(
{'direction': 'Recv', 'info_type': 'test', 'info_data': 'TEST',
'user_name': '1', 'text': 'text1'})
self.message.append(
{'direction': 'Send', 'info_type': 'aaa', 'info_data': '111',
'user_name': '2', 'text': 'text2'})
self.message.append(
{'direction': 'Send', 'info_type': 'bbb', 'info_data': '222',
'user_name': '3', 'text': 'text3'})
self.message.append(
{'direction': 'Send', 'info_type': 'bbb', 'info_data': '222',
'user_name': '4', 'text': 'text4'})
self.message.append(
{'direction': 'Send', 'info_type': '', 'user_name': '5',
'text': 'text5'})
self.assertEqual(self.message.index(3)['direction'], 'Send')
self.assertEqual(self.message.index(3)['info_type'], 'bbb')
self.assertEqual(self.message.index(3)['info_data'], '222')
self.assertEqual(self.message.index(3)['user_name'], '4')
self.assertEqual(self.message.index(3)['text'], 'text4')
def test_last(self):
self.message.append(
{'direction': 'Recv', 'info_type': 'test', 'info_data': 'TEST',
'user_name': '1', 'text': 'text1'})
self.message.append(
{'direction': 'Send', 'info_type': 'aaa', 'info_data': '111',
'user_name': '2', 'text': 'text2'})
self.message.append(
{'direction': 'Send', 'info_type': 'bbb', 'info_data': '222',
'user_name': '3', 'text': 'text3'})
last = self.message.last()
self.assertEqual(self.message.last()['direction'], 'Send')
self.assertEqual(self.message.last()['info_type'], 'bbb')
self.assertEqual(self.message.last()['info_data'], '222')
self.assertEqual(self.message.last()['user_name'], '3')
self.assertEqual(self.message.last()['text'], 'text3')
def test_tostring_origin(self):
self.message.append(
{'direction': 'Recv', 'info_type': 'test', 'info_data': 'TEST',
'user_name': '1', 'text': 'text1'})
self.message.append(
{'direction': 'Send', 'info_type': 'aaa', 'info_data': '111',
'user_name': '2', 'text': 'text2'})
self.message.append(
{'direction': 'Send', 'info_type': 'bbb', 'info_data': '222',
'user_name': '3', 'text': 'text3'})
self.message.append(
{'direction': 'Send', 'info_type': 'bbb', 'info_data': '222',
'user_name': '4', 'text': 'text4'})
self.message.append(
{'direction': 'Send', 'info_type': '', 'user_name': '5',
'text': 'text5'})
tmp = Message(origin=self.message.tostring().decode())
self.assertEqual(self.message.tostring(), tmp.tostring())
def test_pretty_print(self):
self.message.append(
{'direction': 'Recv', 'info_type': 'test', 'info_data': 'TEST',
'user_name': '1', 'text': 'text1'})
self.message.append(
{'direction': 'Send', 'info_type': 'aaa', 'info_data': '111',
'user_name': '2', 'text': 'text2'})
self.message.append(
{'direction': 'Send', 'info_type': 'bbb', 'info_data': '222',
'user_name': '3', 'text': 'text3'})
self.message.append(
{'direction': 'Send', 'info_type': 'bbb', 'info_data': '222',
'user_name': '4', 'text': 'text4'})
self.message.append(
{'direction': 'Send', 'info_type': '', 'user_name': '5',
'text': 'text5'})
tmp = Message(origin=self.message.pretty_print().decode())
self.assertEqual(tmp.index(3)['direction'], 'Send')
self.assertEqual(tmp.index(3)['info_type'], 'bbb')
self.assertEqual(tmp.index(3)['info_data'], '222')
self.assertEqual(tmp.index(3)['user_name'], '4')
self.assertEqual(tmp.index(3)['text'], 'text4')
def test_sizeof(self):
self.message.append(
{'direction': 'Recv', 'info_type': 'test', 'info_data': 'TEST',
'user_name': '1', 'text': 'text1'})
self.message.append(
{'direction': 'Send', 'info_type': 'aaa', 'info_data': '111',
'user_name': '2', 'text': 'text2'})
self.message.append(
{'direction': 'Send', 'info_type': 'bbb', 'info_data': '222',
'user_name': '3', 'text': 'text3'})
self.message.append(
{'direction': 'Send', 'info_type': 'bbb', 'info_data': '222',
'user_name': '4', 'text': 'text4'})
self.message.append(
{'direction': 'Send', 'info_type': '', 'user_name': '5',
'text': 'text5'})
self.assertEqual(self.message.__sizeof__(), 5)
