1 #!/usr/bin/python 2 # 3 # 4 # Copyright 2011, The Android Open Source Project 5 # 6 # Licensed under the Apache License, Version 2.0 (the "License"); 7 # you may not use this file except in compliance with the License. 8 # You may obtain a copy of the License at 9 # 10 # http://www.apache.org/licenses/LICENSE-2.0 11 # 12 # Unless required by applicable law or agreed to in writing, software 13 # distributed under the License is distributed on an "AS IS" BASIS, 14 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 # See the License for the specific language governing permissions and 16 # limitations under the License. 17 18 """TestSuite for running C/C++ Android tests using gtest framework.""" 19 20 # python imports 21 import os 22 import re 23 24 # local imports 25 import logger 26 import run_command 27 import test_suite 28 29 30 class GTestSuite(test_suite.AbstractTestSuite): 31 """A test suite for running gtest on device.""" 32 33 def __init__(self): 34 test_suite.AbstractTestSuite.__init__(self) 35 self._target_exec_path = None 36 37 def GetTargetExecPath(self): 38 """Get the target path to gtest executable.""" 39 return self._target_exec_path 40 41 def SetTargetExecPath(self, path): 42 self._target_exec_path = path 43 return self 44 45 def Run(self, options, adb): 46 """Run the provided gtest test suite. 47 48 Args: 49 options: command line options 50 adb: adb interface 51 """ 52 shell_cmd = adb.PreviewShellCommand(self.GetTargetExecPath()) 53 logger.Log(shell_cmd) 54 if not options.preview: 55 # gtest will log to test results to stdout, so no need to do any 56 # extra processing 57 run_command.RunCommand(shell_cmd, return_output=False) 58 59 60 class GTestFactory(test_suite.AbstractTestFactory): 61 62 def __init__(self, test_root_path, build_path): 63 test_suite.AbstractTestFactory.__init__(self, test_root_path, 64 build_path) 65 66 def CreateTests(self, sub_tests_path=None): 67 """Create tests found in sub_tests_path. 68 69 Looks for test files matching a pattern, and assumes each one is a separate 70 binary on target. 71 72 Test files must match one of the following pattern: 73 - test_*.[c|cc|cpp] 74 - *_test.[c|cc|cpp] 75 - *_unittest.[c|cc|cpp] 76 77 """ 78 if not sub_tests_path: 79 sub_tests_path = self.GetTestRootPath() 80 test_file_list = [] 81 if os.path.isfile(sub_tests_path): 82 self._EvaluateFile(test_file_list, os.path.basename(sub_tests_path)) 83 else: 84 os.path.walk(sub_tests_path, self._CollectTestSources, test_file_list) 85 # TODO: obtain this from makefile instead of hardcoding 86 target_root_path = os.path.join('/data', 'nativetest') 87 test_suites = [] 88 for test_file in test_file_list: 89 logger.SilentLog('Creating gtest suite for file %s' % test_file) 90 suite = GTestSuite() 91 suite.SetBuildPath(self.GetBuildPath()) 92 suite.SetTargetExecPath(os.path.join(target_root_path, test_file)) 93 test_suites.append(suite) 94 return test_suites 95 96 def _CollectTestSources(self, test_list, dirname, files): 97 """For each directory, find tests source file and add them to the list. 98 99 Test files must match one of the following pattern: 100 - test_*.[cc|cpp] 101 - *_test.[cc|cpp] 102 - *_unittest.[cc|cpp] 103 104 This method is a callback for os.path.walk. 105 106 Args: 107 test_list: Where new tests should be inserted. 108 dirname: Current directory. 109 files: List of files in the current directory. 110 """ 111 for f in files: 112 self._EvaluateFile(test_list, f) 113 114 def _EvaluateFile(self, test_list, file): 115 (name, ext) = os.path.splitext(file) 116 if ext == ".cc" or ext == ".cpp" or ext == ".c": 117 if re.search("_test$|_test_$|_unittest$|_unittest_$|^test_", name): 118 logger.SilentLog("Found native test file %s" % file) 119 test_list.append(name) 120