From 757ed4fd2eca070da4cfe09a101f2ab36ffa2f37 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Zbigniew=20J=C4=99drzejewski-Szmek?= Date: Wed, 16 May 2018 12:05:43 +0200 Subject: [PATCH] log-generator: make message size configurable, add short options Also remove "b''" from the generated MESSAGE= field. --- src/journal-remote/log-generator.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/journal-remote/log-generator.py b/src/journal-remote/log-generator.py index c2f945bb473..e1725b1a71d 100755 --- a/src/journal-remote/log-generator.py +++ b/src/journal-remote/log-generator.py @@ -5,7 +5,8 @@ import argparse PARSER = argparse.ArgumentParser() PARSER.add_argument('n', type=int) PARSER.add_argument('--dots', action='store_true') -PARSER.add_argument('--data-size', type=int, default=4000) +PARSER.add_argument('-m', '--message-size', type=int, default=200) +PARSER.add_argument('-d', '--data-size', type=int, default=4000) PARSER.add_argument('--data-type', choices={'random', 'simple'}) OPTIONS = PARSER.parse_args() @@ -42,7 +43,9 @@ bytes = 0 counter = 0 for i in range(OPTIONS.n): - message = repr(src.read(2000)) + message = src.read(OPTIONS.message_size) + message = repr(message)[2:-1] + if OPTIONS.data_type == 'random': data = repr(src.read(OPTIONS.data_size)) else: -- 2.47.3