Commit | Line | Data |
---|---|---|
d2d857a8 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 | ||
c4239792 | 19 | from bt2 import value |
811644b8 | 20 | import unittest |
811644b8 PP |
21 | import bt2 |
22 | ||
23 | ||
24 | class UserComponentClassTestCase(unittest.TestCase): | |
25 | def _test_no_init(self, cls): | |
ce4923b0 | 26 | with self.assertRaises(RuntimeError): |
811644b8 PP |
27 | cls() |
28 | ||
29 | def test_no_init_source(self): | |
5602ef81 | 30 | class MyIter(bt2._UserMessageIterator): |
811644b8 PP |
31 | def __next__(self): |
32 | raise bt2.Stop | |
33 | ||
cfbd7cf3 | 34 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
811644b8 PP |
35 | pass |
36 | ||
37 | self._test_no_init(MySource) | |
38 | ||
39 | def test_no_init_filter(self): | |
5602ef81 | 40 | class MyIter(bt2._UserMessageIterator): |
811644b8 PP |
41 | def __next__(self): |
42 | raise bt2.Stop | |
43 | ||
cfbd7cf3 | 44 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
811644b8 PP |
45 | pass |
46 | ||
47 | self._test_no_init(MyFilter) | |
48 | ||
49 | def test_no_init_sink(self): | |
50 | class MySink(bt2._UserSinkComponent): | |
6a91742b | 51 | def _user_consume(self): |
a01b452b SM |
52 | pass |
53 | ||
811644b8 PP |
54 | self._test_no_init(MySink) |
55 | ||
5602ef81 SM |
56 | def test_incomplete_source_no_msg_iter_cls(self): |
57 | class MyIter(bt2._UserMessageIterator): | |
811644b8 PP |
58 | pass |
59 | ||
cb06aa27 | 60 | with self.assertRaises(bt2._IncompleteUserClass): |
cfbd7cf3 | 61 | |
811644b8 PP |
62 | class MySource(bt2._UserSourceComponent): |
63 | pass | |
64 | ||
5602ef81 SM |
65 | def test_incomplete_source_wrong_msg_iter_cls_type(self): |
66 | class MyIter(bt2._UserMessageIterator): | |
811644b8 PP |
67 | pass |
68 | ||
cb06aa27 | 69 | with self.assertRaises(bt2._IncompleteUserClass): |
cfbd7cf3 FD |
70 | |
71 | class MySource(bt2._UserSourceComponent, message_iterator_class=int): | |
811644b8 PP |
72 | pass |
73 | ||
5602ef81 SM |
74 | def test_incomplete_filter_no_msg_iter_cls(self): |
75 | class MyIter(bt2._UserMessageIterator): | |
811644b8 PP |
76 | pass |
77 | ||
cb06aa27 | 78 | with self.assertRaises(bt2._IncompleteUserClass): |
cfbd7cf3 | 79 | |
811644b8 PP |
80 | class MyFilter(bt2._UserFilterComponent): |
81 | pass | |
82 | ||
83 | def test_incomplete_sink_no_consume_method(self): | |
5602ef81 | 84 | class MyIter(bt2._UserMessageIterator): |
811644b8 PP |
85 | pass |
86 | ||
cb06aa27 | 87 | with self.assertRaises(bt2._IncompleteUserClass): |
cfbd7cf3 | 88 | |
811644b8 PP |
89 | class MySink(bt2._UserSinkComponent): |
90 | pass | |
91 | ||
92 | def test_minimal_source(self): | |
5602ef81 | 93 | class MyIter(bt2._UserMessageIterator): |
811644b8 PP |
94 | pass |
95 | ||
cfbd7cf3 | 96 | class MySource(bt2._UserSourceComponent, message_iterator_class=MyIter): |
811644b8 PP |
97 | pass |
98 | ||
99 | def test_minimal_filter(self): | |
5602ef81 | 100 | class MyIter(bt2._UserMessageIterator): |
811644b8 PP |
101 | pass |
102 | ||
cfbd7cf3 | 103 | class MyFilter(bt2._UserFilterComponent, message_iterator_class=MyIter): |
811644b8 PP |
104 | pass |
105 | ||
106 | def test_minimal_sink(self): | |
107 | class MySink(bt2._UserSinkComponent): | |
6a91742b | 108 | def _user_consume(self): |
a01b452b SM |
109 | pass |
110 | ||
811644b8 PP |
111 | def test_default_name(self): |
112 | class MySink(bt2._UserSinkComponent): | |
6a91742b | 113 | def _user_consume(self): |
a01b452b SM |
114 | pass |
115 | ||
811644b8 PP |
116 | self.assertEqual(MySink.name, 'MySink') |
117 | ||
118 | def test_custom_name(self): | |
119 | class MySink(bt2._UserSinkComponent, name='salut'): | |
6a91742b | 120 | def _user_consume(self): |
a01b452b SM |
121 | pass |
122 | ||
811644b8 PP |
123 | self.assertEqual(MySink.name, 'salut') |
124 | ||
125 | def test_invalid_custom_name(self): | |
126 | with self.assertRaises(TypeError): | |
cfbd7cf3 | 127 | |
811644b8 | 128 | class MySink(bt2._UserSinkComponent, name=23): |
6a91742b | 129 | def _user_consume(self): |
a01b452b SM |
130 | pass |
131 | ||
811644b8 PP |
132 | def test_description(self): |
133 | class MySink(bt2._UserSinkComponent): | |
134 | """ | |
135 | The description. | |
136 | ||
137 | Bacon ipsum dolor amet ribeye t-bone corned beef, beef jerky | |
138 | porchetta burgdoggen prosciutto chicken frankfurter boudin | |
139 | hamburger doner bacon turducken. Sirloin shank sausage, | |
140 | boudin meatloaf alcatra meatball t-bone tongue pastrami | |
141 | cupim flank tenderloin. | |
142 | """ | |
143 | ||
6a91742b | 144 | def _user_consume(self): |
a01b452b SM |
145 | pass |
146 | ||
811644b8 PP |
147 | self.assertEqual(MySink.description, 'The description.') |
148 | ||
149 | def test_empty_description(self): | |
150 | class MySink(bt2._UserSinkComponent): | |
151 | """ | |
152 | """ | |
153 | ||
6a91742b | 154 | def _user_consume(self): |
a01b452b SM |
155 | pass |
156 | ||
811644b8 PP |
157 | self.assertIsNone(MySink.description) |
158 | ||
159 | def test_help(self): | |
160 | class MySink(bt2._UserSinkComponent): | |
161 | """ | |
162 | The description. | |
163 | ||
164 | The help | |
165 | text is | |
166 | here. | |
167 | """ | |
168 | ||
6a91742b | 169 | def _user_consume(self): |
a01b452b SM |
170 | pass |
171 | ||
811644b8 PP |
172 | self.assertEqual(MySink.help, 'The help\ntext is\nhere.') |
173 | ||
174 | def test_addr(self): | |
175 | class MySink(bt2._UserSinkComponent): | |
6a91742b | 176 | def _user_consume(self): |
a01b452b SM |
177 | pass |
178 | ||
811644b8 PP |
179 | self.assertIsInstance(MySink.addr, int) |
180 | self.assertNotEqual(MySink.addr, 0) | |
181 | ||
182 | def test_query_not_implemented(self): | |
183 | class MySink(bt2._UserSinkComponent): | |
6a91742b | 184 | def _user_consume(self): |
a01b452b SM |
185 | pass |
186 | ||
76b6c2f7 | 187 | with self.assertRaises(bt2.UnknownObject): |
c7eee084 | 188 | bt2.QueryExecutor().query(MySink, 'obj', 23) |
811644b8 PP |
189 | |
190 | def test_query_raises(self): | |
191 | class MySink(bt2._UserSinkComponent): | |
6a91742b | 192 | def _user_consume(self): |
a01b452b SM |
193 | pass |
194 | ||
811644b8 | 195 | @classmethod |
6a91742b | 196 | def _user_query(cls, query_exec, obj, params, log_level): |
811644b8 PP |
197 | raise ValueError |
198 | ||
694c792b | 199 | with self.assertRaises(bt2._Error): |
c7eee084 | 200 | bt2.QueryExecutor().query(MySink, 'obj', 23) |
811644b8 PP |
201 | |
202 | def test_query_wrong_return_type(self): | |
203 | class MySink(bt2._UserSinkComponent): | |
6a91742b | 204 | def _user_consume(self): |
a01b452b SM |
205 | pass |
206 | ||
811644b8 | 207 | @classmethod |
6a91742b | 208 | def _user_query(cls, query_exec, obj, params, log_level): |
811644b8 PP |
209 | return ... |
210 | ||
694c792b | 211 | with self.assertRaises(bt2._Error): |
c7eee084 | 212 | bt2.QueryExecutor().query(MySink, 'obj', 23) |
811644b8 PP |
213 | |
214 | def test_query_params_none(self): | |
215 | class MySink(bt2._UserSinkComponent): | |
6a91742b | 216 | def _user_consume(self): |
a01b452b SM |
217 | pass |
218 | ||
811644b8 | 219 | @classmethod |
6a91742b | 220 | def _user_query(cls, query_exec, obj, params, log_level): |
811644b8 PP |
221 | nonlocal query_params |
222 | query_params = params | |
223 | return None | |
224 | ||
225 | query_params = None | |
226 | params = None | |
c7eee084 | 227 | res = bt2.QueryExecutor().query(MySink, 'obj', params) |
811644b8 PP |
228 | self.assertEqual(query_params, params) |
229 | self.assertIsNone(res) | |
230 | del query_params | |
231 | ||
761e1890 PP |
232 | def test_query_logging_level(self): |
233 | class MySink(bt2._UserSinkComponent): | |
6a91742b | 234 | def _user_consume(self): |
a01b452b SM |
235 | pass |
236 | ||
761e1890 | 237 | @classmethod |
6a91742b | 238 | def _user_query(cls, query_exec, obj, params, log_level): |
761e1890 PP |
239 | nonlocal query_log_level |
240 | query_log_level = log_level | |
241 | ||
242 | query_log_level = None | |
cfbd7cf3 | 243 | res = bt2.QueryExecutor().query(MySink, 'obj', None, bt2.LoggingLevel.WARNING) |
770538dd | 244 | self.assertEqual(query_log_level, bt2.LoggingLevel.WARNING) |
761e1890 PP |
245 | del query_log_level |
246 | ||
b5947615 SM |
247 | def test_query_returns_none(self): |
248 | class MySink(bt2._UserSinkComponent): | |
6a91742b | 249 | def _user_consume(self): |
a01b452b SM |
250 | pass |
251 | ||
b5947615 | 252 | @staticmethod |
6a91742b | 253 | def _user_query(query_exec, obj, params, log_level): |
b5947615 SM |
254 | return |
255 | ||
256 | res = bt2.QueryExecutor().query(MySink, 'obj', None) | |
257 | self.assertIsNone(res) | |
258 | ||
811644b8 PP |
259 | def test_query_simple(self): |
260 | class MySink(bt2._UserSinkComponent): | |
6a91742b | 261 | def _user_consume(self): |
a01b452b SM |
262 | pass |
263 | ||
811644b8 | 264 | @classmethod |
6a91742b | 265 | def _user_query(cls, query_exec, obj, params, log_level): |
811644b8 PP |
266 | nonlocal query_params |
267 | query_params = params | |
268 | return 17.5 | |
269 | ||
270 | query_params = None | |
271 | params = ['coucou', 23, None] | |
c7eee084 | 272 | res = bt2.QueryExecutor().query(MySink, 'obj', params) |
811644b8 PP |
273 | self.assertEqual(query_params, params) |
274 | self.assertEqual(res, 17.5) | |
275 | del query_params | |
276 | ||
277 | def test_query_complex(self): | |
278 | class MySink(bt2._UserSinkComponent): | |
6a91742b | 279 | def _user_consume(self): |
a01b452b SM |
280 | pass |
281 | ||
811644b8 | 282 | @classmethod |
6a91742b | 283 | def _user_query(cls, query_exec, obj, params, log_level): |
811644b8 PP |
284 | nonlocal query_params |
285 | query_params = params | |
cfbd7cf3 | 286 | return {'null': None, 'bt2': 'BT2'} |
811644b8 PP |
287 | |
288 | query_params = None | |
289 | params = { | |
290 | 'array': ['coucou', 23, None], | |
cfbd7cf3 | 291 | 'other_map': {'yes': 'yeah', '19': 19, 'minus 1.5': -1.5}, |
811644b8 PP |
292 | 'null': None, |
293 | } | |
294 | ||
c7eee084 | 295 | res = bt2.QueryExecutor().query(MySink, 'obj', params) |
811644b8 | 296 | self.assertEqual(query_params, params) |
cfbd7cf3 | 297 | self.assertEqual(res, {'null': None, 'bt2': 'BT2'}) |
811644b8 PP |
298 | del query_params |
299 | ||
300 | def test_eq(self): | |
301 | class MySink(bt2._UserSinkComponent): | |
6a91742b | 302 | def _user_consume(self): |
a01b452b SM |
303 | pass |
304 | ||
811644b8 PP |
305 | self.assertEqual(MySink, MySink) |
306 | ||
307 | ||
68e606ef | 308 | class ComponentClassTestCase(unittest.TestCase): |
811644b8 PP |
309 | def setUp(self): |
310 | class MySink(bt2._UserSinkComponent): | |
311 | ''' | |
312 | The description. | |
313 | ||
314 | The help. | |
315 | ''' | |
cfbd7cf3 | 316 | |
6a91742b | 317 | def _user_consume(self): |
a01b452b SM |
318 | pass |
319 | ||
811644b8 | 320 | @classmethod |
6a91742b | 321 | def _user_query(cls, query_exec, obj, params, log_level): |
811644b8 PP |
322 | return [obj, params, 23] |
323 | ||
324 | self._py_comp_cls = MySink | |
325 | graph = bt2.Graph() | |
894a8df5 | 326 | comp = graph.add_component(MySink, 'salut') |
e8ac1aae | 327 | self._comp_cls = comp.cls |
3fb99a22 | 328 | self.assertTrue(issubclass(type(self._comp_cls), bt2._SinkComponentClass)) |
811644b8 PP |
329 | |
330 | def tearDown(self): | |
331 | del self._py_comp_cls | |
332 | del self._comp_cls | |
333 | ||
334 | def test_description(self): | |
335 | self.assertEqual(self._comp_cls.description, 'The description.') | |
336 | ||
337 | def test_help(self): | |
338 | self.assertEqual(self._comp_cls.help, 'The help.') | |
339 | ||
340 | def test_name(self): | |
341 | self.assertEqual(self._comp_cls.name, 'MySink') | |
342 | ||
343 | def test_addr(self): | |
344 | self.assertIsInstance(self._comp_cls.addr, int) | |
345 | self.assertNotEqual(self._comp_cls.addr, 0) | |
346 | ||
347 | def test_eq_invalid(self): | |
348 | self.assertFalse(self._comp_cls == 23) | |
349 | ||
350 | def test_eq(self): | |
351 | self.assertEqual(self._comp_cls, self._comp_cls) | |
352 | self.assertEqual(self._py_comp_cls, self._comp_cls) | |
353 | ||
354 | def test_query(self): | |
cfbd7cf3 FD |
355 | res = bt2.QueryExecutor().query( |
356 | self._comp_cls, 'an object', {'yes': 'no', 'book': -17} | |
357 | ) | |
811644b8 PP |
358 | expected = ['an object', {'yes': 'no', 'book': -17}, 23] |
359 | self.assertEqual(res, expected) |