2 # Copyright (C) 2019 EfficiOS Inc.
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
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.
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.
23 class UserComponentClassTestCase(unittest
.TestCase
):
24 def _test_no_init(self
, cls
):
25 with self
.assertRaises(RuntimeError):
28 def test_no_init_source(self
):
30 bt2
._UserSourceComponent
, message_iterator_class
=bt2
._UserMessageIterator
34 self
._test
_no
_init
(MySource
)
36 def test_no_init_filter(self
):
38 bt2
._UserFilterComponent
, message_iterator_class
=bt2
._UserMessageIterator
42 self
._test
_no
_init
(MyFilter
)
44 def test_no_init_sink(self
):
45 class MySink(bt2
._UserSinkComponent
):
46 def _user_consume(self
):
49 self
._test
_no
_init
(MySink
)
51 def test_incomplete_source_no_msg_iter_cls(self
):
52 with self
.assertRaises(bt2
._IncompleteUserClass
):
54 class MySource(bt2
._UserSourceComponent
):
57 def test_incomplete_source_wrong_msg_iter_cls_type(self
):
58 with self
.assertRaises(bt2
._IncompleteUserClass
):
60 class MySource(bt2
._UserSourceComponent
, message_iterator_class
=int):
63 def test_incomplete_filter_no_msg_iter_cls(self
):
64 with self
.assertRaises(bt2
._IncompleteUserClass
):
66 class MyFilter(bt2
._UserFilterComponent
):
69 def test_incomplete_sink_no_consume_method(self
):
70 with self
.assertRaises(bt2
._IncompleteUserClass
):
72 class MySink(bt2
._UserSinkComponent
):
75 def test_minimal_source(self
):
77 bt2
._UserSourceComponent
, message_iterator_class
=bt2
._UserMessageIterator
81 def test_minimal_filter(self
):
83 bt2
._UserFilterComponent
, message_iterator_class
=bt2
._UserMessageIterator
87 def test_minimal_sink(self
):
88 class MySink(bt2
._UserSinkComponent
):
89 def _user_consume(self
):
92 def test_default_name(self
):
93 class MySink(bt2
._UserSinkComponent
):
94 def _user_consume(self
):
97 self
.assertEqual(MySink
.name
, 'MySink')
99 def test_custom_name(self
):
100 class MySink(bt2
._UserSinkComponent
, name
='salut'):
101 def _user_consume(self
):
104 self
.assertEqual(MySink
.name
, 'salut')
106 def test_invalid_custom_name(self
):
107 with self
.assertRaises(TypeError):
109 class MySink(bt2
._UserSinkComponent
, name
=23):
110 def _user_consume(self
):
113 def test_description(self
):
114 class MySink(bt2
._UserSinkComponent
):
118 Bacon ipsum dolor amet ribeye t-bone corned beef, beef jerky
119 porchetta burgdoggen prosciutto chicken frankfurter boudin
120 hamburger doner bacon turducken. Sirloin shank sausage,
121 boudin meatloaf alcatra meatball t-bone tongue pastrami
122 cupim flank tenderloin.
125 def _user_consume(self
):
128 self
.assertEqual(MySink
.description
, 'The description.')
130 def test_empty_description_no_lines(self
):
131 class MySink(bt2
._UserSinkComponent
):
136 def _user_consume(self
):
139 self
.assertIsNone(MySink
.description
)
141 def test_empty_description_no_contents(self
):
142 class MySink(bt2
._UserSinkComponent
):
148 def _user_consume(self
):
151 self
.assertIsNone(MySink
.description
)
153 def test_empty_description_single_line(self
):
154 class MySink(bt2
._UserSinkComponent
):
157 def _user_consume(self
):
160 self
.assertEqual(MySink
.description
, "my description")
163 class MySink(bt2
._UserSinkComponent
):
172 def _user_consume(self
):
175 self
.assertEqual(MySink
.help, 'The help\ntext is\nhere.')
178 class MySink(bt2
._UserSinkComponent
):
179 def _user_consume(self
):
182 self
.assertIsInstance(MySink
.addr
, int)
183 self
.assertNotEqual(MySink
.addr
, 0)
185 def test_query_not_implemented(self
):
186 class MySink(bt2
._UserSinkComponent
):
187 def _user_consume(self
):
190 with self
.assertRaises(bt2
.UnknownObject
):
191 bt2
.QueryExecutor(MySink
, 'obj', 23).query()
193 def test_query_raises(self
):
194 class MySink(bt2
._UserSinkComponent
):
195 def _user_consume(self
):
199 def _user_query(cls
, priv_query_exec
, obj
, params
):
202 with self
.assertRaises(bt2
._Error
):
203 bt2
.QueryExecutor(MySink
, 'obj', 23).query()
205 def test_query_wrong_return_type(self
):
206 class MySink(bt2
._UserSinkComponent
):
207 def _user_consume(self
):
211 def _user_query(cls
, priv_query_exec
, obj
, params
, method_obj
):
214 with self
.assertRaises(bt2
._Error
):
215 bt2
.QueryExecutor(MySink
, 'obj', 23).query()
217 def test_query_params_none(self
):
218 class MySink(bt2
._UserSinkComponent
):
219 def _user_consume(self
):
223 def _user_query(cls
, priv_query_exec
, obj
, params
, method_obj
):
224 nonlocal query_params
225 query_params
= params
230 res
= bt2
.QueryExecutor(MySink
, 'obj', params
).query()
231 self
.assertEqual(query_params
, params
)
232 self
.assertIsNone(res
)
235 def test_query_logging_level(self
):
236 class MySink(bt2
._UserSinkComponent
):
237 def _user_consume(self
):
241 def _user_query(cls
, priv_query_exec
, obj
, params
, method_obj
):
242 nonlocal query_log_level
243 query_log_level
= priv_query_exec
.logging_level
245 query_log_level
= None
246 query_exec
= bt2
.QueryExecutor(MySink
, 'obj', None)
247 query_exec
.logging_level
= bt2
.LoggingLevel
.WARNING
249 self
.assertEqual(query_log_level
, bt2
.LoggingLevel
.WARNING
)
252 def test_query_returns_none(self
):
253 class MySink(bt2
._UserSinkComponent
):
254 def _user_consume(self
):
258 def _user_query(priv_query_exec
, obj
, params
, method_obj
):
261 res
= bt2
.QueryExecutor(MySink
, 'obj', None).query()
262 self
.assertIsNone(res
)
264 def test_query_simple(self
):
265 class MySink(bt2
._UserSinkComponent
):
266 def _user_consume(self
):
270 def _user_query(cls
, priv_query_exec
, obj
, params
, method_obj
):
271 nonlocal query_params
272 query_params
= params
276 params
= ['coucou', 23, None]
277 res
= bt2
.QueryExecutor(MySink
, 'obj', params
).query()
278 self
.assertEqual(query_params
, params
)
279 self
.assertEqual(res
, 17.5)
282 def test_query_complex(self
):
283 class MySink(bt2
._UserSinkComponent
):
284 def _user_consume(self
):
288 def _user_query(cls
, priv_query_exec
, obj
, params
, method_obj
):
289 nonlocal query_params
290 query_params
= params
291 return {'null': None, 'bt2': 'BT2'}
295 'array': ['coucou', 23, None],
296 'other_map': {'yes': 'yeah', '19': 19, 'minus 1.5': -1.5},
300 res
= bt2
.QueryExecutor(MySink
, 'obj', params
).query()
301 self
.assertEqual(query_params
, params
)
302 self
.assertEqual(res
, {'null': None, 'bt2': 'BT2'})
306 class MySink(bt2
._UserSinkComponent
):
307 def _user_consume(self
):
310 self
.assertEqual(MySink
, MySink
)
313 class ComponentClassTestCase(unittest
.TestCase
):
315 class MySink(bt2
._UserSinkComponent
):
322 def _user_consume(self
):
326 def _user_query(cls
, priv_query_exec
, obj
, params
, method_obj
):
327 return [obj
, params
, 23]
329 self
._py
_comp
_cls
= MySink
331 comp
= graph
.add_component(MySink
, 'salut')
332 self
._comp
_cls
= comp
.cls
333 self
.assertIs(type(self
._comp
_cls
), bt2
._SinkComponentClassConst
)
336 del self
._py
_comp
_cls
339 def test_description(self
):
340 self
.assertEqual(self
._comp
_cls
.description
, 'The description.')
343 self
.assertEqual(self
._comp
_cls
.help, 'The help.')
346 self
.assertEqual(self
._comp
_cls
.name
, 'MySink')
349 self
.assertIsInstance(self
._comp
_cls
.addr
, int)
350 self
.assertNotEqual(self
._comp
_cls
.addr
, 0)
352 def test_eq_invalid(self
):
353 self
.assertFalse(self
._comp
_cls
== 23)
356 self
.assertEqual(self
._comp
_cls
, self
._comp
_cls
)
357 self
.assertEqual(self
._py
_comp
_cls
, self
._comp
_cls
)
359 def test_query(self
):
360 res
= bt2
.QueryExecutor(
361 self
._comp
_cls
, 'an object', {'yes': 'no', 'book': -17}
363 expected
= ['an object', {'yes': 'no', 'book': -17}, 23]
364 self
.assertEqual(res
, expected
)
367 if __name__
== '__main__':
This page took 0.036835 seconds and 4 git commands to generate.