1# Copyright 2015 The Chromium OS Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5AUTHOR = "shuqianz, chromeos-infra"
6NAME = "suite_attr_wrapper"
7
8TIME = "SHORT"
9TEST_CATEGORY = "General"
10TEST_CLASS = "suite"
11TEST_TYPE = "Server"
12
13DOC = """
14This is a wrapper suite to trigger tests that satisfy any attribute boolean
15expression.
16
17@param build: The name of the image to test.
18              Ex: x86-mario-release/R17-1412.33.0-a1-b29
19@param board: The board to test on. Ex: x86-mario
20@param pool: The pool of machines to utilize for scheduling. If pool=None
21             board is used.
22@param check_hosts: require appropriate live hosts to exist in the lab.
23@param SKIP_IMAGE: (optional) If present and True, don't re-image devices.
24"""
25import ast, logging
26
27import common
28from autotest_lib.server.cros.dynamic_suite import dynamic_suite
29from autotest_lib.server.cros.dynamic_suite import suite
30from autotest_lib.server.cros import provision
31
32suite_args_dict = ast.literal_eval(args_dict['suite_args'])
33
34# Get the predicate object from attr_filter argument
35try:
36    attr = suite_args_dict['attr_filter']
37    attr_predicate = suite.Suite.matches_attribute_expression_predicate(attr)
38except KeyError:
39    logging.error('Unable to find the attribute boolean expression used to '
40                  'trigger tests in suite_args %s, please use attr_filter '
41                  'keyword to specify it. suite aborting' % suite_args_dict)
42    raise
43
44# Update suite_args_dict into args_dict
45args_dict.update(suite_args_dict)
46args_dict['predicate'] = attr_predicate
47args_dict.setdefault('name', NAME)
48args_dict['job'] = job
49args_dict.setdefault('version_prefix', provision.CROS_VERSION_PREFIX)
50
51dynamic_suite.reimage_and_run(**args_dict)
52