• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1#!/usr/bin/env python
2
3# Copyright (C) 2016 The Android Open Source Project
4#
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8#
9#      http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16
17import os
18import subprocess
19import sys
20
21INSTRUMENTED_PACKAGE_RUNNER = ('com.android.frameworks.servicestests/'
22                               'androidx.test.runner.AndroidJUnitRunner')
23
24PACKAGE_WHITELIST = (
25    "com.android.server",
26)
27
28COLOR_RED = '\033[0;31m'
29COLOR_NONE ='\033[0m'
30
31def run(shell_command, echo=True):
32    if echo:
33        print '%s + %s%s' % (
34                COLOR_RED,
35                echo if isinstance(echo, str) else shell_command,
36                COLOR_NONE)
37    return subprocess.check_call(shell_command, shell=True)
38
39# usage:
40#   ${ANDROID_BUILD_TOP}/frameworks/base/services/tests/runtests.py : run tests in com.android.server
41#   ${ANDROID_BUILD_TOP}/frameworks/base/services/tests/runtests.py -e package [package name, e.g. com.android.server]
42#   ${ANDROID_BUILD_TOP}/frameworks/base/services/tests/runtests.py -e class [class name, e.g. com.android.server.MountServiceTests]
43#
44#   The available INSTRUMENTED_PACKAGE_RUNNER may differ in different environments.
45#   In this case, use "adb shell pm list instrumentation" to query available runners
46#   and use the environment variable INSTRUMENTED_PACKAGE_RUNNER to overwrite
47#   the default one, e.g.,
48#   INSTRUMENTED_PACKAGE_RUNNER=com.android.frameworks.servicestests/androidx.test.runner.AndroidJUnitRunner \
49#       ${ANDROID_BUILD_TOP}/frameworks/base/services/tests/runtests.py
50#
51def main():
52    build_top = os.environ.get('ANDROID_BUILD_TOP', None)
53    out_dir = os.environ.get('OUT', None)
54    runner = os.environ.get('INSTRUMENTED_PACKAGE_RUNNER', None)
55    if build_top is None or out_dir is None:
56        print 'You need to source and lunch before you can use this script'
57        return 1
58    if runner is None:
59        runner = INSTRUMENTED_PACKAGE_RUNNER
60    print 'Building tests...'
61    run('make -j32 -C %s -f build/core/main.mk '
62        'MODULES-IN-frameworks-base-services-tests-servicestests' % build_top,
63        echo='mmma -j32 %s/frameworks/base/services/tests/servicestests' %
64             build_top)
65
66    print 'Installing tests...'
67    run('adb root')
68    run('adb wait-for-device')
69    apk_path = (
70            '%s/data/app/FrameworksServicesTests/FrameworksServicesTests.apk' %
71            out_dir)
72    run('adb install -t -r -g "%s"' % apk_path)
73
74    print 'Running tests...'
75    if len(sys.argv) != 1:
76        run('adb shell am instrument -w %s "%s"' %
77            (' '.join(sys.argv[1:]), runner))
78        return 0
79
80    # It would be nice if the activity manager accepted a list of packages, but
81    # in lieu of that...
82    for package in PACKAGE_WHITELIST:
83        run('adb shell am instrument -w -e package %s %s' %
84            (package, runner))
85
86    return 0
87
88
89if __name__ == '__main__':
90    sys.exit(main())
91