6 class UserComponentClassTestCase(unittest
.TestCase
):
7 def _test_no_init(self
, cls
):
8 with self
.assertRaises(bt2
.Error
):
11 def test_no_init_source(self
):
12 class MyIter(bt2
._UserMessageIterator
):
16 class MySource(bt2
._UserSourceComponent
,
17 message_iterator_class
=MyIter
):
20 self
._test
_no
_init
(MySource
)
22 def test_no_init_filter(self
):
23 class MyIter(bt2
._UserMessageIterator
):
27 class MyFilter(bt2
._UserFilterComponent
,
28 message_iterator_class
=MyIter
):
31 self
._test
_no
_init
(MyFilter
)
33 def test_no_init_sink(self
):
34 class MySink(bt2
._UserSinkComponent
):
38 self
._test
_no
_init
(MySink
)
40 def test_incomplete_source_no_msg_iter_cls(self
):
41 class MyIter(bt2
._UserMessageIterator
):
44 with self
.assertRaises(bt2
.IncompleteUserClass
):
45 class MySource(bt2
._UserSourceComponent
):
48 def test_incomplete_source_wrong_msg_iter_cls_type(self
):
49 class MyIter(bt2
._UserMessageIterator
):
52 with self
.assertRaises(bt2
.IncompleteUserClass
):
53 class MySource(bt2
._UserSourceComponent
,
54 message_iterator_class
=int):
57 def test_incomplete_filter_no_msg_iter_cls(self
):
58 class MyIter(bt2
._UserMessageIterator
):
61 with self
.assertRaises(bt2
.IncompleteUserClass
):
62 class MyFilter(bt2
._UserFilterComponent
):
65 def test_incomplete_sink_no_consume_method(self
):
66 class MyIter(bt2
._UserMessageIterator
):
69 with self
.assertRaises(bt2
.IncompleteUserClass
):
70 class MySink(bt2
._UserSinkComponent
):
73 def test_minimal_source(self
):
74 class MyIter(bt2
._UserMessageIterator
):
77 class MySource(bt2
._UserSourceComponent
,
78 message_iterator_class
=MyIter
):
81 def test_minimal_filter(self
):
82 class MyIter(bt2
._UserMessageIterator
):
85 class MyFilter(bt2
._UserFilterComponent
,
86 message_iterator_class
=MyIter
):
89 def test_minimal_sink(self
):
90 class MySink(bt2
._UserSinkComponent
):
94 def test_default_name(self
):
95 class MySink(bt2
._UserSinkComponent
):
99 self
.assertEqual(MySink
.name
, 'MySink')
101 def test_custom_name(self
):
102 class MySink(bt2
._UserSinkComponent
, name
='salut'):
106 self
.assertEqual(MySink
.name
, 'salut')
108 def test_invalid_custom_name(self
):
109 with self
.assertRaises(TypeError):
110 class MySink(bt2
._UserSinkComponent
, name
=23):
114 def test_description(self
):
115 class MySink(bt2
._UserSinkComponent
):
119 Bacon ipsum dolor amet ribeye t-bone corned beef, beef jerky
120 porchetta burgdoggen prosciutto chicken frankfurter boudin
121 hamburger doner bacon turducken. Sirloin shank sausage,
122 boudin meatloaf alcatra meatball t-bone tongue pastrami
123 cupim flank tenderloin.
129 self
.assertEqual(MySink
.description
, 'The description.')
131 def test_empty_description(self
):
132 class MySink(bt2
._UserSinkComponent
):
139 self
.assertIsNone(MySink
.description
)
142 class MySink(bt2
._UserSinkComponent
):
154 self
.assertEqual(MySink
.help, 'The help\ntext is\nhere.')
157 class MySink(bt2
._UserSinkComponent
):
161 self
.assertIsInstance(MySink
.addr
, int)
162 self
.assertNotEqual(MySink
.addr
, 0)
164 def test_query_not_implemented(self
):
165 class MySink(bt2
._UserSinkComponent
):
169 with self
.assertRaises(bt2
.Error
):
170 bt2
.QueryExecutor().query(MySink
, 'obj', 23)
172 def test_query_raises(self
):
173 class MySink(bt2
._UserSinkComponent
):
178 def _query(cls
, query_exec
, obj
, params
, log_level
):
181 with self
.assertRaises(bt2
.Error
):
182 bt2
.QueryExecutor().query(MySink
, 'obj', 23)
184 def test_query_wrong_return_type(self
):
185 class MySink(bt2
._UserSinkComponent
):
190 def _query(cls
, query_exec
, obj
, params
, log_level
):
193 with self
.assertRaises(bt2
.Error
):
194 bt2
.QueryExecutor().query(MySink
, 'obj', 23)
196 def test_query_params_none(self
):
197 class MySink(bt2
._UserSinkComponent
):
202 def _query(cls
, query_exec
, obj
, params
, log_level
):
203 nonlocal query_params
204 query_params
= params
209 res
= bt2
.QueryExecutor().query(MySink
, 'obj', params
)
210 self
.assertEqual(query_params
, params
)
211 self
.assertIsNone(res
)
214 def test_query_logging_level(self
):
215 class MySink(bt2
._UserSinkComponent
):
220 def _query(cls
, query_exec
, obj
, params
, log_level
):
221 nonlocal query_log_level
222 query_log_level
= log_level
224 query_log_level
= None
225 res
= bt2
.QueryExecutor().query(MySink
, 'obj', None,
226 bt2
.LoggingLevel
.WARN
)
227 self
.assertEqual(query_log_level
, bt2
.LoggingLevel
.WARN
)
230 def test_query_simple(self
):
231 class MySink(bt2
._UserSinkComponent
):
236 def _query(cls
, query_exec
, obj
, params
, log_level
):
237 nonlocal query_params
238 query_params
= params
242 params
= ['coucou', 23, None]
243 res
= bt2
.QueryExecutor().query(MySink
, 'obj', params
)
244 self
.assertEqual(query_params
, params
)
245 self
.assertEqual(res
, 17.5)
248 def test_query_complex(self
):
249 class MySink(bt2
._UserSinkComponent
):
254 def _query(cls
, query_exec
, obj
, params
, log_level
):
255 nonlocal query_params
256 query_params
= params
264 'array': ['coucou', 23, None],
273 res
= bt2
.QueryExecutor().query(MySink
, 'obj', params
)
274 self
.assertEqual(query_params
, params
)
275 self
.assertEqual(res
, {
282 class MySink(bt2
._UserSinkComponent
):
286 self
.assertEqual(MySink
, MySink
)
289 class GenericComponentClassTestCase(unittest
.TestCase
):
291 class MySink(bt2
._UserSinkComponent
):
301 def _query(cls
, query_exec
, obj
, params
, log_level
):
302 return [obj
, params
, 23]
304 self
._py
_comp
_cls
= MySink
306 comp
= graph
.add_component(MySink
, 'salut')
307 self
._comp
_cls
= comp
.cls
308 self
.assertTrue(issubclass(type(self
._comp
_cls
),
309 bt2
.component
._GenericComponentClass
))
312 del self
._py
_comp
_cls
315 def test_description(self
):
316 self
.assertEqual(self
._comp
_cls
.description
, 'The description.')
319 self
.assertEqual(self
._comp
_cls
.help, 'The help.')
322 self
.assertEqual(self
._comp
_cls
.name
, 'MySink')
325 self
.assertIsInstance(self
._comp
_cls
.addr
, int)
326 self
.assertNotEqual(self
._comp
_cls
.addr
, 0)
328 def test_eq_invalid(self
):
329 self
.assertFalse(self
._comp
_cls
== 23)
332 self
.assertEqual(self
._comp
_cls
, self
._comp
_cls
)
333 self
.assertEqual(self
._py
_comp
_cls
, self
._comp
_cls
)
335 def test_query(self
):
336 res
= bt2
.QueryExecutor().query(self
._comp
_cls
, 'an object',
337 {'yes': 'no', 'book': -17})
338 expected
= ['an object', {'yes': 'no', 'book': -17}, 23]
339 self
.assertEqual(res
, expected
)
This page took 0.04095 seconds and 4 git commands to generate.