extract_test_cases.py

#!/usr/bin/env python3
#
# This script reads C++ or RST source files and writes all
# multi-line strings into individual files.
# This can be used to extract the Solidity test cases
# into files for e.g. fuzz testing as
# scripts/isolate_tests.py test/libsolidity/*

import sys
import re

def extract_test_cases(_path):
    with open(_path, mode='rb', encoding='utf8') as f:
        lines = f.read().splitlines()

    inside = False
    delimiter = ''
    test = ''

    ctr = 1
    test_name = ''

    for l in lines:
        if inside:
            if l.strip().endswith(')' + delimiter + '";'):
                with open(f'{ctr:03d}_{test_name}.sol', mode='wb', encoding='utf8') as f:
                    f.write(test)
                ctr += 1
                inside = False
                test = ''
            else:
                l = re.sub('^\t\t', '', l)
                l = l.replace('\t', '        ')
                test += l + '\n'
        else:
            m = re.search(r'BOOST_AUTO_TEST_CASE\(([^(]*)\)', l.strip())
            if m:
                test_name = m.group(1)
            m = re.search(r'R"([^(]*)\($', l.strip())
            if m:
                inside = True
                delimiter = m.group(1)

if __name__ == '__main__':
    extract_test_cases(sys.argv[1])