src/build/android/pylib/linker/test_runner.py

   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.adb.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