Add parametrized large array stress test

Signed-off-by: Mathieu Desnoyers <mathieu.desnoyers@efficios.com>

diff --git a/tests/1.8/stress/stream/pass/array-2GB/test.py b/tests/1.8/stress/stream/pass/array-2GB/test.py
deleted file mode 100755 (executable)
index c4431b4..0000000
--- a/tests/1.8/stress/stream/pass/array-2GB/test.py
+++ /dev/null
@@ -1,89 +0,0 @@
-#!/usr/bin/env python3
-
-# Expects either "prepare" or "clean" argument
-
-import sys
-import os
-
-output_metadata = './metadata'
-output_stream = './stream'
-
-def arg_err():
-       print('Please specify "prepare" or "clean" as first argument.')
-       sys.exit(-1)
-
-if (len(sys.argv) <= 1):
-       arg_err()
-mode = sys.argv[1]
-
-metadata_str = \
-'/* CTF 1.8 */\n' \
-'\n' \
-'typealias integer { size = 8; align = 8; signed = false; base = 10; } := uint8_t;\n' \
-'typealias integer { size = 32; align = 32; signed = false; base = hex; } := uint32_t;\n' \
-'\n' \
-'trace {\n' \
-'      major = 0;\n' \
-'      minor = 0;\n' \
-'      uuid = "2a6422d0-6cee-11e0-8c08-cb07d7b3a564";\n' \
-'      byte_order = le;\n' \
-'      packet.header := struct {\n' \
-'              uint32_t magic;\n' \
-'              uint8_t uuid[16];\n' \
-'      };\n' \
-'};\n' \
-'\n' \
-'event {\n' \
-'      name = seq;\n' \
-'      fields := struct {\n' \
-'              uint8_t field[2147483648]; /* 2GB array */\n' \
-'      };\n' \
-'};\n'
-
-stream_packet_header = [
-  0xC1, 0x1F, 0xFC, 0xC1,      # magic
-  0x2A, 0x64, 0x22, 0xD0, 0x6C, 0xEE, 0x11, 0xE0,
-  0x8C, 0x08, 0xCB, 0x07, 0xD7, 0xB3, 0xA5, 0x64, # uuid
-]
-
-array_byte = [ 0x00, ]
-
-def write_binary(f, arr):
-       f.write(bytes(arr))
-
-def generate_metadata():
-       # Generate metadata
-       metadata_f = open(output_metadata, 'w')
-       metadata_f.write(metadata_str)
-       metadata_f.close()
-
-def generate_stream():
-       # Generate stream
-       stream_f = open(output_stream, 'wb')
-       write_binary(stream_f, stream_packet_header)
-       # generate 2GB array as event content
-       os.ftruncate(stream_f.fileno(), 2147483648)
-       stream_f.close()
-
-def test_prepare():
-       print('Preparing test')
-       generate_metadata()
-       generate_stream()
-
-def test_clean():
-       print('Cleaning up test')
-       try:
-               os.remove(output_metadata)
-       except:
-               pass
-       try:
-               os.remove(output_stream)
-       except:
-               pass
-
-if (mode == "prepare"):
-       test_prepare()
-elif (mode == "clean"):
-       test_clean()
-else:
-       arg_err()

diff --git a/tests/1.8/stress/stream/pass/array-large/param-list.txt b/tests/1.8/stress/stream/pass/array-large/param-list.txt
new file mode 100644 (file)
index 0000000..97ad2d8
--- /dev/null
+++ b/tests/1.8/stress/stream/pass/array-large/param-list.txt
@@ -0,0 +1,8 @@
+512
+1024
+2048
+4096
+8192
+16384
+32768
+65536

