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 UserComponentTestCase(unittest
.TestCase
):
25 def _create_comp(comp_cls
, name
=None, log_level
=bt2
.LoggingLevel
.NONE
):
31 return graph
.add_component(comp_cls
, name
, logging_level
=log_level
)
34 class MySink(bt2
._UserSinkComponent
):
35 def __init__(comp_self
, params
, obj
):
36 self
.assertEqual(comp_self
.name
, 'yaes')
38 def _user_consume(self
):
41 comp
= self
._create
_comp
(MySink
, 'yaes')
43 def test_logging_level(self
):
44 class MySink(bt2
._UserSinkComponent
):
45 def __init__(comp_self
, params
, obj
):
46 self
.assertEqual(comp_self
.logging_level
, bt2
.LoggingLevel
.INFO
)
48 def _user_consume(self
):
51 comp
= self
._create
_comp
(MySink
, 'yaes', bt2
.LoggingLevel
.INFO
)
54 class MySink(bt2
._UserSinkComponent
):
55 def __init__(comp_self
, params
, obj
):
56 self
.assertEqual(comp_self
.cls
, MySink
)
58 def _user_consume(self
):
61 self
._create
_comp
(MySink
)
64 class MySink(bt2
._UserSinkComponent
):
65 def __init__(comp_self
, params
, obj
):
66 self
.assertIsInstance(comp_self
.addr
, int)
67 self
.assertNotEqual(comp_self
.addr
, 0)
69 def _user_consume(self
):
72 self
._create
_comp
(MySink
)
74 def test_finalize(self
):
77 class MySink(bt2
._UserSinkComponent
):
78 def _user_consume(self
):
81 def _user_finalize(comp_self
):
86 comp
= graph
.add_component(MySink
, 'lel')
90 self
.assertTrue(finalized
)
93 class GenericComponentTestCase(unittest
.TestCase
):
95 def _create_comp(comp_cls
, name
=None, log_level
=bt2
.LoggingLevel
.NONE
):
101 return graph
.add_component(comp_cls
, name
, logging_level
=log_level
)
104 class MySink(bt2
._UserSinkComponent
):
105 def _user_consume(self
):
108 comp
= self
._create
_comp
(MySink
, 'yaes')
109 self
.assertEqual(comp
.name
, 'yaes')
111 def test_logging_level(self
):
112 class MySink(bt2
._UserSinkComponent
):
113 def _user_consume(self
):
116 comp
= self
._create
_comp
(MySink
, 'yaes', bt2
.LoggingLevel
.WARNING
)
117 self
.assertEqual(comp
.logging_level
, bt2
.LoggingLevel
.WARNING
)
119 def test_class(self
):
120 class MySink(bt2
._UserSinkComponent
):
121 def _user_consume(self
):
124 comp
= self
._create
_comp
(MySink
)
125 self
.assertEqual(comp
.cls
, MySink
)
128 class MySink(bt2
._UserSinkComponent
):
129 def _user_consume(self
):
132 comp
= self
._create
_comp
(MySink
)
133 self
.assertIsInstance(comp
.addr
, int)
134 self
.assertNotEqual(comp
.addr
, 0)
This page took 0.034493 seconds and 4 git commands to generate.