8 @unittest.skip("this is broken")
9 class UserMessageIteratorTestCase(unittest
.TestCase
):
11 def _create_graph(src_comp_cls
):
12 class MySink(bt2
._UserSinkComponent
):
13 def __init__(self
, params
):
14 self
._add
_input
_port
('in')
19 def _port_connected(self
, port
, other_port
):
20 self
._msg
_iter
= port
.connection
.create_message_iterator()
23 src_comp
= graph
.add_component(src_comp_cls
, 'src')
24 sink_comp
= graph
.add_component(MySink
, 'sink')
25 graph
.connect_ports(src_comp
.output_ports
['out'],
26 sink_comp
.input_ports
['in'])
30 class MyIter(bt2
._UserMessageIterator
):
35 class MySource(bt2
._UserSourceComponent
,
36 message_iterator_class
=MyIter
):
37 def __init__(self
, params
):
38 self
._add
_output
_port
('out')
41 graph
= self
._create
_graph
(MySource
)
42 self
.assertTrue(initialized
)
44 def test_finalize(self
):
45 class MyIter(bt2
._UserMessageIterator
):
50 class MySource(bt2
._UserSourceComponent
,
51 message_iterator_class
=MyIter
):
52 def __init__(self
, params
):
53 self
._add
_output
_port
('out')
56 graph
= self
._create
_graph
(MySource
)
58 self
.assertTrue(finalized
)
60 def test_component(self
):
61 class MyIter(bt2
._UserMessageIterator
):
64 salut
= self
._component
._salut
66 class MySource(bt2
._UserSourceComponent
,
67 message_iterator_class
=MyIter
):
68 def __init__(self
, params
):
69 self
._add
_output
_port
('out')
73 graph
= self
._create
_graph
(MySource
)
74 self
.assertEqual(salut
, 23)
77 class MyIter(bt2
._UserMessageIterator
):
82 class MySource(bt2
._UserSourceComponent
,
83 message_iterator_class
=MyIter
):
84 def __init__(self
, params
):
85 self
._add
_output
_port
('out')
88 graph
= self
._create
_graph
(MySource
)
89 self
.assertIsNotNone(addr
)
90 self
.assertNotEqual(addr
, 0)
93 @unittest.skip("this is broken")
94 class PrivateConnectionMessageIteratorTestCase(unittest
.TestCase
):
95 def test_component(self
):
96 class MyIter(bt2
._UserMessageIterator
):
99 class MySource(bt2
._UserSourceComponent
,
100 message_iterator_class
=MyIter
):
101 def __init__(self
, params
):
102 self
._add
_output
_port
('out')
104 class MySink(bt2
._UserSinkComponent
):
105 def __init__(self
, params
):
106 self
._add
_input
_port
('in')
111 def _port_connected(self
, port
, other_port
):
112 nonlocal upstream_comp
113 self
._msg
_iter
= port
.connection
.create_message_iterator()
114 upstream_comp
= self
._msg
_iter
.component
118 src_comp
= graph
.add_component(MySource
, 'src')
119 sink_comp
= graph
.add_component(MySink
, 'sink')
120 graph
.connect_ports(src_comp
.output_ports
['out'],
121 sink_comp
.input_ports
['in'])
122 self
.assertEqual(src_comp
, upstream_comp
)
126 @unittest.skip("this is broken")
127 class OutputPortMessageIteratorTestCase(unittest
.TestCase
):
128 def test_component(self
):
129 class MyIter(bt2
._UserMessageIterator
):
134 def _build_meta(self
):
135 self
._trace
= bt2
.Trace()
136 self
._sc
= bt2
.StreamClass()
137 self
._ec
= bt2
.EventClass('salut')
138 self
._my
_int
_fc
= bt2
.IntegerFieldClass(32)
139 self
._ec
.payload_field_class
= bt2
.StructureFieldClass()
140 self
._ec
.payload_field_class
+= collections
.OrderedDict([
141 ('my_int', self
._my
_int
_fc
),
143 self
._sc
.add_event_class(self
._ec
)
144 self
._trace
.add_stream_class(self
._sc
)
145 self
._stream
= self
._sc
()
146 self
._packet
= self
._stream
.create_packet()
148 def _create_event(self
, value
):
150 ev
.payload_field
['my_int'] = value
151 ev
.packet
= self
._packet
158 msg
= bt2
.EventMessage(self
._create
_event
(self
._at
* 3))
162 class MySource(bt2
._UserSourceComponent
,
163 message_iterator_class
=MyIter
):
164 def __init__(self
, params
):
165 self
._add
_output
_port
('out')
168 src
= graph
.add_component(MySource
, 'src')
169 types
= [bt2
.EventMessage
]
170 msg_iter
= src
.output_ports
['out'].create_message_iterator(types
)
172 for at
, msg
in enumerate(msg_iter
):
173 self
.assertIsInstance(msg
, bt2
.EventMessage
)
174 self
.assertEqual(msg
.event
.event_class
.name
, 'salut')
175 field
= msg
.event
.payload_field
['my_int']
176 self
.assertEqual(field
, at
* 3)
This page took 0.035026 seconds and 4 git commands to generate.