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
7import logging
8import os.path
9import sys
10import traceback
11
12from pylib import constants
13from pylib.base import base_test_result
14from pylib.base import base_test_runner
15from pylib.linker import test_case
16from 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
23class 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
44class 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