Christopher Wiley | 012ab10 | 2016-10-20 12:33:31 -0700 | [diff] [blame] | 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 | |
| 17 | import os |
| 18 | import subprocess |
| 19 | import sys |
| 20 | |
| 21 | INSTRUMENTED_PACKAGE_RUNNER = ('com.android.frameworks.servicestests/' |
| 22 | 'android.support.test.runner.AndroidJUnitRunner') |
| 23 | |
| 24 | PACKAGE_WHITELIST = ( |
Howard Chen | 3fd94f4 | 2018-09-28 15:15:22 +0800 | [diff] [blame] | 25 | "com.android.server", |
Christopher Wiley | 012ab10 | 2016-10-20 12:33:31 -0700 | [diff] [blame] | 26 | ) |
| 27 | |
| 28 | COLOR_RED = '\033[0;31m' |
| 29 | COLOR_NONE ='\033[0m' |
| 30 | |
| 31 | def 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 | |
Howard Chen | 3fd94f4 | 2018-09-28 15:15:22 +0800 | [diff] [blame] | 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 | # |
Christopher Wiley | 012ab10 | 2016-10-20 12:33:31 -0700 | [diff] [blame] | 51 | def main(): |
| 52 | build_top = os.environ.get('ANDROID_BUILD_TOP', None) |
| 53 | out_dir = os.environ.get('OUT', None) |
Howard Chen | 3fd94f4 | 2018-09-28 15:15:22 +0800 | [diff] [blame] | 54 | runner = os.environ.get('INSTRUMENTED_PACKAGE_RUNNER', None) |
Christopher Wiley | 012ab10 | 2016-10-20 12:33:31 -0700 | [diff] [blame] | 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 |
Howard Chen | 3fd94f4 | 2018-09-28 15:15:22 +0800 | [diff] [blame] | 58 | if runner is None: |
| 59 | runner = INSTRUMENTED_PACKAGE_RUNNER |
Christopher Wiley | 012ab10 | 2016-10-20 12:33:31 -0700 | [diff] [blame] | 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) |
Howard Chen | 3fd94f4 | 2018-09-28 15:15:22 +0800 | [diff] [blame] | 72 | run('adb install -t -r -g "%s"' % apk_path) |
Christopher Wiley | 012ab10 | 2016-10-20 12:33:31 -0700 | [diff] [blame] | 73 | |
| 74 | print 'Running tests...' |
| 75 | if len(sys.argv) != 1: |
Etan Cohen | 31dc3db | 2016-10-27 10:33:56 -0700 | [diff] [blame] | 76 | run('adb shell am instrument -w %s "%s"' % |
Howard Chen | 3fd94f4 | 2018-09-28 15:15:22 +0800 | [diff] [blame] | 77 | (' '.join(sys.argv[1:]), runner)) |
Christopher Wiley | 012ab10 | 2016-10-20 12:33:31 -0700 | [diff] [blame] | 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' % |
Howard Chen | 3fd94f4 | 2018-09-28 15:15:22 +0800 | [diff] [blame] | 84 | (package, runner)) |
Christopher Wiley | 012ab10 | 2016-10-20 12:33:31 -0700 | [diff] [blame] | 85 | |
| 86 | return 0 |
| 87 | |
| 88 | |
| 89 | if __name__ == '__main__': |
| 90 | sys.exit(main()) |