Add support for getting real instrumentation value in incremental build

devil/devil/android/apk_helper.py

 # 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.
 
 """Module containing utilities for apk packages."""
 
 import itertools
 import re
 
 from devil import base_error
@@ skipping 139 matching lines @@
     if len(all_instrumentations) != 1:
       raise base_error.BaseError(
           'There is more than one instrumentation. Expected one.')
     else:
       return self._ResolveName(all_instrumentations[0]['android:name'])
 
   def GetAllInstrumentations(
       self, default='android.test.InstrumentationTestRunner'):
     """Returns a list of all Instrumentations in the apk."""
     try:
-      return self._GetManifest()['manifest'][0]['instrumentation']
+      manifest = self._GetManifest()['manifest'][0]
+      real_instrumentation_values = self._GetRealInstrumentationValue()
+      if real_instrumentation_values:
+        for index, instr in enumerate(manifest['instrumentation']):
+          key = 'incremental-install-real-instrumentation-' + str(index)
+          instr['android:name'] = real_instrumentation_values[key]
+      return manifest['instrumentation']
     except KeyError:
       return [{'android:name': default}]
 
   def GetPackageName(self):
     """Returns the package name of the apk."""
     manifest_info = self._GetManifest()
     try:
       return manifest_info['manifest'][0]['package']
     except KeyError:
       raise Exception('Failed to determine package name of %s' % self._apk_path)
@@ skipping 19 matching lines @@
     manifest_info = self._GetManifest()
     try:
       applications = manifest_info['manifest'][0].get('application', [])
       services = itertools.chain(
           *(application.get('service', []) for application in applications))
       return any(
           _ParseNumericKey(s, 'android:isolatedProcess') for s in services)
     except KeyError:
       return False
 
+  def _GetRealInstrumentationValue(self):
+    manifest = self._GetManifest()['manifest'][0]
+    real_instrumentation_values = {}
+    if (manifest.get('application') and
+        manifest['application'][0].get('meta-data')):
+      meta_data = manifest.get('application')[0]['meta-data']
+      for i in meta_data:
+        if 'incremental-install-real-instrumentation' in i['android:name']:
+          real_instrumentation_values[i['android:name']] = i['android:value']
+    return real_instrumentation_values
+
   def _GetManifest(self):
     if not self._manifest:
       self._manifest = _ParseManifestFromApk(self._apk_path)
     return self._manifest
 
   def _ResolveName(self, name):
     name = name.lstrip('.')
     if '.' not in name:
       return '%s.%s' % (self.GetPackageName(), name)
     return name