Apply black code formatter on all Python code
[babeltrace.git] / tests / bindings / python / bt2 / test_event_class.py
1 #
2 # Copyright (C) 2019 EfficiOS Inc.
3 #
4 # This program is free software; you can redistribute it and/or
5 # modify it under the terms of the GNU General Public License
6 # as published by the Free Software Foundation; only version 2
7 # of the License.
8 #
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
13 #
14 # You should have received a copy of the GNU General Public License
15 # along with this program; if not, write to the Free Software
16 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
17 #
18
19 import unittest
20 import bt2
21 from utils import get_default_trace_class
22
23
24 class EventClassTestCase(unittest.TestCase):
25 def setUp(self):
26 self._tc = get_default_trace_class()
27
28 self._context_fc = self._tc.create_structure_field_class()
29 self._context_fc.append_member('allo', self._tc.create_string_field_class())
30 self._context_fc.append_member(
31 'zola', self._tc.create_signed_integer_field_class(18)
32 )
33
34 self._payload_fc = self._tc.create_structure_field_class()
35 self._payload_fc.append_member('zoom', self._tc.create_string_field_class())
36
37 self._stream_class = self._tc.create_stream_class(
38 assigns_automatic_event_class_id=True
39 )
40
41 def test_create_default(self):
42 ec = self._stream_class.create_event_class()
43
44 self.assertIsNone(ec.name, 'my_event')
45 self.assertTrue(type(ec.id), int)
46 self.assertIsNone(ec.specific_context_field_class)
47 self.assertIsNone(ec.payload_field_class)
48 self.assertIsNone(ec.emf_uri)
49 self.assertIsNone(ec.log_level)
50
51 def test_create_invalid_id(self):
52 sc = self._tc.create_stream_class(assigns_automatic_event_class_id=False)
53 with self.assertRaises(TypeError):
54 sc.create_event_class(id='lel')
55
56 def test_create_specific_context_field_class(self):
57 fc = self._tc.create_structure_field_class()
58 ec = self._stream_class.create_event_class(specific_context_field_class=fc)
59 self.assertEqual(ec.specific_context_field_class.addr, fc.addr)
60
61 def test_create_invalid_specific_context_field_class(self):
62 with self.assertRaises(TypeError):
63 self._stream_class.create_event_class(specific_context_field_class='lel')
64
65 def test_create_payload_field_class(self):
66 fc = self._tc.create_structure_field_class()
67 ec = self._stream_class.create_event_class(payload_field_class=fc)
68 self.assertEqual(ec.payload_field_class.addr, fc.addr)
69
70 def test_create_invalid_payload_field_class(self):
71 with self.assertRaises(TypeError):
72 self._stream_class.create_event_class(payload_field_class='lel')
73
74 def test_create_name(self):
75 ec = self._stream_class.create_event_class(name='viande à chien')
76 self.assertEqual(ec.name, 'viande à chien')
77
78 def test_create_invalid_name(self):
79 with self.assertRaises(TypeError):
80 self._stream_class.create_event_class(name=2)
81
82 def test_emf_uri(self):
83 ec = self._stream_class.create_event_class(emf_uri='salut')
84 self.assertEqual(ec.emf_uri, 'salut')
85
86 def test_create_invalid_emf_uri(self):
87 with self.assertRaises(TypeError):
88 self._stream_class.create_event_class(emf_uri=23)
89
90 def test_create_log_level(self):
91 ec = self._stream_class.create_event_class(
92 log_level=bt2.EventClassLogLevel.EMERGENCY
93 )
94 self.assertEqual(ec.log_level, bt2.EventClassLogLevel.EMERGENCY)
95
96 def test_create_invalid_log_level(self):
97 with self.assertRaises(ValueError):
98 self._stream_class.create_event_class(log_level='zoom')
99
100 def test_stream_class(self):
101 ec = self._stream_class.create_event_class()
102 self.assertEqual(ec.stream_class.addr, self._stream_class.addr)
This page took 0.035126 seconds and 4 git commands to generate.