Index: build/android/pylib/instrumentation/test_package.py |
diff --git a/build/android/pylib/instrumentation/test_package.py b/build/android/pylib/instrumentation/test_package.py |
index 1f56b5e0892316e134ff3ba9f0ed786283660c7d..d6469b2340c8a4147fda3b0e7baaee2c5d523972 100644 |
--- a/build/android/pylib/instrumentation/test_package.py |
+++ b/build/android/pylib/instrumentation/test_package.py |
@@ -11,13 +11,16 @@ from pylib.instrumentation import test_jar |
class TestPackage(test_jar.TestJar): |
- def __init__(self, apk_path, jar_path, test_support_apk_path): |
+ 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._apk_path = apk_path |
+ self._additional_apks = additional_apks or [] |
self._apk_name = os.path.splitext(os.path.basename(apk_path))[0] |
+ self._apk_path = apk_path |
+ self._apk_under_test = apk_under_test |
self._package_name = apk_helper.GetPackageName(self._apk_path) |
self._test_support_apk_path = test_support_apk_path |
@@ -35,7 +38,11 @@ class TestPackage(test_jar.TestJar): |
# Override. |
def Install(self, device): |
+ if self._apk_under_test: |
+ device.Install(self._apk_under_test) |
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) |