X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=tests%2Fdata%2Fplugins%2Fsink.ctf.fs%2Fstream-names%2Fbt_plugin_foo.py;fp=tests%2Fdata%2Fplugins%2Fsink.ctf.fs%2Fstream-names%2Fbt_plugin_foo.py;h=107be40db3eca39b02db4a70e6f2692b37341d7c;hp=0000000000000000000000000000000000000000;hb=d19b245bad1fcd0ac56045819f927e206088e8a5;hpb=72d458a311ee9b8c2cc84d8ade5754e5ea517fcb

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
index 00000000..107be40d
--- /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")