| OLD | NEW |
| (Empty) |
| 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 # Use of this source code is governed by a BSD-style license that can be | |
| 3 # found in the LICENSE file. | |
| 4 | |
| 5 | |
| 6 import logging | |
| 7 import os | |
| 8 import pexpect | |
| 9 import shutil | |
| 10 import sys | |
| 11 import tempfile | |
| 12 | |
| 13 import cmd_helper | |
| 14 from test_package import TestPackage | |
| 15 | |
| 16 | |
| 17 class TestPackageExecutable(TestPackage): | |
| 18 """A helper class for running stand-alone executables.""" | |
| 19 | |
| 20 _TEST_RUNNER_RET_VAL_FILE = '/data/local/tmp/gtest_retval' | |
| 21 | |
| 22 def __init__(self, adb, device, test_suite, timeout, rebaseline, | |
| 23 performance_test, cleanup_test_files, tool, dump_debug_info, | |
| 24 symbols_dir=None): | |
| 25 """ | |
| 26 Args: | |
| 27 adb: ADB interface the tests are using. | |
| 28 device: Device to run the tests. | |
| 29 test_suite: A specific test suite to run, empty to run all. | |
| 30 timeout: Timeout for each test. | |
| 31 rebaseline: Whether or not to run tests in isolation and update the | |
| 32 filter. | |
| 33 performance_test: Whether or not performance test(s). | |
| 34 cleanup_test_files: Whether or not to cleanup test files on device. | |
| 35 tool: Name of the Valgrind tool. | |
| 36 dump_debug_info: A debug_info object. | |
| 37 symbols_dir: Directory to put the stripped binaries. | |
| 38 """ | |
| 39 TestPackage.__init__(self, adb, device, test_suite, timeout, | |
| 40 rebaseline, performance_test, cleanup_test_files, | |
| 41 tool, dump_debug_info) | |
| 42 self.symbols_dir = symbols_dir | |
| 43 | |
| 44 def _GetGTestReturnCode(self): | |
| 45 ret = None | |
| 46 ret_code = 1 # Assume failure if we can't find it | |
| 47 ret_code_file = tempfile.NamedTemporaryFile() | |
| 48 try: | |
| 49 if not self.adb.Adb().Pull( | |
| 50 TestPackageExecutable._TEST_RUNNER_RET_VAL_FILE, ret_code_file.name): | |
| 51 logging.critical('Unable to pull gtest ret val file %s', | |
| 52 ret_code_file.name) | |
| 53 raise ValueError | |
| 54 ret_code = file(ret_code_file.name).read() | |
| 55 ret = int(ret_code) | |
| 56 except ValueError: | |
| 57 logging.critical('Error reading gtest ret val file %s [%s]', | |
| 58 ret_code_file.name, ret_code) | |
| 59 ret = 1 | |
| 60 return ret | |
| 61 | |
| 62 def _AddNativeCoverageExports(self): | |
| 63 # export GCOV_PREFIX set the path for native coverage results | |
| 64 # export GCOV_PREFIX_STRIP indicates how many initial directory | |
| 65 # names to strip off the hardwired absolute paths. | |
| 66 # This value is calculated in buildbot.sh and | |
| 67 # depends on where the tree is built. | |
| 68 # Ex: /usr/local/google/code/chrome will become | |
| 69 # /code/chrome if GCOV_PREFIX_STRIP=3 | |
| 70 try: | |
| 71 depth = os.environ['NATIVE_COVERAGE_DEPTH_STRIP'] | |
| 72 except KeyError: | |
| 73 logging.info('NATIVE_COVERAGE_DEPTH_STRIP is not defined: ' | |
| 74 'No native coverage.') | |
| 75 return '' | |
| 76 export_string = 'export GCOV_PREFIX="/data/local/tmp/gcov"\n' | |
| 77 export_string += 'export GCOV_PREFIX_STRIP=%s\n' % depth | |
| 78 return export_string | |
| 79 | |
| 80 def GetAllTests(self): | |
| 81 """Returns a list of all tests available in the test suite.""" | |
| 82 all_tests = self.adb.RunShellCommand( | |
| 83 '/data/local/tmp/%s --gtest_list_tests' % self.test_suite_basename) | |
| 84 return self._ParseGTestListTests(all_tests) | |
| 85 | |
| 86 def CreateTestRunnerScript(self, gtest_filter, test_arguments): | |
| 87 """Creates a test runner script and pushes to the device. | |
| 88 | |
| 89 Args: | |
| 90 gtest_filter: A gtest_filter flag. | |
| 91 test_arguments: Additional arguments to pass to the test binary. | |
| 92 """ | |
| 93 tool_wrapper = self.tool.GetTestWrapper() | |
| 94 sh_script_file = tempfile.NamedTemporaryFile() | |
| 95 # We need to capture the exit status from the script since adb shell won't | |
| 96 # propagate to us. | |
| 97 sh_script_file.write('cd /data/local/tmp\n' | |
| 98 '%s' | |
| 99 '%s /data/local/tmp/%s --gtest_filter=%s %s\n' | |
| 100 'echo $? > %s' % | |
| 101 (self._AddNativeCoverageExports(), | |
| 102 tool_wrapper, self.test_suite_basename, | |
| 103 gtest_filter, test_arguments, | |
| 104 TestPackageExecutable._TEST_RUNNER_RET_VAL_FILE)) | |
| 105 sh_script_file.flush() | |
| 106 cmd_helper.RunCmd(['chmod', '+x', sh_script_file.name]) | |
| 107 self.adb.PushIfNeeded(sh_script_file.name, | |
| 108 '/data/local/tmp/chrome_test_runner.sh') | |
| 109 logging.info('Conents of the test runner script: ') | |
| 110 for line in open(sh_script_file.name).readlines(): | |
| 111 logging.info(' ' + line.rstrip()) | |
| 112 | |
| 113 def RunTestsAndListResults(self): | |
| 114 """Runs all the tests and checks for failures. | |
| 115 | |
| 116 Returns: | |
| 117 A TestResults object. | |
| 118 """ | |
| 119 args = ['adb', '-s', self.device, 'shell', 'sh', | |
| 120 '/data/local/tmp/chrome_test_runner.sh'] | |
| 121 logging.info(args) | |
| 122 p = pexpect.spawn(args[0], args[1:], logfile=sys.stdout) | |
| 123 return self._WatchTestOutput(p) | |
| 124 | |
| 125 def StripAndCopyExecutable(self): | |
| 126 """Strips and copies the executable to the device.""" | |
| 127 if self.tool.NeedsDebugInfo(): | |
| 128 target_name = self.test_suite | |
| 129 else: | |
| 130 target_name = self.test_suite + '_' + self.device + '_stripped' | |
| 131 should_strip = True | |
| 132 if os.path.isfile(target_name): | |
| 133 logging.info('Found target file %s' % target_name) | |
| 134 target_mtime = os.stat(target_name).st_mtime | |
| 135 source_mtime = os.stat(self.test_suite).st_mtime | |
| 136 if target_mtime > source_mtime: | |
| 137 logging.info('Target mtime (%d) is newer than source (%d), assuming ' | |
| 138 'no change.' % (target_mtime, source_mtime)) | |
| 139 should_strip = False | |
| 140 | |
| 141 if should_strip: | |
| 142 logging.info('Did not find up-to-date stripped binary. Generating a ' | |
| 143 'new one (%s).' % target_name) | |
| 144 # Whenever we generate a stripped binary, copy to the symbols dir. If we | |
| 145 # aren't stripping a new binary, assume it's there. | |
| 146 if self.symbols_dir: | |
| 147 if not os.path.exists(self.symbols_dir): | |
| 148 os.makedirs(self.symbols_dir) | |
| 149 shutil.copy(self.test_suite, self.symbols_dir) | |
| 150 strip = os.environ['STRIP'] | |
| 151 cmd_helper.RunCmd([strip, self.test_suite, '-o', target_name]) | |
| 152 test_binary = '/data/local/tmp/' + self.test_suite_basename | |
| 153 self.adb.PushIfNeeded(target_name, test_binary) | |
| 154 | |
| 155 def _GetTestSuiteBaseName(self): | |
| 156 """Returns the base name of the test suite.""" | |
| 157 return os.path.basename(self.test_suite) | |
| OLD | NEW |