Home | History | Annotate | Download | only in linker 
      1 # Copyright 2013 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 """Runs linker tests on a particular device."""
      6 
      7 import logging
      8 import os.path
      9 import sys
     10 import traceback
     11 
     12 from pylib import constants
     13 from pylib.base import base_test_result
     14 from pylib.base import base_test_runner
     15 from pylib.linker import test_case
     16 from pylib.utils import apk_helper
     17 
     18 
     19 # Name of the Android package to install for this to work.
     20 _PACKAGE_NAME = 'ChromiumLinkerTest'
     21 
     22 
     23 class LinkerExceptionTestResult(base_test_result.BaseTestResult):
     24   """Test result corresponding to a python exception in a host-custom test."""
     25 
     26   def __init__(self, test_name, exc_info):
     27     """Constructs a LinkerExceptionTestResult object.
     28 
     29     Args:
     30       test_name: name of the test which raised an exception.
     31       exc_info: exception info, ostensibly from sys.exc_info().
     32     """
     33     exc_type, exc_value, exc_traceback = exc_info
     34     trace_info = ''.join(traceback.format_exception(exc_type, exc_value,
     35                                                     exc_traceback))
     36     log_msg = 'Exception:\n' + trace_info
     37 
     38     super(LinkerExceptionTestResult, self).__init__(
     39         test_name,
     40         base_test_result.ResultType.FAIL,
     41         log = "%s %s" % (exc_type, log_msg))
     42 
     43 
     44 class LinkerTestRunner(base_test_runner.BaseTestRunner):
     45   """Orchestrates running a set of linker tests.
     46 
     47   Any Python exceptions in the tests are caught and translated into a failed
     48   result, rather than being re-raised on the main thread.
     49   """
     50 
     51   #override
     52   def __init__(self, device, tool, push_deps, cleanup_test_files):
     53     """Creates a new LinkerTestRunner.
     54 
     55     Args:
     56       device: Attached android device.
     57       tool: Name of the Valgrind tool.
     58       push_deps: If True, push all dependencies to the device.
     59       cleanup_test_files: Whether or not to cleanup test files on device.
     60     """
     61 
     62     super(LinkerTestRunner, self).__init__(device, tool, push_deps,
     63                                                cleanup_test_files)
     64 
     65   #override
     66   def InstallTestPackage(self):
     67     apk_path = os.path.join(
     68         constants.GetOutDirectory(), 'apks', '%s.apk' % _PACKAGE_NAME)
     69 
     70     if not os.path.exists(apk_path):
     71       raise Exception('%s not found, please build it' % apk_path)
     72 
     73     package_name = apk_helper.GetPackageName(apk_path)
     74     self.device.old_interface.ManagedInstall(apk_path, package_name)
     75 
     76   #override
     77   def RunTest(self, test):
     78     """Sets up and runs a test case.
     79 
     80     Args:
     81       test: An object which is ostensibly a subclass of LinkerTestCaseBase.
     82 
     83     Returns:
     84       A TestRunResults object which contains the result produced by the test
     85       and, in the case of a failure, the test that should be retried.
     86     """
     87 
     88     assert isinstance(test, test_case.LinkerTestCaseBase)
     89 
     90     try:
     91       results = test.Run(self.device)
     92     except Exception:
     93       logging.exception('Caught exception while trying to run test: ' +
     94                         test.tagged_name)
     95       exc_info = sys.exc_info()
     96       results = base_test_result.TestRunResults()
     97       results.AddResult(LinkerExceptionTestResult(
     98           test.tagged_name, exc_info))
     99 
    100     if not results.DidRunPass():
    101       return results, test
    102     else:
    103       return results, None
    104