X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=tests%2Fdata%2Fplugins%2Fflt.utils.trimmer%2Fbt_plugin_trimmer_test.py;h=857fcc927db0d1a09c898d8e50b6c8307debd30c;hp=0fdb78f9f56ff4f94b5cdfff10c69ec67200b83f;hb=f5567ea88d172767b34373bc6e402da8bfd85ef8;hpb=cfbd7cf3bde05e8a6606478889dcd663604ef7b5

diff --git a/tests/data/plugins/flt.utils.trimmer/bt_plugin_trimmer_test.py b/tests/data/plugins/flt.utils.trimmer/bt_plugin_trimmer_test.py
index 0fdb78f9..857fcc92 100644
--- a/tests/data/plugins/flt.utils.trimmer/bt_plugin_trimmer_test.py
+++ b/tests/data/plugins/flt.utils.trimmer/bt_plugin_trimmer_test.py
@@ -1,39 +1,54 @@
+# SPDX-License-Identifier: GPL-2.0-only
+#
+# Copyright (C) 2019 EfficiOS Inc.
+#
+
 import bt2
 
 
 class TheIteratorOfAllEvil(bt2._UserMessageIterator):
-    def __init__(self, port):
+    def __init__(self, config, port):
         tc, sc, ec1, ec2, params = port.user_data
         trace = tc()
         stream = trace.create_stream(sc)
 
-        # Test with and without packets, once packets are optional.
-        packet = stream.create_packet()
+        if params["with-packet-msgs"]:
+            packet = stream.create_packet()
 
-        if params['with-stream-msgs-cs']:
+        if params["with-stream-msgs-cs"]:
             sb_msg = self._create_stream_beginning_message(stream, 100)
         else:
             sb_msg = self._create_stream_beginning_message(stream)
 
-        ev_msg1 = self._create_event_message(ec1, packet, 300)
-        ev_msg2 = self._create_event_message(ec2, packet, 400)
+        parent = packet if params["with-packet-msgs"] else stream
+
+        ev_msg1 = self._create_event_message(ec1, parent, 300)
+        ev_msg2 = self._create_event_message(ec2, parent, 400)
 
-        if params['with-stream-msgs-cs']:
+        if params["with-stream-msgs-cs"]:
             se_msg = self._create_stream_end_message(stream, 1000)
         else:
             se_msg = self._create_stream_end_message(stream)
 
-        self._msgs = [
-            sb_msg,
-            self._create_packet_beginning_message(packet, 200),
-            ev_msg1,
-            ev_msg2,
-            self._create_packet_end_message(packet, 900),
-            se_msg,
-        ]
+        self._msgs = []
+
+        self._msgs.append(sb_msg)
+
+        if params["with-packet-msgs"]:
+            self._msgs.append(self._create_packet_beginning_message(packet, 200))
+
+        self._msgs.append(ev_msg1)
+        self._msgs.append(ev_msg2)
+
+        if params["with-packet-msgs"]:
+            self._msgs.append(self._create_packet_end_message(packet, 900))
+
+        self._msgs.append(se_msg)
+
         self._at = 0
+        config.can_seek_forward = True
 
-    def _seek_beginning(self):
+    def _user_seek_beginning(self):
         self._at = 0
 
     def __next__(self):
@@ -49,22 +64,24 @@ class TheIteratorOfAllEvil(bt2._UserMessageIterator):
 class TheSourceOfAllEvil(
     bt2._UserSourceComponent, message_iterator_class=TheIteratorOfAllEvil
 ):
-    def __init__(self, params):
+    def __init__(self, config, params, obj):
         tc = self._create_trace_class()
 
+        with_packets = bool(params["with-packet-msgs"])
+
         # Use a clock class with an offset, so we can test with --begin or --end
         # smaller than this offset (in other words, a time that it's not
         # possible to represent with this clock class).
         cc = self._create_clock_class(frequency=1, offset=bt2.ClockClassOffset(10000))
         sc = tc.create_stream_class(
             default_clock_class=cc,
-            supports_packets=True,
-            packets_have_beginning_default_clock_snapshot=True,
-            packets_have_end_default_clock_snapshot=True,
+            supports_packets=with_packets,
+            packets_have_beginning_default_clock_snapshot=with_packets,
+            packets_have_end_default_clock_snapshot=with_packets,
         )
-        ec1 = sc.create_event_class(name='event 1')
-        ec2 = sc.create_event_class(name='event 2')
-        self._add_output_port('out', (tc, sc, ec1, ec2, params))
+        ec1 = sc.create_event_class(name="event 1")
+        ec2 = sc.create_event_class(name="event 2")
+        self._add_output_port("out", (tc, sc, ec1, ec2, params))
 
 
-bt2.register_plugin(__name__, 'test-trimmer')
+bt2.register_plugin(__name__, "test-trimmer")