Refactor test infrastructure for metadata stress

[ctf-testsuite.git] / tests / 1.8 / stress / metadata / pass / many-stream-class / test.py

diff --git a/tests/1.8/stress/metadata/pass/many-stream-class/test.py b/tests/1.8/stress/metadata/pass/many-stream-class/test.py
index 06334404e3087da9da5f572176a6df47bc6d4352..ea7556f43e8c0beb942409f645265cd7495b339e 100755 (executable)
--- a/tests/1.8/stress/metadata/pass/many-stream-class/test.py
+++ b/tests/1.8/stress/metadata/pass/many-stream-class/test.py
@@ -1,83 +1,46 @@
 #!/usr/bin/env python3
+from ctftestsuite.stress import MetadataTest
 
-# Expects either "prepare" or "clean" argument
 
-import sys
-import os
-import struct
+class Test(MetadataTest):
+    what = '{size} stream classes'
 
-def arg_err():
-       print('Please specify "prepare <SIZE>" or "clean <SIZE>" as first argument.')
-       sys.exit(-1)
+    def write_metadata(self, f):
+        p1 = \
+'''/* CTF 1.8 */
 
-if (len(sys.argv) <= 1):
-       arg_err()
-mode = sys.argv[1]
-if (len(sys.argv) <= 2):
-       arg_err()
-array_len = int(sys.argv[2])
+typealias integer { size = 8; align = 8; signed = false; base = 10; } := uint8_t;
+typealias integer { size = 32; align = 8; signed = false; base = hex; } := uint32_t;
+typealias integer { size = 64; align = 8; signed = false; base = hex; } := uint64_t;
 
-tracedir_name = './trace-' + str(array_len) + '/'
-output_metadata = tracedir_name + 'metadata'
-output_stream = tracedir_name + 'stream'
+trace {
+       major = 0;
+       minor = 0;
+       uuid = "2a6422d0-6cee-11e0-8c08-cb07d7b3a564";
+       byte_order = le;
+       packet.header := struct {
+               uint32_t magic;
+               uint8_t uuid[16];
+               uint64_t stream_id;
+       };
+};
 
+'''
 
-def write_binary(f, arr):
-       f.write(bytes(arr))
+        stream_fmt = \
+'''stream {{
+       id = {id};
+}};
 
-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 = 8; signed = false; base = hex; }} := uint32_t;\n'
-       'typealias integer {{ size = 64; align = 8; signed = false; base = hex; }} := uint64_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'
-       '               uint64_t stream_id;\n'
-       '       }};\n'
-       '}};\n'
-       '\n').format()
+'''
 
-       metadata_stream_class = \
-       ('stream {{\n'
-       '       id = {};\n'
-       '}};\n')
+        f.write(p1)
 
-       metadata_f = open(output_metadata, 'w')
-       metadata_f.write(metadata_str)
-       for i in range(array_len):
-               metadata_f.write(metadata_stream_class.format(i))
-       metadata_f.close()
+        for i in range(self.size):
+            stream = stream_fmt.format(id=i)
+            f.write(stream)
 
-def test_prepare():
-       print('Preparing test for ' + str(array_len) + ' stream class')
-       os.mkdir(tracedir_name)
-       generate_metadata(array_len)
 
-def test_clean():
-       print('Cleaning up test for ' + str(array_len) + ' stream class')
-       try:
-               os.remove(output_metadata)
-       except:
-               pass
-       try:
-               os.rmdir(tracedir_name)
-       except:
-               pass
-
-if (mode == "prepare"):
-       test_clean()
-       test_prepare()
-elif (mode == "clean"):
-       test_clean()
-else:
-       arg_err()
+if __name__ == '__main__':
+    test = Test()
+    test.main()