Add ability to parse multiple files. Additionally add the
ability to parse all results in the KUnit debugfs repository.

How to parse multiple files:

./tools/testing/kunit/kunit.py parse results.log results2.log

How to parse all files in directory:

./tools/testing/kunit/kunit.py parse directory_path/*

How to parse KUnit debugfs repository:

./tools/testing/kunit/kunit.py parse debugfs

For each file, the parser outputs the file name, results, and test
summary. At the end of all parsing, the parser outputs a total summary
line.

This feature can be easily tested on the tools/testing/kunit/test_data/
directory.

Signed-off-by: Rae Moar <rm...@google.com>
---
Changes since v3:
- Changing from input() to stdin
- Add checking for non-regular files
- Spacing fix
- Small printing fix

 tools/testing/kunit/kunit.py | 54 +++++++++++++++++++++++++-----------
 1 file changed, 38 insertions(+), 16 deletions(-)

diff --git a/tools/testing/kunit/kunit.py b/tools/testing/kunit/kunit.py
index bc74088c458a..641b8ca83e3e 100755
--- a/tools/testing/kunit/kunit.py
+++ b/tools/testing/kunit/kunit.py
@@ -511,19 +511,40 @@ def exec_handler(cli_args: argparse.Namespace) -> None:
 
 
 def parse_handler(cli_args: argparse.Namespace) -> None:
-       if cli_args.file is None:
-               sys.stdin.reconfigure(errors='backslashreplace')  # type: ignore
-               kunit_output = sys.stdin  # type: Iterable[str]
-       else:
-               with open(cli_args.file, 'r', errors='backslashreplace') as f:
-                       kunit_output = f.read().splitlines()
-       # We know nothing about how the result was created!
-       metadata = kunit_json.Metadata()
-       request = KunitParseRequest(raw_output=cli_args.raw_output,
-                                       json=cli_args.json)
-       result, _ = parse_tests(request, metadata, kunit_output)
-       if result.status != KunitStatus.SUCCESS:
-               sys.exit(1)
+       parsed_files = cli_args.files # type: List[str]
+       total_test = kunit_parser.Test()
+       total_test.status = kunit_parser.TestStatus.SUCCESS
+       if not parsed_files:
+               parsed_files.append('/dev/stdin')
+       elif len(parsed_files) == 1 and parsed_files[0] == "debugfs":
+               parsed_files.pop()
+               for (root, _, files) in os.walk("/sys/kernel/debug/kunit"):
+                       parsed_files.extend(os.path.join(root, f) for f in 
files if f == "results")
+               if not parsed_files:
+                       print("No files found.")
+
+       for file in parsed_files:
+               if os.path.isdir(file):
+                       print(f'Ignoring directory "{file}"')
+               elif os.path.exists(file):
+                       print(file)
+                       with open(file, 'r', errors='backslashreplace') as f:
+                               kunit_output = f.read().splitlines()
+                       # We know nothing about how the result was created!
+                       metadata = kunit_json.Metadata()
+                       request = 
KunitParseRequest(raw_output=cli_args.raw_output,
+                                                       json=cli_args.json)
+                       _, test = parse_tests(request, metadata, kunit_output)
+                       total_test.subtests.append(test)
+               else:
+                       print(f'Could not find "{file}"')
+
+       if len(parsed_files) > 1: # if more than one file was parsed output 
total summary
+               print('All files parsed.')
+               if not request.raw_output:
+                       stdout.print_with_timestamp(kunit_parser.DIVIDER)
+                       kunit_parser.bubble_up_test_results(total_test)
+                       kunit_parser.print_summary_line(total_test)
 
 
 subcommand_handlers_map = {
@@ -569,9 +590,10 @@ def main(argv: Sequence[str]) -> None:
                                            help='Parses KUnit results from a 
file, '
                                            'and parses formatted results.')
        add_parse_opts(parse_parser)
-       parse_parser.add_argument('file',
-                                 help='Specifies the file to read results 
from.',
-                                 type=str, nargs='?', metavar='input_file')
+       parse_parser.add_argument('files',
+                                 help='List of file paths to read results from 
or keyword'
+                                 '"debugfs" to read all results from the 
debugfs directory.',
+                                 type=str, nargs='*', metavar='input_files')
 
        cli_args = parser.parse_args(massage_argv(argv))
 

base-commit: 806cb2270237ce2ec672a407d66cee17a07d3aa2
-- 
2.44.0.291.gc1ea87d7ee-goog


Reply via email to