bt2: let components attach "user data" to ports

[babeltrace.git] / tests / bindings / python / bt2 / test_message_iterator.py

diff --git a/tests/bindings/python/bt2/test_message_iterator.py b/tests/bindings/python/bt2/test_message_iterator.py
index 0ad56329da087136891185765fa6f18d5b83f8df..9c8277b7f6b1ef2e3e0054d9f51c0fffdd7aae0b 100644 (file)
--- a/tests/bindings/python/bt2/test_message_iterator.py
+++ b/tests/bindings/python/bt2/test_message_iterator.py
@@ -40,13 +40,14 @@ class UserMessageIteratorTestCase(unittest.TestCase):
                        message_iterator_class=MyIter):
             def __init__(self, params):
                 nonlocal the_output_port_from_source
-                the_output_port_from_source = self._add_output_port('out')
+                the_output_port_from_source = self._add_output_port('out', 'user data')
 
         initialized = False
         graph = self._create_graph(MySource)
         graph.run()
         self.assertTrue(initialized)
         self.assertEqual(the_output_port_from_source.addr, the_output_port_from_iter.addr)
+        self.assertEqual(the_output_port_from_iter.user_data, 'user data')
 
     def test_finalize(self):
         class MyIter(bt2._UserMessageIterator):