1#!/usr/bin/env python3
2# Copyright 2019, The Android Open Source Project
3#
4# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
7#
8#     http://www.apache.org/licenses/LICENSE-2.0
9#
10# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
15
16"""asuite_run_unittests
17
18This is a unit test wrapper to run tests of aidegen, atest or both.
19"""
20
21
22from __future__ import print_function
23
24import argparse
25import os
26import shlex
27import subprocess
28import sys
29
30
31EXIT_ALL_CLEAN = 0
32EXIT_TEST_FAIL = 1
33ASUITE_PLUGIN_PATH = "tools/asuite/asuite_plugin"
34# TODO: remove echo when atest migration has done.
35ATEST_CMD = "echo {}/tools/asuite/atest/atest_run_unittests.py".format(
36    os.getenv('ANDROID_BUILD_TOP'))
37AIDEGEN_CMD = "atest aidegen_unittests --host"
38PLUGIN_LIB_CMD = "atest plugin_lib_unittests --host"
39GRADLE_TEST = "/gradlew test"
40
41
42def run_unittests(files):
43    """Parse modified files and tell if they belong to aidegen, atest or both.
44
45    Args:
46        files: a list of files.
47
48    Returns:
49        True if subprocess.check_call() returns 0.
50    """
51    cmd_dict = {}
52    for f in files:
53        if 'atest' in f:
54            cmd_dict.update({ATEST_CMD: None})
55        if 'aidegen' in f:
56            cmd_dict.update({AIDEGEN_CMD: None})
57        if 'plugin_lib' in f:
58            cmd_dict.update({PLUGIN_LIB_CMD: None})
59        if 'asuite_plugin' in f:
60            full_path = os.path.join(
61                os.getenv('ANDROID_BUILD_TOP'), ASUITE_PLUGIN_PATH)
62            cmd = full_path + GRADLE_TEST
63            cmd_dict.update({cmd : full_path})
64    try:
65        for cmd, path in cmd_dict.items():
66            subprocess.check_call(shlex.split(cmd), cwd=path)
67    except subprocess.CalledProcessError as error:
68        print('Unit test failed at:\n\n{}'.format(error.output))
69        raise
70    return True
71
72
73def get_files_to_upload():
74    """Parse args or modified files and return them as a list.
75
76    Returns:
77        A list of files to upload.
78    """
79    parser = argparse.ArgumentParser()
80    parser.add_argument('preupload_files', nargs='*', help='Files to upload.')
81    args = parser.parse_args()
82    files_to_upload = args.preupload_files
83    if not files_to_upload:
84        # When running by users directly, only consider:
85        # added(A), renamed(R) and modified(M) files
86        # and store them in files_to_upload.
87        cmd = "git status --short | egrep ^[ARM] | awk '{print $NF}'"
88        preupload_files = subprocess.check_output(cmd, shell=True,
89                                                  encoding='utf-8').splitlines()
90        if preupload_files:
91            print('Files to upload: %s' % preupload_files)
92            files_to_upload = preupload_files
93        else:
94            sys.exit(EXIT_ALL_CLEAN)
95    return files_to_upload
96
97if __name__ == '__main__':
98    if not run_unittests(get_files_to_upload()):
99        sys.exit(EXIT_TEST_FAIL)
100