def test_ne_prio(self):
cc1 = bt2.ClockClass('meow', 1234)
cc2 = bt2.ClockClass('mix', 5678)
cc3 = bt2.ClockClass('lel', 1548)
cc_prio_map = bt2.ClockClassPriorityMap({cc1: 17, cc2: 2, cc3: 25})
other_cc1 = bt2.ClockClass('meow', 1234)
other_cc2 = bt2.ClockClass('mix', 5678)
other_cc3 = bt2.ClockClass('lel', 1548)
other_cc_prio_map = bt2.ClockClassPriorityMap({other_cc1: 17, other_cc2: 3, other_cc3: 25})
self.assertNotEqual(cc_prio_map, other_cc_prio_map)
def setUp(self):
self._trace = bt2.Trace()
self._sc = bt2.StreamClass()
self._ec = bt2.EventClass('salut')
self._my_int_ft = bt2.IntegerFieldType(32)
self._ec.payload_field_type = bt2.StructureFieldType()
self._ec.payload_field_type += collections.OrderedDict([
('my_int', self._my_int_ft),
])
self._sc.add_event_class(self._ec)
self._clock_class = bt2.ClockClass('allo', 1000)
self._trace.add_clock_class(self._clock_class)
self._trace.packet_header_field_type = bt2.StructureFieldType()
self._trace.packet_header_field_type += collections.OrderedDict([
('hello', self._my_int_ft),
])
self._trace.add_stream_class(self._sc)
self._cc_prio_map = bt2.ClockClassPriorityMap()
self._cc_prio_map[self._clock_class] = 231
self._stream = self._sc()
self._packet = self._stream.create_packet()
self._packet.header_field['hello'] = 19487
self._event = self._ec()
self._event.add_clock_value(self._clock_class(1772))
self._event.payload_field['my_int'] = 23
self._event.packet = self._packet
def test_setitem(self):
cc_prio_map = bt2.ClockClassPriorityMap()
cc1 = bt2.ClockClass('mix', 5678)
cc_prio_map[cc1] = 184
self.assertEqual(len(cc_prio_map), 1)
self.assertEqual(cc_prio_map[cc1], 184)
self.assertEqual(cc_prio_map.highest_priority_clock_class, cc1)
def test_deep_copy(self):
cc1 = bt2.ClockClass('meow', 1234)
cc2 = bt2.ClockClass('mix', 5678)
cc3 = bt2.ClockClass('lel', 1548)
cc_prio_map = bt2.ClockClassPriorityMap({cc1: 17, cc2: 2, cc3: 25})
cc_prio_map_cpy = copy.deepcopy(cc_prio_map)
self.assertEqual(cc_prio_map, cc_prio_map_cpy)
self.assertNotEqual(
list(cc_prio_map.keys())[0].addr,
list(cc_prio_map_cpy.keys())[0].addr)
self.assertNotEqual(
list(cc_prio_map.keys())[1].addr,
list(cc_prio_map_cpy.keys())[1].addr)
self.assertNotEqual(
list(cc_prio_map.keys())[2].addr,
list(cc_prio_map_cpy.keys())[2].addr)
self.assertEqual(
list(cc_prio_map.values())[0],
list(cc_prio_map_cpy.values())[0])
self.assertEqual(
list(cc_prio_map.values())[1],
list(cc_prio_map_cpy.values())[1])
self.assertEqual(
list(cc_prio_map.values())[2],
list(cc_prio_map_cpy.values())[2])
self.assertNotEqual(cc_prio_map.highest_priority_clock_class.addr,
cc_prio_map_cpy.highest_priority_clock_class.addr)
def test_getitem_wrong_key(self):
cc_prio_map = bt2.ClockClassPriorityMap()
cc1 = bt2.ClockClass('mix', 5678)
cc2 = bt2.ClockClass('mix', 5678)
cc_prio_map[cc1] = 2
with self.assertRaises(KeyError):
cc_prio_map[cc2]
def test_create_full(self):
cc1 = bt2.ClockClass('meow', 1234)
cc2 = bt2.ClockClass('mix', 5678)
cc_prio_map = bt2.ClockClassPriorityMap({cc1: 17, cc2: 2})
self.assertEqual(len(cc_prio_map), 2)
self.assertEqual(cc_prio_map[cc1], 17)
self.assertEqual(cc_prio_map[cc2], 2)
self.assertEqual(cc_prio_map.highest_priority_clock_class, cc2)
def test_iter(self):
cc1 = bt2.ClockClass('meow', 1234)
cc2 = bt2.ClockClass('mix', 5678)
cc3 = bt2.ClockClass('lel', 1548)
cc_prio_map = bt2.ClockClassPriorityMap({cc1: 17, cc2: 2, cc3: 25})
cc_prios = {}
for cc, prio in cc_prio_map.items():
cc_prios[cc] = prio
self.assertEqual(len(cc_prios), 3)
self.assertEqual(cc_prios[cc1], 17)
self.assertEqual(cc_prios[cc2], 2)
self.assertEqual(cc_prios[cc3], 25)
def setUp(self):
self._values = {
'ph_field_1': 42,
'ph_field_2': 'bla bla',
'spc_field': 'some string',
'seh_field': 'another string',
'sec_field': 68752,
'ec_field': 89,
'ef_field': 8476,
}
self._int_ft = bt2.IntegerFieldType(32)
self._str_ft = bt2.StringFieldType()
self._trace = bt2.Trace()
self._trace.packet_header_field_type = bt2.StructureFieldType()
self._trace.packet_header_field_type += collections.OrderedDict([
('ph_field_1', self._int_ft),
('ph_field_2', self._str_ft),
])
self._sc = bt2.StreamClass()
self._sc.packet_context_field_type = bt2.StructureFieldType()
self._sc.packet_context_field_type += collections.OrderedDict([
('spc_field', self._str_ft),
])
self._sc.event_header_field_type = bt2.StructureFieldType()
self._sc.event_header_field_type += collections.OrderedDict([
('seh_field', self._str_ft),
])
self._sc.event_context_field_type = bt2.StructureFieldType()
self._sc.event_context_field_type += collections.OrderedDict([
('sec_field', self._int_ft),
])
self._clock_class = bt2.ClockClass('allo', 1000)
self._trace.add_clock_class(self._clock_class)
self._ec = bt2.EventClass('event_class_name')
self._ec.context_field_type = bt2.StructureFieldType()
self._ec.context_field_type += collections.OrderedDict([
('ec_field', self._int_ft),
])
self._ec.payload_field_type = bt2.StructureFieldType()
self._ec.payload_field_type += collections.OrderedDict([
('ef_field', self._int_ft),
])
self._sc.add_event_class(self._ec)
self._trace.add_stream_class(self._sc)
self._cc_prio_map = bt2.ClockClassPriorityMap()
self._cc_prio_map[self._clock_class] = 231
self._stream = self._sc()
self._packet = self._stream.create_packet()
self._packet.header_field['ph_field_1'] = self._values['ph_field_1']
self._packet.header_field['ph_field_2'] = self._values['ph_field_2']
self._packet.context_field['spc_field'] = self._values['spc_field']
self._event = self._ec()
self._event.add_clock_value(self._clock_class(1772))
self._event.header_field['seh_field'] = self._values['seh_field']
self._event.stream_event_context_field['sec_field'] = self._values[
'sec_field']
self._event.context_field['ec_field'] = self._values['ec_field']
self._event.payload_field['ef_field'] = self._values['ef_field']
self._event.packet = self._packet
def setUp(self):
self._cc1 = bt2.ClockClass('cc1', 1000)
self._cc2 = bt2.ClockClass('cc2', 2000)
self._cc_prio_map = bt2.ClockClassPriorityMap()
self._cc_prio_map[self._cc1] = 25
self._cc_prio_map[self._cc2] = 50
def test_create_empty(self):
cc_prio_map = bt2.ClockClassPriorityMap()
self.assertEqual(len(cc_prio_map), 0)
self.assertIsNone(cc_prio_map.highest_priority_clock_class)
def test_getitem_wrong_key_type(self):
cc_prio_map = bt2.ClockClassPriorityMap()
with self.assertRaises(TypeError):
cc_prio_map[23]
def test_setitem_wrong_value_type(self):
cc_prio_map = bt2.ClockClassPriorityMap()
cc1 = bt2.ClockClass('mix', 5678)
with self.assertRaises(TypeError):
cc_prio_map[cc1] = 'priority'
def test_eq_invalid(self):
self.assertFalse(bt2.ClockClassPriorityMap() == 23)
