X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=tests%2Fbindings%2Fpython%2Fbt2%2Ftest_stream_class.py;h=be73d46e4dafac2c842023ea971ba06d352579fe;hb=d5a22ce88b28f87cfcace263243510c1ad48b133;hp=f8d6446c0fa0a39793a7651be4081138279db9fc;hpb=61d96b89a9c8b3b523558a3c007c2aaedaba2a55;p=babeltrace.git diff --git a/tests/bindings/python/bt2/test_stream_class.py b/tests/bindings/python/bt2/test_stream_class.py index f8d6446c..be73d46e 100644 --- a/tests/bindings/python/bt2/test_stream_class.py +++ b/tests/bindings/python/bt2/test_stream_class.py @@ -17,8 +17,12 @@ # import unittest -import bt2 from utils import run_in_component_init +from bt2 import stream_class as bt2_stream_class +from bt2 import trace_class as bt2_trace_class +from bt2 import clock_class as bt2_clock_class +from bt2 import event_class as bt2_event_class +from bt2 import field_class as bt2_field_class class StreamClassTestCase(unittest.TestCase): @@ -34,6 +38,7 @@ class StreamClassTestCase(unittest.TestCase): def test_create_default(self): sc = self._tc.create_stream_class() + self.assertIs(type(sc), bt2_stream_class._StreamClass) self.assertIsNone(sc.name) self.assertIsNone(sc.packet_context_field_class) self.assertIsNone(sc.event_common_context_field_class) @@ -47,13 +52,14 @@ class StreamClassTestCase(unittest.TestCase): self.assertFalse(sc.discarded_events_have_default_clock_snapshots) self.assertFalse(sc.supports_discarded_packets) self.assertFalse(sc.discarded_packets_have_default_clock_snapshots) + self.assertEqual(len(sc.user_attributes), 0) def test_create_name(self): sc = self._tc.create_stream_class(name='bozo') self.assertEqual(sc.name, 'bozo') def test_create_invalid_name(self): - with self.assertRaises(TypeError): + with self.assertRaisesRegex(TypeError, "'int' is not a 'str' object"): self._tc.create_stream_class(name=17) def test_create_packet_context_field_class(self): @@ -62,34 +68,70 @@ class StreamClassTestCase(unittest.TestCase): packet_context_field_class=fc, supports_packets=True ) self.assertEqual(sc.packet_context_field_class, fc) + self.assertIs( + type(sc.packet_context_field_class), bt2_field_class._StructureFieldClass + ) def test_create_invalid_packet_context_field_class(self): - with self.assertRaises(TypeError): + with self.assertRaisesRegex( + TypeError, + "'int' is not a '' object", + ): self._tc.create_stream_class(packet_context_field_class=22) def test_create_invalid_packet_context_field_class_no_packets(self): fc = self._tc.create_structure_field_class() - with self.assertRaises(ValueError): + with self.assertRaisesRegex( + ValueError, + "cannot have a packet context field class without supporting packets", + ): self._tc.create_stream_class(packet_context_field_class=fc) def test_create_event_common_context_field_class(self): fc = self._tc.create_structure_field_class() sc = self._tc.create_stream_class(event_common_context_field_class=fc) self.assertEqual(sc.event_common_context_field_class, fc) + self.assertIs( + type(sc.event_common_context_field_class), + bt2_field_class._StructureFieldClass, + ) def test_create_invalid_event_common_context_field_class(self): - with self.assertRaises(TypeError): + with self.assertRaisesRegex( + TypeError, + "'int' is not a '' object", + ): self._tc.create_stream_class(event_common_context_field_class=22) def test_create_default_clock_class(self): sc = self._tc.create_stream_class(default_clock_class=self._cc) self.assertEqual(sc.default_clock_class.addr, self._cc.addr) + self.assertIs(type(sc.default_clock_class), bt2_clock_class._ClockClass) def test_create_invalid_default_clock_class(self): - with self.assertRaises(TypeError): + with self.assertRaisesRegex( + TypeError, "'int' is not a '' object" + ): self._tc.create_stream_class(default_clock_class=12) + def test_create_user_attributes(self): + sc = self._tc.create_stream_class(user_attributes={'salut': 23}) + self.assertEqual(sc.user_attributes, {'salut': 23}) + + def test_create_invalid_user_attributes(self): + with self.assertRaisesRegex( + TypeError, "cannot create value object from 'object' object" + ): + self._tc.create_stream_class(user_attributes=object()) + + def test_create_invalid_user_attributes_value_type(self): + with self.assertRaisesRegex( + TypeError, + "'SignedIntegerValue' is not a '' object", + ): + self._tc.create_stream_class(user_attributes=23) + def test_automatic_stream_ids(self): sc = self._tc.create_stream_class(assigns_automatic_stream_id=True) self.assertTrue(sc.assigns_automatic_stream_id) @@ -101,7 +143,9 @@ class StreamClassTestCase(unittest.TestCase): sc = self._tc.create_stream_class(assigns_automatic_stream_id=True) self.assertTrue(sc.assigns_automatic_stream_id) - with self.assertRaises(ValueError): + with self.assertRaisesRegex( + ValueError, "id provided, but stream class assigns automatic stream ids" + ): self._trace.create_stream(sc, id=123) def test_no_automatic_stream_ids(self): @@ -115,7 +159,10 @@ class StreamClassTestCase(unittest.TestCase): sc = self._tc.create_stream_class(assigns_automatic_stream_id=False) self.assertFalse(sc.assigns_automatic_stream_id) - with self.assertRaises(ValueError): + with self.assertRaisesRegex( + ValueError, + "id not provided, but stream class does not assign automatic stream ids", + ): self._trace.create_stream(sc) def test_automatic_event_class_ids(self): @@ -129,7 +176,10 @@ class StreamClassTestCase(unittest.TestCase): sc = self._tc.create_stream_class(assigns_automatic_event_class_id=True) self.assertTrue(sc.assigns_automatic_event_class_id) - with self.assertRaises(ValueError): + with self.assertRaisesRegex( + ValueError, + "id provided, but stream class assigns automatic event class ids", + ): sc.create_event_class(id=123) def test_no_automatic_event_class_ids(self): @@ -143,7 +193,10 @@ class StreamClassTestCase(unittest.TestCase): sc = self._tc.create_stream_class(assigns_automatic_event_class_id=False) self.assertFalse(sc.assigns_automatic_event_class_id) - with self.assertRaises(ValueError): + with self.assertRaisesRegex( + ValueError, + "id not provided, but stream class does not assign automatic event class ids", + ): sc.create_event_class() def test_supports_packets_without_cs(self): @@ -175,34 +228,40 @@ class StreamClassTestCase(unittest.TestCase): self.assertTrue(sc.packets_have_end_default_clock_snapshot) def test_supports_packets_raises_type_error(self): - with self.assertRaises(TypeError): - sc = self._tc.create_stream_class( + with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): + self._tc.create_stream_class( default_clock_class=self._cc, supports_packets=23 ) def test_packets_have_begin_default_cs_raises_type_error(self): - with self.assertRaises(TypeError): - sc = self._tc.create_stream_class( + with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): + self._tc.create_stream_class( default_clock_class=self._cc, packets_have_beginning_default_clock_snapshot=23, ) def test_packets_have_end_default_cs_raises_type_error(self): - with self.assertRaises(TypeError): - sc = self._tc.create_stream_class( + with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): + self._tc.create_stream_class( default_clock_class=self._cc, packets_have_end_default_clock_snapshot=23 ) def test_does_not_support_packets_raises_with_begin_cs(self): - with self.assertRaises(ValueError): - sc = self._tc.create_stream_class( + with self.assertRaisesRegex( + ValueError, + "cannot not support packets, but have packet beginning default clock snapshot", + ): + self._tc.create_stream_class( default_clock_class=self._cc, packets_have_beginning_default_clock_snapshot=True, ) def test_does_not_support_packets_raises_with_end_cs(self): - with self.assertRaises(ValueError): - sc = self._tc.create_stream_class( + with self.assertRaisesRegex( + ValueError, + "cannot not support packets, but have packet end default clock snapshots", + ): + self._tc.create_stream_class( default_clock_class=self._cc, packets_have_end_default_clock_snapshot=True, ) @@ -224,21 +283,24 @@ class StreamClassTestCase(unittest.TestCase): self.assertTrue(sc.discarded_events_have_default_clock_snapshots) def test_supports_discarded_events_raises_type_error(self): - with self.assertRaises(TypeError): - sc = self._tc.create_stream_class( + with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): + self._tc.create_stream_class( default_clock_class=self._cc, supports_discarded_events=23 ) def test_discarded_events_have_default_cs_raises_type_error(self): - with self.assertRaises(TypeError): - sc = self._tc.create_stream_class( + with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): + self._tc.create_stream_class( default_clock_class=self._cc, discarded_events_have_default_clock_snapshots=23, ) def test_does_not_support_discarded_events_raises_with_cs(self): - with self.assertRaises(ValueError): - sc = self._tc.create_stream_class( + with self.assertRaisesRegex( + ValueError, + "cannot not support discarded events, but have default clock snapshots for discarded event messages", + ): + self._tc.create_stream_class( default_clock_class=self._cc, discarded_events_have_default_clock_snapshots=True, ) @@ -263,30 +325,35 @@ class StreamClassTestCase(unittest.TestCase): self.assertTrue(sc.discarded_packets_have_default_clock_snapshots) def test_supports_discarded_packets_raises_without_packet_support(self): - with self.assertRaises(ValueError): - sc = self._tc.create_stream_class( + with self.assertRaisesRegex( + ValueError, "cannot support discarded packets, but not support packets" + ): + self._tc.create_stream_class( default_clock_class=self._cc, supports_discarded_packets=True ) def test_supports_discarded_packets_raises_type_error(self): - with self.assertRaises(TypeError): - sc = self._tc.create_stream_class( + with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): + self._tc.create_stream_class( default_clock_class=self._cc, supports_discarded_packets=23, supports_packets=True, ) def test_discarded_packets_have_default_cs_raises_type_error(self): - with self.assertRaises(TypeError): - sc = self._tc.create_stream_class( + with self.assertRaisesRegex(TypeError, "'int' is not a 'bool' object"): + self._tc.create_stream_class( default_clock_class=self._cc, discarded_packets_have_default_clock_snapshots=23, supports_packets=True, ) def test_does_not_support_discarded_packets_raises_with_cs(self): - with self.assertRaises(ValueError): - sc = self._tc.create_stream_class( + with self.assertRaisesRegex( + ValueError, + "cannot not support discarded packets, but have default clock snapshots for discarded packet messages", + ): + self._tc.create_stream_class( default_clock_class=self._cc, discarded_packets_have_default_clock_snapshots=True, supports_packets=True, @@ -295,6 +362,7 @@ class StreamClassTestCase(unittest.TestCase): def test_trace_class(self): sc = self._tc.create_stream_class() self.assertEqual(sc.trace_class.addr, self._tc.addr) + self.assertIs(type(sc.trace_class), bt2_trace_class._TraceClass) def _create_stream_class_with_event_classes(self): sc = self._tc.create_stream_class(assigns_automatic_event_class_id=False) @@ -306,18 +374,20 @@ class StreamClassTestCase(unittest.TestCase): sc, ec1, ec2 = self._create_stream_class_with_event_classes() self.assertEqual(sc[23].addr, ec1.addr) + self.assertEqual(type(sc[23]), bt2_event_class._EventClass) self.assertEqual(sc[17].addr, ec2.addr) + self.assertEqual(type(sc[17]), bt2_event_class._EventClass) def test_getitem_wrong_key_type(self): sc, _, _ = self._create_stream_class_with_event_classes() - with self.assertRaises(TypeError): + with self.assertRaisesRegex(TypeError, "'str' is not an 'int' object"): sc['event23'] def test_getitem_wrong_key(self): sc, _, _ = self._create_stream_class_with_event_classes() - with self.assertRaises(KeyError): + with self.assertRaisesRegex(KeyError, '19'): sc[19] def test_len(self): @@ -330,3 +400,7 @@ class StreamClassTestCase(unittest.TestCase): ec_ids = sorted(sc) self.assertEqual(ec_ids, [17, 23]) + + +if __name__ == '__main__': + unittest.main()