#!/usr/bin/env vpython3
#
# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import argparse
import json
import os
import re
import sys
def build_filter_file(args):
# TODO(b/227381644): This expression is hard to follow and should be
# simplified. This would require a change on the cros infra side as well
tast_expr_dict = {}
default_disabled_tests = []
if args.disabled_tests is not None:
default_disabled_tests = [
'!"name:{0}"'.format(test) for test in args.disabled_tests
]
default_enabled_test_term = ''
if args.enabled_tests is not None:
default_enabled_test_term = (' || ').join(
['"name:{0}"'.format(test) for test in args.enabled_tests])
# Generate the default expression to be used when there is no known key
tast_expr = args.tast_expr if args.tast_expr else ""
if default_disabled_tests:
default_disabled_term = " && ".join(default_disabled_tests)
tast_expr = "{0} && {1}".format(tast_expr, default_disabled_term) if \
tast_expr else default_disabled_term
if default_enabled_test_term:
tast_expr = "{0} && ({1})".format(
tast_expr,
default_enabled_test_term) if tast_expr else default_enabled_test_term
tast_expr_dict['default'] = "({0})".format(tast_expr)
# Generate an expression for each collection in the gni file
if args.tast_control is not None:
with open(args.tast_control, 'r') as tast_control_file:
gni = tast_control_file.read()
filter_lists = re.findall(r'(.*) = \[([^\]]*)\]', gni)
for filter_list in filter_lists:
tast_expr = args.tast_expr if args.tast_expr else ""
milestone_disabled_tests = {
'!"name:{0}"'.format(test)
for test in re.findall(r'"([^"]+)"', filter_list[1])
}
milestone_disabled_tests.update(default_disabled_tests)
if milestone_disabled_tests:
tast_expr = "{0} && {1}".format(
tast_expr, " && ".join(milestone_disabled_tests)
) if tast_expr else " && ".join(milestone_disabled_tests)
if default_enabled_test_term:
tast_expr = "{0} && ({1})".format(
tast_expr, default_enabled_test_term
) if tast_expr else default_enabled_test_term
if tast_expr:
tast_expr_dict[filter_list[0]] = "({0})".format(tast_expr)
if len(tast_expr_dict) > 0:
with open(args.output, "w") as file:
json.dump(tast_expr_dict, file, indent=2)
os.chmod(args.output, 0o644)
def main():
parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers(dest='command')
filter_gen_parser = subparsers.add_parser('generate-filter')
filter_gen_parser.add_argument(
'--tast-expr',
type=str,
required=False,
help='Tast expression to determine tests to run. This creates the '
'initial set of tests that can be further filtered.')
filter_gen_parser.add_argument(
'--enabled-tests',
type=str,
required=False,
action='append',
help='Name of tests to allow to test (unnamed tests will not run).')
filter_gen_parser.add_argument(
'--disabled-tests',
type=str,
required=False,
action='append',
help='Names of tests to disable from running')
filter_gen_parser.add_argument(
'--tast-control',
type=str,
required=False,
help='Filename for the tast_control file containing version skew '
'test filters to generate.')
filter_gen_parser.add_argument(
'--output',
required=True,
type=str,
help='Path to create the plain text filter file.')
filter_gen_parser.set_defaults(func=build_filter_file)
args = parser.parse_args()
if (args.command == "generate-filter" and args.disabled_tests is None and
args.enabled_tests is None and args.tast_expr is None):
parser.error(
'--disabled-tests, --enabled-tests, or --tast-expr must be provided '
'to generate-filter')
args.func(args)
return 0
if __name__ == '__main__':
sys.exit(main())
Codebase Browser
chromium