Fix: sink.ctf.fs: fix logic of make_unique_stream_file_name

[babeltrace.git] / tests / data / plugins / sink.ctf.fs / stream-names / bt_plugin_foo.py

diff --git a/tests/data/plugins/sink.ctf.fs/stream-names/bt_plugin_foo.py b/tests/data/plugins/sink.ctf.fs/stream-names/bt_plugin_foo.py
new file mode 100644 (file)
index 0000000..107be40
--- /dev/null
+++ b/tests/data/plugins/sink.ctf.fs/stream-names/bt_plugin_foo.py
@@ -0,0 +1,50 @@
+# SPDX-License-Identifier: GPL-2.0-only
+#
+# Copyright (C) 2020 EfficiOS Inc.
+#
+
+import bt2
+
+
+class TheSourceIterator(bt2._UserMessageIterator):
+    def __init__(self, config, port):
+        tc, sc, ec = port.user_data
+
+        trace = tc()
+
+        # Make two streams with the same name, to verify the stream filenames
+        # are de-duplicated properly.  Make one with the name "metadata" to
+        # verify the resulting data file is not named "metadata".
+        stream1 = trace.create_stream(sc, name='the-stream')
+        stream2 = trace.create_stream(sc, name='the-stream')
+        stream3 = trace.create_stream(sc, name='metadata')
+
+        self._msgs = [
+            self._create_stream_beginning_message(stream1),
+            self._create_stream_beginning_message(stream2),
+            self._create_stream_beginning_message(stream3),
+            self._create_event_message(ec, stream1),
+            self._create_event_message(ec, stream2),
+            self._create_event_message(ec, stream3),
+            self._create_stream_end_message(stream1),
+            self._create_stream_end_message(stream2),
+            self._create_stream_end_message(stream3),
+        ]
+
+    def __next__(self):
+        if len(self._msgs) == 0:
+            raise StopIteration
+
+        return self._msgs.pop(0)
+
+
+@bt2.plugin_component_class
+class TheSource(bt2._UserSourceComponent, message_iterator_class=TheSourceIterator):
+    def __init__(self, config, params, obj):
+        tc = self._create_trace_class()
+        sc = tc.create_stream_class()
+        ec = sc.create_event_class(name='the-event')
+        self._add_output_port('out', user_data=(tc, sc, ec))
+
+
+bt2.register_plugin(__name__, "foo")