diff options
Diffstat (limited to 'build/android/test_package_executable.py')
| -rw-r--r-- | build/android/test_package_executable.py | 153 |
1 files changed, 153 insertions, 0 deletions
diff --git a/build/android/test_package_executable.py b/build/android/test_package_executable.py new file mode 100644 index 0000000..6390735 --- /dev/null +++ b/build/android/test_package_executable.py @@ -0,0 +1,153 @@ +#!/usr/bin/python +# Copyright (c) 2011 The Chromium Authors. All rights reserved. +# Use of this source code is governed by a BSD-style license that can be +# found in the LICENSE file. + + +import logging +import os +import pexpect +import shutil +import sys +import tempfile + +import cmd_helper +from test_package import TestPackage + + +class TestPackageExecutable(TestPackage): + """A helper class for running stand-alone executables.""" + + _TEST_RUNNER_RET_VAL_FILE = '/data/local/tmp/gtest_retval' + + def __init__(self, adb, device, test_suite, timeout, rebaseline, + performance_test, cleanup_test_files, tool, dump_debug_info, + symbols_dir=None): + """ + Args: + adb: ADB interface the tests are using. + device: Device to run the tests. + test_suite: A specific test suite to run, empty to run all. + timeout: Timeout for each test. + rebaseline: Whether or not to run tests in isolation and update the + filter. + performance_test: Whether or not performance test(s). + cleanup_test_files: Whether or not to cleanup test files on device. + tool: Name of the Valgrind tool. + dump_debug_info: A debug_info object. + symbols_dir: Directory to put the stripped binaries. + """ + TestPackage.__init__(self, adb, device, test_suite, timeout, + rebaseline, performance_test, cleanup_test_files, + tool, dump_debug_info) + self.symbols_dir = symbols_dir + + def _GetGTestReturnCode(self): + ret = None + ret_code_file = tempfile.NamedTemporaryFile() + try: + if not self.adb.Adb().Pull( + TestPackageExecutable._TEST_RUNNER_RET_VAL_FILE, ret_code_file.name): + logging.critical('Unable to pull gtest ret val file %s', + ret_code_file.name) + raise ValueError + ret_code = file(ret_code_file.name).read() + ret = int(ret_code) + except ValueError: + logging.critical('Error reading gtest ret val file %s [%s]', + ret_code_file.name, ret_code) + ret = 1 + return ret + + def _AddNativeCoverageExports(self): + # export GCOV_PREFIX set the path for native coverage results + # export GCOV_PREFIX_STRIP indicates how many initial directory + # names to strip off the hardwired absolute paths. + # This value is calculated in buildbot.sh and + # depends on where the tree is built. + # Ex: /usr/local/google/code/chrome will become + # /code/chrome if GCOV_PREFIX_STRIP=3 + try: + depth = os.environ['NATIVE_COVERAGE_DEPTH_STRIP'] + except KeyError: + logging.info('NATIVE_COVERAGE_DEPTH_STRIP is not defined: ' + 'No native coverage.') + return '' + export_string = 'export GCOV_PREFIX="/data/local/gcov"\n' + export_string += 'export GCOV_PREFIX_STRIP=%s\n' % depth + return export_string + + def GetAllTests(self): + """Returns a list of all tests available in the test suite.""" + all_tests = self.adb.RunShellCommand( + '/data/local/%s --gtest_list_tests' % self.test_suite_basename) + return self._ParseGTestListTests(all_tests) + + def CreateTestRunnerScript(self, gtest_filter, test_arguments): + """Creates a test runner script and pushes to the device. + + Args: + gtest_filter: A gtest_filter flag. + test_arguments: Additional arguments to pass to the test binary. + """ + tool_wrapper = self.tool.GetTestWrapper() + sh_script_file = tempfile.NamedTemporaryFile() + # We need to capture the exit status from the script since adb shell won't + # propagate to us. + sh_script_file.write('cd /data/local\n' + '%s' + '%s /data/local/%s --gtest_filter=%s %s\n' + 'echo $? > %s' % + (self._AddNativeCoverageExports(), + tool_wrapper, self.test_suite_basename, + gtest_filter, test_arguments, + TestPackageExecutable._TEST_RUNNER_RET_VAL_FILE)) + sh_script_file.flush() + cmd_helper.RunCmd(['chmod', '+x', sh_script_file.name]) + self.adb.PushIfNeeded(sh_script_file.name, + '/data/local/chrome_test_runner.sh') + + def RunTestsAndListResults(self): + """Runs all the tests and checks for failures. + + Returns: + A TestResults object. + """ + args = ['adb', '-s', self.device, 'shell', 'sh', + '/data/local/chrome_test_runner.sh'] + logging.info(args) + p = pexpect.spawn(args[0], args[1:], logfile=sys.stdout) + return self._WatchTestOutput(p) + + def StripAndCopyExecutable(self): + """Strips and copies the executable to the device.""" + if self.tool.NeedsDebugInfo(): + target_name = self.test_suite + elif self.test_suite_basename == 'webkit_unit_tests': + # webkit_unit_tests has been stripped in build step. + target_name = self.test_suite + else: + target_name = self.test_suite + '_' + self.device + '_stripped' + should_strip = True + if os.path.isfile(target_name): + logging.info('Found target file %s' % target_name) + target_mtime = os.stat(target_name).st_mtime + source_mtime = os.stat(self.test_suite).st_mtime + if target_mtime > source_mtime: + logging.info('Target mtime (%d) is newer than source (%d), assuming ' + 'no change.' % (target_mtime, source_mtime)) + should_strip = False + + if should_strip: + logging.info('Did not find up-to-date stripped binary. Generating a ' + 'new one (%s).' % target_name) + # Whenever we generate a stripped binary, copy to the symbols dir. If we + # aren't stripping a new binary, assume it's there. + if self.symbols_dir: + if not os.path.exists(self.symbols_dir): + os.makedirs(self.symbols_dir) + shutil.copy(self.test_suite, self.symbols_dir) + strip = os.environ['STRIP'] + cmd_helper.RunCmd([strip, self.test_suite, '-o', target_name]) + test_binary = '/data/local/' + self.test_suite_basename + self.adb.PushIfNeeded(target_name, test_binary) |