Commit | Line | Data |
---|---|---|
32d2d479 MJ |
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 | ||
f6a5e476 PP |
19 | import collections |
20 | import unittest | |
f6a5e476 PP |
21 | import bt2 |
22 | ||
23 | ||
0010c8b0 | 24 | class AllMessagesTestCase(unittest.TestCase): |
f6a5e476 | 25 | def setUp(self): |
fa4c33e3 | 26 | class MyIter(bt2._UserMessageIterator): |
a4dcfa96 | 27 | def __init__(self, self_port_output): |
0010c8b0 | 28 | self._at = 0 |
61d96b89 FD |
29 | self._with_stream_msgs_clock_snapshots = self_port_output.user_data.get( |
30 | 'with_stream_msgs_clock_snapshots', False | |
31 | ) | |
f6a5e476 PP |
32 | |
33 | def __next__(self): | |
0010c8b0 SM |
34 | if test_obj._clock_class: |
35 | if self._at == 0: | |
b7cbc799 | 36 | if self._with_stream_msgs_clock_snapshots: |
61d96b89 FD |
37 | msg = self._create_stream_beginning_message( |
38 | test_obj._stream, default_clock_snapshot=self._at | |
39 | ) | |
b7cbc799 | 40 | else: |
61d96b89 FD |
41 | msg = self._create_stream_beginning_message( |
42 | test_obj._stream | |
43 | ) | |
0010c8b0 | 44 | elif self._at == 1: |
61d96b89 FD |
45 | msg = self._create_packet_beginning_message( |
46 | test_obj._packet, self._at | |
47 | ) | |
b7cbc799 | 48 | elif self._at == 2: |
61d96b89 FD |
49 | msg = self._create_event_message( |
50 | test_obj._event_class, test_obj._packet, self._at | |
51 | ) | |
b7cbc799 | 52 | elif self._at == 3: |
61d96b89 FD |
53 | msg = self._create_message_iterator_inactivity_message( |
54 | test_obj._clock_class, self._at | |
55 | ) | |
b7cbc799 | 56 | elif self._at == 4: |
61d96b89 FD |
57 | msg = self._create_discarded_events_message( |
58 | test_obj._stream, 890, self._at, self._at | |
59 | ) | |
b7cbc799 | 60 | elif self._at == 5: |
61d96b89 FD |
61 | msg = self._create_packet_end_message( |
62 | test_obj._packet, self._at | |
63 | ) | |
b7cbc799 | 64 | elif self._at == 6: |
61d96b89 FD |
65 | msg = self._create_discarded_packets_message( |
66 | test_obj._stream, 678, self._at, self._at | |
67 | ) | |
b7cbc799 SM |
68 | elif self._at == 7: |
69 | if self._with_stream_msgs_clock_snapshots: | |
61d96b89 FD |
70 | msg = self._create_stream_end_message( |
71 | test_obj._stream, default_clock_snapshot=self._at | |
72 | ) | |
b7cbc799 SM |
73 | else: |
74 | msg = self._create_stream_end_message(test_obj._stream) | |
75 | elif self._at >= 8: | |
0010c8b0 | 76 | raise bt2.Stop |
f6a5e476 | 77 | else: |
0010c8b0 SM |
78 | if self._at == 0: |
79 | msg = self._create_stream_beginning_message(test_obj._stream) | |
80 | elif self._at == 1: | |
0010c8b0 | 81 | msg = self._create_packet_beginning_message(test_obj._packet) |
b7cbc799 | 82 | elif self._at == 2: |
61d96b89 FD |
83 | msg = self._create_event_message( |
84 | test_obj._event_class, test_obj._packet | |
85 | ) | |
b7cbc799 | 86 | elif self._at == 3: |
61d96b89 FD |
87 | msg = self._create_discarded_events_message( |
88 | test_obj._stream, 890 | |
89 | ) | |
b7cbc799 | 90 | elif self._at == 4: |
0010c8b0 | 91 | msg = self._create_packet_end_message(test_obj._packet) |
b7cbc799 | 92 | elif self._at == 5: |
61d96b89 FD |
93 | msg = self._create_discarded_packets_message( |
94 | test_obj._stream, 678 | |
95 | ) | |
b7cbc799 | 96 | elif self._at == 6: |
0010c8b0 | 97 | msg = self._create_stream_end_message(test_obj._stream) |
b7cbc799 | 98 | elif self._at >= 7: |
0010c8b0 | 99 | raise bt2.Stop |
f6a5e476 PP |
100 | |
101 | self._at += 1 | |
fa4c33e3 | 102 | return msg |
f6a5e476 | 103 | |
0010c8b0 | 104 | class MySrc(bt2._UserSourceComponent, message_iterator_class=MyIter): |
b20382e2 | 105 | def __init__(self, params, obj): |
b7cbc799 | 106 | self._add_output_port('out', params) |
f6a5e476 | 107 | |
77037b2b | 108 | with_cc = bool(params['with_cc']) |
0010c8b0 SM |
109 | tc = self._create_trace_class() |
110 | if with_cc: | |
111 | cc = self._create_clock_class() | |
f6a5e476 | 112 | else: |
0010c8b0 | 113 | cc = None |
0010c8b0 | 114 | |
61d96b89 FD |
115 | sc = tc.create_stream_class( |
116 | default_clock_class=cc, | |
117 | supports_packets=True, | |
118 | packets_have_beginning_default_clock_snapshot=with_cc, | |
119 | packets_have_end_default_clock_snapshot=with_cc, | |
120 | supports_discarded_events=True, | |
121 | discarded_events_have_default_clock_snapshots=with_cc, | |
122 | supports_discarded_packets=True, | |
123 | discarded_packets_have_default_clock_snapshots=with_cc, | |
124 | ) | |
0010c8b0 SM |
125 | |
126 | # Create payload field class | |
127 | my_int_fc = tc.create_signed_integer_field_class(32) | |
128 | payload_fc = tc.create_structure_field_class() | |
61d96b89 | 129 | payload_fc += [('my_int', my_int_fc)] |
0010c8b0 SM |
130 | |
131 | ec = sc.create_event_class(name='salut', payload_field_class=payload_fc) | |
132 | ||
133 | trace = tc() | |
134 | stream = trace.create_stream(sc) | |
135 | packet = stream.create_packet() | |
136 | ||
137 | test_obj._trace = trace | |
138 | test_obj._stream = stream | |
139 | test_obj._packet = packet | |
140 | test_obj._event_class = ec | |
141 | test_obj._clock_class = cc | |
142 | ||
143 | test_obj = self | |
144 | self._graph = bt2.Graph() | |
145 | self._src = MySrc | |
146 | self._iter = MyIter | |
147 | ||
148 | def test_all_msg_with_cc(self): | |
149 | params = {'with_cc': True} | |
150 | self._src_comp = self._graph.add_component(self._src, 'my_source', params) | |
61d96b89 FD |
151 | self._msg_iter = self._graph.create_output_port_message_iterator( |
152 | self._src_comp.output_ports['out'] | |
153 | ) | |
0010c8b0 SM |
154 | |
155 | for i, msg in enumerate(self._msg_iter): | |
156 | if i == 0: | |
c946c9de | 157 | self.assertIsInstance(msg, bt2._StreamBeginningMessage) |
0010c8b0 | 158 | self.assertEqual(msg.stream.addr, self._stream.addr) |
61d96b89 | 159 | self.assertIsInstance( |
c946c9de | 160 | msg.default_clock_snapshot, bt2._UnknownClockSnapshot |
61d96b89 | 161 | ) |
0010c8b0 | 162 | elif i == 1: |
c946c9de | 163 | self.assertIsInstance(msg, bt2._PacketBeginningMessage) |
0010c8b0 SM |
164 | self.assertEqual(msg.packet.addr, self._packet.addr) |
165 | self.assertEqual(msg.default_clock_snapshot.value, i) | |
b7cbc799 | 166 | elif i == 2: |
c946c9de | 167 | self.assertIsInstance(msg, bt2._EventMessage) |
c88be1c8 | 168 | self.assertEqual(msg.event.cls.addr, self._event_class.addr) |
0010c8b0 | 169 | self.assertEqual(msg.default_clock_snapshot.value, i) |
b7cbc799 | 170 | elif i == 3: |
c946c9de | 171 | self.assertIsInstance(msg, bt2._MessageIteratorInactivityMessage) |
0010c8b0 | 172 | self.assertEqual(msg.default_clock_snapshot.value, i) |
b7cbc799 | 173 | elif i == 4: |
c946c9de | 174 | self.assertIsInstance(msg, bt2._DiscardedEventsMessage) |
0010c8b0 SM |
175 | self.assertEqual(msg.stream.addr, self._stream.addr) |
176 | self.assertEqual(msg.count, 890) | |
61d96b89 FD |
177 | self.assertEqual( |
178 | msg.stream.cls.default_clock_class.addr, self._clock_class.addr | |
179 | ) | |
0010c8b0 SM |
180 | self.assertEqual(msg.beginning_default_clock_snapshot.value, i) |
181 | self.assertEqual(msg.end_default_clock_snapshot.value, i) | |
b7cbc799 | 182 | elif i == 5: |
c946c9de | 183 | self.assertIsInstance(msg, bt2._PacketEndMessage) |
0010c8b0 SM |
184 | self.assertEqual(msg.packet.addr, self._packet.addr) |
185 | self.assertEqual(msg.default_clock_snapshot.value, i) | |
b7cbc799 | 186 | elif i == 6: |
c946c9de | 187 | self.assertIsInstance(msg, bt2._DiscardedPacketsMessage) |
0010c8b0 SM |
188 | self.assertEqual(msg.stream.addr, self._stream.addr) |
189 | self.assertEqual(msg.count, 678) | |
61d96b89 FD |
190 | self.assertEqual( |
191 | msg.stream.cls.default_clock_class.addr, self._clock_class.addr | |
192 | ) | |
0010c8b0 SM |
193 | self.assertEqual(msg.beginning_default_clock_snapshot.value, i) |
194 | self.assertEqual(msg.end_default_clock_snapshot.value, i) | |
b7cbc799 | 195 | elif i == 7: |
c946c9de | 196 | self.assertIsInstance(msg, bt2._StreamEndMessage) |
0010c8b0 | 197 | self.assertEqual(msg.stream.addr, self._stream.addr) |
61d96b89 | 198 | self.assertIsInstance( |
c946c9de | 199 | msg.default_clock_snapshot, bt2._UnknownClockSnapshot |
61d96b89 | 200 | ) |
0010c8b0 SM |
201 | else: |
202 | raise Exception | |
203 | ||
204 | def test_all_msg_without_cc(self): | |
205 | params = {'with_cc': False} | |
206 | self._src_comp = self._graph.add_component(self._src, 'my_source', params) | |
61d96b89 FD |
207 | self._msg_iter = self._graph.create_output_port_message_iterator( |
208 | self._src_comp.output_ports['out'] | |
209 | ) | |
0010c8b0 SM |
210 | |
211 | for i, msg in enumerate(self._msg_iter): | |
212 | if i == 0: | |
c946c9de | 213 | self.assertIsInstance(msg, bt2._StreamBeginningMessage) |
0010c8b0 | 214 | self.assertEqual(msg.stream.addr, self._stream.addr) |
8b18e5b9 SM |
215 | with self.assertRaisesRegex( |
216 | ValueError, 'stream class has no default clock class' | |
217 | ): | |
b7cbc799 | 218 | msg.default_clock_snapshot |
0010c8b0 | 219 | elif i == 1: |
c946c9de | 220 | self.assertIsInstance(msg, bt2._PacketBeginningMessage) |
0010c8b0 | 221 | self.assertEqual(msg.packet.addr, self._packet.addr) |
b7cbc799 | 222 | elif i == 2: |
c946c9de | 223 | self.assertIsInstance(msg, bt2._EventMessage) |
c88be1c8 | 224 | self.assertEqual(msg.event.cls.addr, self._event_class.addr) |
8b18e5b9 SM |
225 | with self.assertRaisesRegex( |
226 | ValueError, 'stream class has no default clock class' | |
227 | ): | |
0010c8b0 | 228 | msg.default_clock_snapshot |
b7cbc799 | 229 | elif i == 3: |
c946c9de | 230 | self.assertIsInstance(msg, bt2._DiscardedEventsMessage) |
0010c8b0 SM |
231 | self.assertEqual(msg.stream.addr, self._stream.addr) |
232 | self.assertEqual(msg.count, 890) | |
c88be1c8 | 233 | self.assertIsNone(msg.stream.cls.default_clock_class) |
8b18e5b9 SM |
234 | with self.assertRaisesRegex( |
235 | ValueError, | |
236 | 'such a message has no clock snapshots for this stream class', | |
237 | ): | |
0010c8b0 | 238 | msg.beginning_default_clock_snapshot |
8b18e5b9 SM |
239 | with self.assertRaisesRegex( |
240 | ValueError, | |
241 | 'such a message has no clock snapshots for this stream class', | |
242 | ): | |
0010c8b0 | 243 | msg.end_default_clock_snapshot |
b7cbc799 | 244 | elif i == 4: |
c946c9de | 245 | self.assertIsInstance(msg, bt2._PacketEndMessage) |
0010c8b0 | 246 | self.assertEqual(msg.packet.addr, self._packet.addr) |
b7cbc799 | 247 | elif i == 5: |
c946c9de | 248 | self.assertIsInstance(msg, bt2._DiscardedPacketsMessage) |
0010c8b0 SM |
249 | self.assertEqual(msg.stream.addr, self._stream.addr) |
250 | self.assertEqual(msg.count, 678) | |
c88be1c8 | 251 | self.assertIsNone(msg.stream.cls.default_clock_class) |
8b18e5b9 SM |
252 | with self.assertRaisesRegex( |
253 | ValueError, | |
254 | 'such a message has no clock snapshots for this stream class', | |
255 | ): | |
0010c8b0 | 256 | msg.beginning_default_clock_snapshot |
8b18e5b9 SM |
257 | with self.assertRaisesRegex( |
258 | ValueError, | |
259 | 'such a message has no clock snapshots for this stream class', | |
260 | ): | |
0010c8b0 | 261 | msg.end_default_clock_snapshot |
b7cbc799 | 262 | elif i == 6: |
c946c9de | 263 | self.assertIsInstance(msg, bt2._StreamEndMessage) |
0010c8b0 | 264 | self.assertEqual(msg.stream.addr, self._stream.addr) |
8b18e5b9 SM |
265 | with self.assertRaisesRegex( |
266 | ValueError, 'stream class has no default clock class' | |
267 | ): | |
b7cbc799 | 268 | msg.default_clock_snapshot |
0010c8b0 SM |
269 | else: |
270 | raise Exception | |
f6a5e476 | 271 | |
b7cbc799 | 272 | def test_msg_stream_with_clock_snapshots(self): |
61d96b89 | 273 | params = {'with_cc': True, 'with_stream_msgs_clock_snapshots': True} |
dcd94213 | 274 | |
b7cbc799 | 275 | self._src_comp = self._graph.add_component(self._src, 'my_source', params) |
61d96b89 FD |
276 | self._msg_iter = self._graph.create_output_port_message_iterator( |
277 | self._src_comp.output_ports['out'] | |
278 | ) | |
b7cbc799 | 279 | msgs = list(self._msg_iter) |
dcd94213 | 280 | |
b7cbc799 | 281 | msg_stream_beg = msgs[0] |
c946c9de | 282 | self.assertIsInstance(msg_stream_beg, bt2._StreamBeginningMessage) |
b7cbc799 | 283 | self.assertEqual(msg_stream_beg.default_clock_snapshot.value, 0) |
dcd94213 | 284 | |
b7cbc799 | 285 | msg_stream_end = msgs[7] |
c946c9de | 286 | self.assertIsInstance(msg_stream_end, bt2._StreamEndMessage) |
b7cbc799 | 287 | self.assertEqual(msg_stream_end.default_clock_snapshot.value, 7) |