Home | History | Annotate | Download | only in test_defs
      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       # expect tests in /data/nativetest/test_file/test_file
     93       suite.SetTargetExecPath(os.path.join(target_root_path, test_file, test_file))
     94       test_suites.append(suite)
     95     return test_suites
     96 
     97   def _CollectTestSources(self, test_list, dirname, files):
     98     """For each directory, find tests source file and add them to the list.
     99 
    100     Test files must match one of the following pattern:
    101       - test_*.[cc|cpp]
    102       - *_test.[cc|cpp]
    103       - *_unittest.[cc|cpp]
    104 
    105     This method is a callback for os.path.walk.
    106 
    107     Args:
    108       test_list: Where new tests should be inserted.
    109       dirname: Current directory.
    110       files: List of files in the current directory.
    111     """
    112     for f in files:
    113       self._EvaluateFile(test_list, f)
    114 
    115   def _EvaluateFile(self, test_list, file):
    116     (name, ext) = os.path.splitext(file)
    117     if ext == ".cc" or ext == ".cpp" or ext == ".c":
    118       if re.search("_test$|_test_$|_unittest$|_unittest_$|^test_", name):
    119         logger.SilentLog("Found native test file %s" % file)
    120         test_list.append(name)
    121