1#! /usr/bin/env vpython3 2# 3# Copyright 2023 The ANGLE Project Authors. All rights reserved. 4# Use of this source code is governed by a BSD-style license that can be 5# found in the LICENSE file. 6# 7# angle_android_test_runner.py: 8# Runs (a subset of) ANGLE tests on Android using android_helper 9# instead of chromium test_runner scripts. This script is integrated into 10# gn builds via the android_test_runner_script var, which creates wrappers 11# (out/<config>/angle_trace_tests) calling into this script 12# with additional args like --output-directory=out/<config> etc 13 14import argparse 15import json 16import logging 17import os 18import pathlib 19import sys 20 21PY_UTILS = str(pathlib.Path(__file__).resolve().parent / 'py_utils') 22if PY_UTILS not in sys.path: 23 os.stat(PY_UTILS) and sys.path.insert(0, PY_UTILS) 24import android_helper 25import angle_test_util 26 27 28def AddCommonParserArgs(parser): 29 parser.add_argument( 30 '--suite', 31 help='Test suite to run.', 32 required=True, 33 choices=['angle_end2end_tests', 'angle_perftests', 'angle_trace_tests']) 34 parser.add_argument('-l', '--log', help='Logging level.', default='info') 35 parser.add_argument('--list-tests', help='List tests.', action='store_true') 36 parser.add_argument( 37 '-f', 38 '--filter', 39 '--isolated-script-test-filter', 40 '--gtest_filter', 41 type=str, 42 help='Test filter.') 43 44 45def RunAndroidTestSuite(args, extra_args): 46 angle_test_util.SetupLogging(args.log.upper()) 47 48 android_helper.Initialize(args.suite) 49 assert android_helper.IsAndroid() 50 51 rc, output, _ = android_helper.RunTests( 52 args.suite, ['--list-tests', '--verbose'] + extra_args, log_output=False) 53 if rc != 0: 54 logging.fatal('Could not find test list from test output:\n%s' % output) 55 return rc 56 57 tests = angle_test_util.GetTestsFromOutput(output) 58 if args.filter: 59 tests = angle_test_util.FilterTests(tests, args.filter) 60 61 if args.list_tests: 62 print('\n'.join(['Tests list:'] + tests + ['End tests list.'])) 63 return 0 64 65 if args.suite == 'angle_trace_tests': 66 traces = set(android_helper.GetTraceFromTestName(test) for test in tests) 67 android_helper.PrepareRestrictedTraces(traces) 68 69 if args.prepare_only: 70 print('Prepared traces: %s' % traces) 71 return 0 72 73 flags = ['--gtest_filter=' + args.filter] if args.filter else [] 74 return android_helper.RunTests(args.suite, flags + extra_args)[0] 75 76 77def main(): 78 parser = argparse.ArgumentParser() 79 parser.add_argument('test_type', choices=['gtest']) 80 parser.add_argument('--output-directory', required=True) 81 parser.add_argument('--wrapper-script-args') 82 parser.add_argument('--runtime-deps-path') 83 parser.add_argument('--prepare-only', action='store_true') 84 AddCommonParserArgs(parser) 85 86 args, extra_args = parser.parse_known_args() 87 88 os.chdir(args.output_directory) 89 90 return RunAndroidTestSuite(args, extra_args) 91 92 93if __name__ == "__main__": 94 sys.exit(main()) 95