def test_from_list(self):
mess = BasicMessage(source=[(23, 2323), (33, 3333), (11, 1111)])
self.assertEquals(3, len(mess))
items = [(k, v) for k, v in mess.items()]
self.assertEquals('23', items[0][0])
self.assertEquals(2323, items[0][1])
self.assertEquals('33', items[1][0])
self.assertEquals(3333, items[1][1])
self.assertEquals('11', items[2][0])
self.assertEquals(1111, items[2][1])
def test_from_ordereddict(self):
new_dict = collections.OrderedDict()
new_dict[22] = 99
new_dict[21] = 101
new_dict[20] = 2020
mess = BasicMessage(source=new_dict)
self.assertEquals(3, len(mess))
items = [(k, v) for k, v in mess.items()]
self.assertEquals('22', items[0][0])
self.assertEquals(99, items[0][1])
self.assertEquals('21', items[1][0])
self.assertEquals(101, items[1][1])
self.assertEquals('20', items[2][0])
self.assertEquals(2020, items[2][1])
def test_from_message(self):
""" Verify that the message has the same fields as source message.
"""
source_mess = BasicMessage()
source_mess[23] = 2323
source_mess[33] = 3333
source_mess[11] = 1111
new_mess = BasicMessage(source=source_mess)
self.assertEquals(3, len(new_mess))
items = [(k, v) for k, v in new_mess.items()]
self.assertEquals('23', items[0][0])
self.assertEquals(2323, items[0][1])
self.assertEquals('33', items[1][0])
self.assertEquals(3333, items[1][1])
self.assertEquals('11', items[2][0])
self.assertEquals(1111, items[2][1])
def test_contains(self):
""" Verify that contains accepts both integer and string keys.
"""
mess = BasicMessage()
mess[22] = 'abcd'
self.assertTrue(22 in mess)
self.assertTrue('22' in mess)
self.assertFalse(33 in mess)
self.assertFalse('33' in mess)
def test_field_ordering(self):
""" Verify that the BasicMessage maintains the field ordering.
That is the fields, when iterated, will appear in the
order they were added.
"""
mess = BasicMessage()
mess[1] = 11
mess[3] = 12
mess[2] = 13
self.assertEquals(3, len(mess))
items = [(k, v) for k, v in mess.items()]
self.assertEquals('1', items[0][0])
self.assertEquals(11, items[0][1])
self.assertEquals('3', items[1][0])
self.assertEquals(12, items[1][1])
self.assertEquals('2', items[2][0])
self.assertEquals(13, items[2][1])
def test_simple_basicmessage(self):
""" Create a message and see if basic operations work.
"""
mess = BasicMessage()
self.assertEquals(0, len(mess))
mess[19] = 111
self.assertEquals(1, len(mess))
self.assertEquals(111, mess[19])
self.assertEquals(111, mess['19'])
def test_ordering(self):
mess = BasicMessage()
self.assertEquals(0, len(mess))
mess[12] = 12
mess[13] = 13
self.assertEquals(2, len(mess))
del mess[12]
self.assertEquals(1, len(mess))
self.assertTrue(13 in mess)
self.assertTrue(12 not in mess)
mess[12] = 14
self.assertEquals(2, len(mess))
items = [(k, v) for k, v in mess.items()]
self.assertEquals('13', items[0][0])
self.assertEquals(13, items[0][1])
self.assertEquals('12', items[1][0])
self.assertEquals(14, items[1][1])
