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 = ( |
| 25 | 'android.net', |
| 26 | 'com.android.server.connectivity', |
| 27 | ) |
| 28 | |
| 29 | COLOR_RED = '\033[0;31m' |
| 30 | COLOR_NONE ='\033[0m' |
| 31 | |
| 32 | def run(shell_command, echo=True): |
| 33 | if echo: |
| 34 | print '%s + %s%s' % ( |
| 35 | COLOR_RED, |
| 36 | echo if isinstance(echo, str) else shell_command, |
| 37 | COLOR_NONE) |
| 38 | return subprocess.check_call(shell_command, shell=True) |
| 39 | |
| 40 | |
| 41 | def main(): |
| 42 | build_top = os.environ.get('ANDROID_BUILD_TOP', None) |
| 43 | out_dir = os.environ.get('OUT', None) |
| 44 | if build_top is None or out_dir is None: |
| 45 | print 'You need to source and lunch before you can use this script' |
| 46 | return 1 |
| 47 | |
| 48 | print 'Building tests...' |
| 49 | run('make -j32 -C %s -f build/core/main.mk ' |
| 50 | 'MODULES-IN-frameworks-base-services-tests-servicestests' % build_top, |
| 51 | echo='mmma -j32 %s/frameworks/base/services/tests/servicestests' % |
| 52 | build_top) |
| 53 | |
| 54 | print 'Installing tests...' |
| 55 | run('adb root') |
| 56 | run('adb wait-for-device') |
| 57 | apk_path = ( |
| 58 | '%s/data/app/FrameworksServicesTests/FrameworksServicesTests.apk' % |
| 59 | out_dir) |
| 60 | run('adb install -r -g "%s"' % apk_path) |
| 61 | |
| 62 | print 'Running tests...' |
| 63 | if len(sys.argv) != 1: |
Etan Cohen | 31dc3db | 2016-10-27 10:33:56 -0700 | [diff] [blame] | 64 | run('adb shell am instrument -w %s "%s"' % |
| 65 | (' '.join(sys.argv[1:]), INSTRUMENTED_PACKAGE_RUNNER)) |
Christopher Wiley | 012ab10 | 2016-10-20 12:33:31 -0700 | [diff] [blame] | 66 | return 0 |
| 67 | |
| 68 | # It would be nice if the activity manager accepted a list of packages, but |
| 69 | # in lieu of that... |
| 70 | for package in PACKAGE_WHITELIST: |
| 71 | run('adb shell am instrument -w -e package %s %s' % |
| 72 | (package, INSTRUMENTED_PACKAGE_RUNNER)) |
| 73 | |
| 74 | return 0 |
| 75 | |
| 76 | |
| 77 | if __name__ == '__main__': |
| 78 | sys.exit(main()) |