diff --git a/tests/1.8/stress/stream/pass/array-large/test.py b/tests/1.8/stress/stream/pass/array-large/test.py
new file mode 100755 (executable)
index 0000000..cbbefae
--- /dev/null
+++ b/tests/1.8/stress/stream/pass/array-large/test.py
@@ -0,0 +1,97 @@
+#!/usr/bin/env python3
+
+# Expects either "prepare" or "clean" argument
+
+import sys
+import os
+
+def arg_err():
+       print('Please specify "prepare <SIZE>" or "clean <SIZE>" as first argument (SIZE is in KB).')
+       sys.exit(-1)
+
+if (len(sys.argv) <= 1):
+       arg_err()
+mode = sys.argv[1]
+if (len(sys.argv) <= 2):
+       arg_err()
+array_len_kb = int(sys.argv[2])
+array_len = array_len_kb * 1024
+
+tracedir_name = './trace-' + str(array_len_kb) + '/'
+output_metadata = tracedir_name + 'metadata'
+output_stream = tracedir_name + 'stream'
+
+
+def write_binary(f, arr):
+       f.write(bytes(arr))
+
+def generate_metadata(array_len):
+       # Generate metadata
+       metadata_str = \
+       ( '/* CTF 1.8 */\n'
+       '\n'
+       'typealias integer {{ size = 8; align = 8; signed = false; base = 10; }} := uint8_t;\n'
+       'typealias integer {{ size = 32; align = 32; signed = false; base = hex; }} := uint32_t;\n'
+       '\n'
+       'trace {{\n'
+       '       major = 0;\n'
+       '       minor = 0;\n'
+       '       uuid = "2a6422d0-6cee-11e0-8c08-cb07d7b3a564";\n'
+       '       byte_order = le;\n'
+       '       packet.header := struct {{\n'
+       '               uint32_t magic;\n'
+       '               uint8_t uuid[16];\n'
+       '       }};\n'
+       '}};\n'
+       '\n'
+       'event {{\n'
+       '       name = seq;\n'
+       '       fields := struct {{\n'
+       '               uint8_t field[{}]; \n'
+       '       }};\n'
+       '}};\n').format(array_len)
+       metadata_f = open(output_metadata, 'w')
+       metadata_f.write(metadata_str)
+       metadata_f.close()
+
+def generate_stream(array_len):
+       # Generate stream
+       stream_packet_header = [
+         0xC1, 0x1F, 0xFC, 0xC1,       # magic
+         0x2A, 0x64, 0x22, 0xD0, 0x6C, 0xEE, 0x11, 0xE0,
+         0x8C, 0x08, 0xCB, 0x07, 0xD7, 0xB3, 0xA5, 0x64, # uuid
+       ]
+       stream_f = open(output_stream, 'wb')
+       write_binary(stream_f, stream_packet_header)
+       # generate array as event content
+       os.ftruncate(stream_f.fileno(), len(stream_packet_header) + array_len)
+       stream_f.close()
+
+def test_prepare():
+       print('Preparing test for array length ' + str(array_len_kb) + ' kb')
+       os.mkdir(tracedir_name)
+       generate_metadata(array_len)
+       generate_stream(array_len)
+
+def test_clean():
+       print('Cleaning up test for array length ' + str(array_len_kb) + ' kb')
+       try:
+               os.remove(output_metadata)
+       except:
+               pass
+       try:
+               os.remove(output_stream)
+       except:
+               pass
+       try:
+               os.rmdir(tracedir_name)
+       except:
+               pass
+
+if (mode == "prepare"):
+       test_clean()
+       test_prepare()
+elif (mode == "clean"):
+       test_clean()
+else:
+       arg_err()

diff --git a/tests/1.8/stress/stream/pass/test.sh b/tests/1.8/stress/stream/pass/test.sh
index 0b22645f36c3897cd221d419373044b9f4a71eab..59094770f36b8b1f24a5a07bd8bb18d25255ac24 100755 (executable)
--- a/tests/1.8/stress/stream/pass/test.sh
+++ b/tests/1.8/stress/stream/pass/test.sh
@@ -20,7 +20,11 @@ UTILS_DIR=$CURDIR/../../../../../utils/
 
 source $UTILS_DIR/tap/tap.sh
 
-NUM_TESTS=3
+NUM_TESTS=0
+
+for param in $(cat $1/param-list.txt); do
+       NUM_TESTS=$((${NUM_TESTS} + 3))
+done
 
 plan_tests $NUM_TESTS
 
@@ -29,16 +33,18 @@ if [ "x${CTF_READER_BIN}" == "x" ]; then
        exit 1
 fi
 
-cd $1 && ./test.py prepare
-result=$?
-is $result 0 $1
-cd ..
-
-$CTF_READER_BIN $CTF_READER_OPTS $1 >/dev/null 2>&1
-result=$?
-echo $result > /tmp/blah
-is $result 0 $1                # expect pass
-
-cd $1 && ./test.py clean
-result=$?
-is $result 0 $1
+for param in $(cat $1/param-list.txt); do
+       cd $1 && ./test.py prepare ${param}
+       result=$?
+       is $result 0 $1
+       cd ..
+
+       $CTF_READER_BIN $CTF_READER_OPTS $1/trace-${param} >/dev/null 2>&1
+       result=$?
+       is $result 0 $1         # expect pass
+
+       cd $1 && ./test.py clean ${param}
+       result=$?
+       is $result 0 $1
+       cd ..
+done