summaryrefslogtreecommitdiffstats
path: root/build/android/pylib/instrumentation/test_package.py
blob: e2e3b77adfed5228d4f8118fb594d049d55b9056 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
# Copyright (c) 2013 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Class representing instrumentation test apk and jar."""

import os

from devil.android import apk_helper
from pylib.instrumentation import test_jar


class TestPackage(test_jar.TestJar):
  def __init__(self, apk_path, jar_path, test_support_apk_path,
               additional_apks=None, apk_under_test=None):
    test_jar.TestJar.__init__(self, jar_path)

    if not os.path.exists(apk_path):
      raise Exception('%s not found, please build it' % apk_path)
    self._additional_apks = additional_apks or []
    self._apk_name = os.path.splitext(os.path.basename(apk_path))[0]
    if apk_under_test:
      self._apk_under_test = apk_helper.ApkHelper(apk_under_test)
    else:
      self._apk_under_test = None
    self._test_apk = apk_helper.ApkHelper(apk_path)
    self._test_support_apk_path = test_support_apk_path

  def GetApkPath(self):
    """Returns the absolute path to the APK."""
    return self._test_apk.path

  def GetApkUnderTest(self):
    """Returns an ApkHelper instance for the apk under test.

    Note that --apk-under-test is not required, so this can be None.
    """
    return self._apk_under_test

  def GetApkName(self):
    """Returns the name of the apk without the suffix."""
    return self._apk_name

  def GetPackageName(self):
    """Returns the package name of this APK."""
    return self._test_apk.GetPackageName()

  def GetTestApk(self):
    """Returns an ApkHelper instance for the test apk."""
    return self._test_apk

  # Override.
  def Install(self, device):
    if self._apk_under_test:
      device.Install(self._apk_under_test.path)
    device.Install(self.GetApkPath())
    if (self._test_support_apk_path and
        os.path.exists(self._test_support_apk_path)):
      device.Install(self._test_support_apk_path)
    for apk in (a for a in self._additional_apks if os.path.exists(a)):
      device.Install(apk)