- 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.clock_values.add(self._clock_class(1772))
- self._event.payload_field['my_int'] = 23
- self._event.packet = self._packet
-
- def tearDown(self):
- del self._trace
- del self._sc
- del self._ec
- del self._my_int_ft
- del self._clock_class
- del self._cc_prio_map
- del self._stream
- del self._packet
- del self._event
-
-
-@unittest.skip("this is broken")
-class EventMessageTestCase(_MessageTestCase):
- def test_create_no_cc_prio_map(self):
- msg = bt2.EventMessage(self._event)
- self.assertEqual(msg.event.addr, self._event.addr)
- self.assertEqual(len(msg.clock_class_priority_map), 0)
-
- def test_create_with_cc_prio_map(self):
- msg = bt2.EventMessage(self._event, self._cc_prio_map)
- self.assertEqual(msg.event.addr, self._event.addr)
- self.assertEqual(len(msg.clock_class_priority_map), 1)
- self.assertEqual(msg.clock_class_priority_map.highest_priority_clock_class.addr,
- self._clock_class.addr)
- self.assertEqual(msg.clock_class_priority_map[self._clock_class], 231)
-
- def test_eq(self):
- msg = bt2.EventMessage(self._event, self._cc_prio_map)
- event_copy = copy.copy(self._event)
- event_copy.packet = self._packet
- cc_prio_map_copy = copy.copy(self._cc_prio_map)
- msg2 = bt2.EventMessage(event_copy, cc_prio_map_copy)
- self.assertEqual(msg, msg2)
-
- def test_ne_event(self):
- msg = bt2.EventMessage(self._event, self._cc_prio_map)
- event_copy = copy.copy(self._event)
- event_copy.payload_field['my_int'] = 17
- event_copy.packet = self._packet
- cc_prio_map_copy = copy.copy(self._cc_prio_map)
- msg2 = bt2.EventMessage(event_copy, cc_prio_map_copy)
- self.assertNotEqual(msg, msg2)
-
- def test_ne_cc_prio_map(self):
- msg = bt2.EventMessage(self._event)
- event_copy = copy.copy(self._event)
- event_copy.packet = self._packet
- cc_prio_map_copy = copy.copy(self._cc_prio_map)
- msg2 = bt2.EventMessage(event_copy, cc_prio_map_copy)
- self.assertNotEqual(msg, msg2)
-
- def test_eq_invalid(self):
- msg = bt2.EventMessage(self._event)
- self.assertNotEqual(msg, 23)
-
- def test_copy(self):
- msg = bt2.EventMessage(self._event, self._cc_prio_map)
- msg2 = copy.copy(msg)
- self.assertEqual(msg, msg2)
-
- def test_deepcopy(self):
- msg = bt2.EventMessage(self._event, self._cc_prio_map)
- msg2 = copy.deepcopy(msg)
- self.assertEqual(msg, msg2)
-
-
-@unittest.skip("this is broken")
-class PacketBeginningMessageTestCase(_MessageTestCase):
- def test_create(self):
- msg = bt2.PacketBeginningMessage(self._packet)
- self.assertEqual(msg.packet.addr, self._packet.addr)
-
- def test_eq(self):
- msg = bt2.PacketBeginningMessage(self._packet)
- packet_copy = copy.copy(self._packet)
- msg2 = bt2.PacketBeginningMessage(packet_copy)
- self.assertEqual(msg, msg2)
-
- def test_ne_packet(self):
- msg = bt2.PacketBeginningMessage(self._packet)
- packet_copy = copy.copy(self._packet)
- packet_copy.header_field['hello'] = 1847
- msg2 = bt2.PacketBeginningMessage(packet_copy)
- self.assertNotEqual(msg, msg2)
-
- def test_eq_invalid(self):
- msg = bt2.PacketBeginningMessage(self._packet)
- self.assertNotEqual(msg, 23)
-
- def test_copy(self):
- msg = bt2.PacketBeginningMessage(self._packet)
- msg2 = copy.copy(msg)
- self.assertEqual(msg, msg2)
-
- def test_deepcopy(self):
- msg = bt2.PacketBeginningMessage(self._packet)
- msg2 = copy.deepcopy(msg)
- self.assertEqual(msg, msg2)
-
-
-@unittest.skip("this is broken")
-class PacketEndMessageTestCase(_MessageTestCase):
- def test_create(self):
- msg = bt2.PacketEndMessage(self._packet)
- self.assertEqual(msg.packet.addr, self._packet.addr)
-
- def test_eq(self):
- msg = bt2.PacketEndMessage(self._packet)
- packet_copy = copy.copy(self._packet)
- msg2 = bt2.PacketEndMessage(packet_copy)
- self.assertEqual(msg, msg2)
-
- def test_ne_packet(self):
- msg = bt2.PacketEndMessage(self._packet)
- packet_copy = copy.copy(self._packet)
- packet_copy.header_field['hello'] = 1847
- msg2 = bt2.PacketEndMessage(packet_copy)
- self.assertNotEqual(msg, msg2)
-
- def test_eq_invalid(self):
- msg = bt2.PacketEndMessage(self._packet)
- self.assertNotEqual(msg, 23)
-
- def test_copy(self):
- msg = bt2.PacketEndMessage(self._packet)
- msg2 = copy.copy(msg)
- self.assertEqual(msg, msg2)
-
- def test_deepcopy(self):
- msg = bt2.PacketEndMessage(self._packet)
- msg2 = copy.deepcopy(msg)
- self.assertEqual(msg, msg2)
-
-
-@unittest.skip("this is broken")
-class StreamBeginningMessageTestCase(_MessageTestCase):
- def test_create(self):
- msg = bt2.StreamBeginningMessage(self._stream)
- self.assertEqual(msg.stream.addr, self._stream.addr)
-
- def test_eq(self):
- msg = bt2.StreamBeginningMessage(self._stream)
- stream_copy = copy.copy(self._stream)
- msg2 = bt2.StreamBeginningMessage(stream_copy)
- self.assertEqual(msg, msg2)
-
- def test_ne_stream(self):
- msg = bt2.StreamBeginningMessage(self._stream)
- stream_copy = self._sc(name='salut')
- msg2 = bt2.StreamBeginningMessage(stream_copy)
- self.assertNotEqual(msg, msg2)
-
- def test_eq_invalid(self):
- msg = bt2.StreamBeginningMessage(self._stream)
- self.assertNotEqual(msg, 23)
-
- def test_copy(self):
- msg = bt2.StreamBeginningMessage(self._stream)
- msg2 = copy.copy(msg)
- self.assertEqual(msg, msg2)
-
- def test_deepcopy(self):
- msg = bt2.StreamBeginningMessage(self._stream)
- msg2 = copy.deepcopy(msg)
- self.assertEqual(msg, msg2)
-
-
-@unittest.skip("this is broken")
-class StreamEndMessageTestCase(_MessageTestCase):
- def test_create(self):
- msg = bt2.StreamEndMessage(self._stream)
- self.assertEqual(msg.stream.addr, self._stream.addr)
-
- def test_eq(self):
- msg = bt2.StreamEndMessage(self._stream)
- stream_copy = copy.copy(self._stream)
- msg2 = bt2.StreamEndMessage(stream_copy)
- self.assertEqual(msg, msg2)
-
- def test_ne_stream(self):
- msg = bt2.StreamEndMessage(self._stream)
- stream_copy = self._sc(name='salut')
- msg2 = bt2.StreamEndMessage(stream_copy)
- self.assertNotEqual(msg, msg2)
-
- def test_eq_invalid(self):
- msg = bt2.StreamEndMessage(self._stream)
- self.assertNotEqual(msg, 23)
-
- def test_copy(self):
- msg = bt2.StreamEndMessage(self._stream)
- msg2 = copy.copy(msg)
- self.assertEqual(msg, msg2)
-
- def test_deepcopy(self):
- msg = bt2.StreamEndMessage(self._stream)
- msg2 = copy.deepcopy(msg)
- self.assertEqual(msg, msg2)
-
-
-@unittest.skip("this is broken")
-class InactivityMessageTestCase(unittest.TestCase):
- 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 tearDown(self):
- del self._cc1
- del self._cc2
- del self._cc_prio_map
-
- def test_create_no_cc_prio_map(self):
- msg = bt2.InactivityMessage()
- self.assertEqual(len(msg.clock_class_priority_map), 0)
-
- def test_create_with_cc_prio_map(self):
- msg = bt2.InactivityMessage(self._cc_prio_map)
- msg.clock_values.add(self._cc1(123))
- msg.clock_values.add(self._cc2(19487))
- self.assertEqual(len(msg.clock_class_priority_map), 2)
- self.assertEqual(msg.clock_class_priority_map, self._cc_prio_map)
- self.assertEqual(msg.clock_values[self._cc1], 123)
- self.assertEqual(msg.clock_values[self._cc2], 19487)
-
- def test_eq(self):
- msg = bt2.InactivityMessage(self._cc_prio_map)
- msg.clock_values.add(self._cc1(123))
- msg.clock_values.add(self._cc2(19487))
- cc_prio_map_copy = copy.copy(self._cc_prio_map)
- msg2 = bt2.InactivityMessage(cc_prio_map_copy)
- msg2.clock_values.add(self._cc1(123))
- msg2.clock_values.add(self._cc2(19487))
- self.assertEqual(msg, msg2)
-
- def test_ne_cc_prio_map(self):
- msg = bt2.InactivityMessage(self._cc_prio_map)
- msg.clock_values.add(self._cc1(123))
- msg.clock_values.add(self._cc2(19487))
- cc_prio_map_copy = copy.copy(self._cc_prio_map)
- cc_prio_map_copy[self._cc2] = 23
- msg2 = bt2.InactivityMessage(cc_prio_map_copy)
- self.assertNotEqual(msg, msg2)
-
- def test_ne_clock_value(self):
- msg = bt2.InactivityMessage(self._cc_prio_map)
- msg.clock_values.add(self._cc1(123))
- msg.clock_values.add(self._cc2(19487))
- msg2 = bt2.InactivityMessage(self._cc_prio_map)
- msg.clock_values.add(self._cc1(123))
- msg.clock_values.add(self._cc2(1847))
- self.assertNotEqual(msg, msg2)
-
- def test_eq_invalid(self):
- msg = bt2.InactivityMessage(self._cc_prio_map)
- self.assertNotEqual(msg, 23)
-
- def test_copy(self):
- msg = bt2.InactivityMessage(self._cc_prio_map)
- msg.clock_values.add(self._cc1(123))
- msg.clock_values.add(self._cc2(19487))
- msg_copy = copy.copy(msg)
- self.assertEqual(msg, msg_copy)
- self.assertNotEqual(msg.addr, msg_copy.addr)
- self.assertEqual(msg.clock_class_priority_map.addr,
- msg_copy.clock_class_priority_map.addr)
- self.assertEqual(msg_copy.clock_values[self._cc1], 123)
- self.assertEqual(msg_copy.clock_values[self._cc2], 19487)
-
- def test_deepcopy(self):
- msg = bt2.InactivityMessage(self._cc_prio_map)
- msg.clock_values.add(self._cc1(123))
- msg.clock_values.add(self._cc2(19487))
- msg_copy = copy.deepcopy(msg)
- self.assertEqual(msg, msg_copy)
- self.assertNotEqual(msg.addr, msg_copy.addr)
- self.assertNotEqual(msg.clock_class_priority_map.addr,
- msg_copy.clock_class_priority_map.addr)
- self.assertEqual(msg.clock_class_priority_map,
- msg_copy.clock_class_priority_map)
- self.assertNotEqual(list(msg.clock_class_priority_map)[0].addr,
- list(msg_copy.clock_class_priority_map)[0].addr)
- self.assertIsNone(msg_copy.clock_values[self._cc1])
- self.assertIsNone(msg_copy.clock_values[self._cc2])
- self.assertEqual(msg_copy.clock_values[list(msg_copy.clock_class_priority_map)[0]], 123)
- self.assertEqual(msg_copy.clock_values[list(msg_copy.clock_class_priority_map)[1]], 19487)
-
-
-@unittest.skip("this is broken")
-class DiscardedPacketsMessageTestCase(unittest.TestCase):
- def setUp(self):
- self._trace = bt2.Trace()
- self._sc = bt2.StreamClass()
- self._ec = bt2.EventClass('salut')
- self._clock_class = bt2.ClockClass('yo', 1000)
- self._uint64_int_ft = bt2.IntegerFieldType(64, mapped_clock_class=self._clock_class)
- 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._sc.packet_context_field_type = bt2.StructureFieldType()
- self._sc.packet_context_field_type += collections.OrderedDict([
- ('packet_seq_num', self._my_int_ft),
- ('timestamp_begin', self._uint64_int_ft),
- ('timestamp_end', self._uint64_int_ft),
- ])
- self._trace.add_clock_class(self._clock_class)
- self._trace.add_stream_class(self._sc)
- self._stream = self._sc()
-
- def tearDown(self):
- del self._trace
- del self._sc
- del self._ec
- del self._clock_class
- del self._uint64_int_ft
- del self._my_int_ft
- del self._stream
-
- def _create_event(self, packet):
- event = self._ec()
- event.payload_field['my_int'] = 23
- event.packet = packet
- return event
-
- def _get_msg(self):