From 2fafa4c187aefe444211b3dacdf789d27647bae1 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=A9mie=20Galarneau?= Date: Thu, 2 Feb 2017 20:37:54 -0500 Subject: [PATCH] Tests: add empty packet writer test MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérémie Galarneau --- tests/Makefile.am | 5 +- tests/lib/writer/Makefile.am | 9 +- .../writer/test_ctf_writer_empty_packet.py | 108 ++++++++++++++++++ 3 files changed, 117 insertions(+), 5 deletions(-) create mode 100755 tests/lib/writer/test_ctf_writer_empty_packet.py diff --git a/tests/Makefile.am b/tests/Makefile.am index 80d8d03b..90340455 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -31,6 +31,7 @@ TESTS += lib/test_dwarf_complete \ endif if USE_PYTHON -TESTS += bin/intersection/test_multi_trace_intersection.py -TESTS += lib/writer/test_ctf_writer_no_packet_context.py +TESTS += bin/intersection/test_multi_trace_intersection.py \ + lib/writer/test_ctf_writer_no_packet_context.py \ + lib/writer/test_ctf_writer_empty_packet.py endif diff --git a/tests/lib/writer/Makefile.am b/tests/lib/writer/Makefile.am index 6a881bbb..eb648367 100644 --- a/tests/lib/writer/Makefile.am +++ b/tests/lib/writer/Makefile.am @@ -1,7 +1,10 @@ -check_SCRIPTS = test_ctf_writer_no_packet_context.py +TESTS = test_ctf_writer_no_packet_context.py \ + test_ctf_writer_empty_packet.py -dist_noinst_SCRIPTS = test_ctf_writer_no_packet_context.py -EXTRA_DIST = test_ctf_writer_no_packet_context.py +check_SCRIPTS = $(TESTS) + +dist_noinst_SCRIPTS = $(TESTS) +EXTRA_DIST = $(TESTS) all-local: @if [ x"$(srcdir)" != x"$(builddir)" ]; then \ diff --git a/tests/lib/writer/test_ctf_writer_empty_packet.py b/tests/lib/writer/test_ctf_writer_empty_packet.py new file mode 100755 index 00000000..c5b6acb4 --- /dev/null +++ b/tests/lib/writer/test_ctf_writer_empty_packet.py @@ -0,0 +1,108 @@ +#!/usr/bin/env python3 +# +# The MIT License (MIT) +# +# Copyright (C) 2017 - Jérémie Galarneau +# +# Permission is hereby granted, free of charge, to any person obtaining a copy +# of this software and associated documentation files (the "Software"), to deal +# in the Software without restriction, including without limitation the rights +# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +# copies of the Software, and to permit persons to whom the Software is +# furnished to do so, subject to the following conditions: +# +# The above copyright notice and this permission notice shall be included in +# all copies or substantial portions of the Software. +# +# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +# SOFTWARE. + +import bt_python_helper +import tempfile +import babeltrace.writer as btw +import babeltrace.reader as btr +import shutil +import uuid + +TEST_COUNT = 4 +EXPECTED_EVENT_COUNT = 10 +trace_path = tempfile.mkdtemp() + + +def print_test_result(test_number, result, description): + result_string = 'ok' if result else 'not ok' + result_string += ' {} - {}'.format(test_number, description) + print(result_string) + +def create_trace(path): + trace = btw.Writer(path) + clock = btw.Clock('test_clock') + trace.add_clock(clock) + + integer_field_type = btw.IntegerFieldDeclaration(32) + + event_class = btw.EventClass('simple_event') + event_class.add_field(integer_field_type, 'int_field') + + stream_class = btw.StreamClass('empty_packet_stream') + stream_class.add_event_class(event_class) + stream_class.clock = clock + + stream = trace.create_stream(stream_class) + + for i in range(EXPECTED_EVENT_COUNT): + event = btw.Event(event_class) + event.payload('int_field').value = i + stream.append_event(event) + stream.flush() + print_test_result(1, True, + 'Flush a packet containing {} events'.format(EXPECTED_EVENT_COUNT)) + + # The CTF writer will not be able to populate the packet context's + # timestamp_begin and timestamp_end fields if it is asked to flush + # without any queued events. + try: + stream.flush() + empty_flush_succeeded = True + except ValueError: + empty_flush_succeeded = False + + print_test_result(2, not empty_flush_succeeded, + 'Flushing an empty packet without explicitly setting packet time bounds should fail') + packet_context = stream.packet_context + packet_context.field('timestamp_begin').value = 1 + packet_context.field('timestamp_end').value = 123456 + try: + stream.flush() + empty_flush_succeeded = True + except ValueError: + empty_flush_succeeded = False + + print_test_result(3, empty_flush_succeeded, + 'Flushing an empty packet after explicitly setting packet time bounds should succeed') + + +# TAP plan +print('1..{}'.format(TEST_COUNT)) +print('# Creating trace at {}'.format(trace_path)) +create_trace(trace_path) + +traces = btr.TraceCollection() +trace_handle = traces.add_trace(trace_path, 'ctf') +if trace_handle is None: + print('Failed to open trace at {}'.format(trace_path)) + +event_count = 0 +for event in traces.events: + event_count += 1 + +print_test_result(4, EXPECTED_EVENT_COUNT == event_count, + 'Trace was found to contain {} events, expected {}'.format( + event_count, EXPECTED_EVENT_COUNT)) + +shutil.rmtree(trace_path) -- 2.34.1