[Android] Enable using isolate files to get a list of data dependencies to push to the device.

build/android/pylib/gtest/test_runner.py

 # Copyright (c) 2012 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.
 
 import glob
 import logging
 import os
+import sys
 
 from pylib import android_commands
+from pylib import cmd_helper
 from pylib import constants
 from pylib import perf_tests_helper
 from pylib.android_commands import errors
 from pylib.base import base_test_result
 from pylib.base import base_test_runner
 from pylib.utils import run_tests_helper
 
 import test_package_apk
 import test_package_executable
 
+sys.path.insert(
+    0, os.path.join(constants.DIR_SOURCE_ROOT, 'tools', 'swarm_client'))
+import run_isolated
 
-def _GetDataFilesForTestSuite(test_suite_basename):
+
+_ISOLATE_FILE_PATHS = {
+  'base_unittests': 'base/base_unittests.isolate',
+  #'net_unittests': 'net/net_unittests.isolate',
+  #'unit_tests': 'chrome/unit_tests.isolate',
+  #'content_browsertests': 'content/content_browsertests.isolate',
+  #'content_unittests': 'content/content_unittests.isolate',
+  }
+_ISOLATE_SCRIPT = os.path.join(
+    constants.DIR_SOURCE_ROOT, 'tools', 'swarm_client', 'isolate.py')
+
+
+def _GetDataFilesForTestSuite(product_dir, test_suite_basename):
   """Returns a list of data files/dirs needed by the test suite.
 
   Args:
-    test_suite_basename: The test suite basename for which to return file paths.
+    product_dir: Absolute path to product directory (e.g. /../out/Debug).
+    test_suite_basename: The test suite basename (e.g. base_unittests).
 
   Returns:
     A list of test file and directory paths.
   """
+  # TODO(frankf): *.isolated should be generated by gyp using foo_tests_run
+  # targets. This is a stop-gap solution as currently there are no such
+  # targets for content_unittests and content_browsertests.
+  isolate_rel_path = _ISOLATE_FILE_PATHS.get(test_suite_basename)
+  if isolate_rel_path:
+    isolate_abs_path = os.path.join(constants.DIR_SOURCE_ROOT, isolate_rel_path)
+    isolated_abs_path = os.path.join(
+        product_dir, '%s.isolated' % test_suite_basename)
+    assert os.path.exists(isolate_abs_path)
+    isolate_cmd = [
+      'python', _ISOLATE_SCRIPT,
+      'check',
+      '--isolate=%s' % isolate_abs_path,
+      '--isolated=%s' % isolated_abs_path,
+      '-V', 'PRODUCT_DIR=%s' % product_dir,
+      '-V', 'OS=android',
+      '--outdir=%s' % product_dir,
+      ]
+    assert not cmd_helper.RunCmd(isolate_cmd)
+    with open(isolated_abs_path) as f:
+      isolated_content = run_isolated.load_isolated(f.read())
+      assert isolated_content['os'] == 'android'
+      return isolated_content['files'].keys()
+
+  logging.info('Did not find an isolate file for the test suite.')
   # Ideally, we'd just push all test data. However, it has >100MB, and a lot
   # of the files are not relevant (some are used for browser_tests, others for
   # features not supported, etc..).
-  if test_suite_basename == 'base_unittests':
-    return [
-        'base/test/data/',
-    ]
-  elif test_suite_basename == 'unit_tests':
+  if test_suite_basename == 'unit_tests':
     test_files = [
         'base/test/data/',
         'chrome/test/data/download-test1.lib',
         'chrome/test/data/extensions/bad_magic.crx',
         'chrome/test/data/extensions/good.crx',
         'chrome/test/data/extensions/icon1.png',
         'chrome/test/data/extensions/icon2.png',
         'chrome/test/data/extensions/icon3.png',
         'chrome/test/data/extensions/allow_silent_upgrade/',
         'chrome/test/data/extensions/app/',
@@ skipping 225 matching lines @@
           symbols_dir)
 
   #override
   def InstallTestPackage(self):
     self.test_package.StripAndCopyExecutable()
     self.test_package.PushDataAndPakFiles()
 
   #override
   def PushDataDeps(self):
     self.tool.CopyFiles()
-    test_data = _GetDataFilesForTestSuite(self.test_package.test_suite_basename)
+    test_data = _GetDataFilesForTestSuite(self.test_package.test_suite_dirname,
+                                          self.test_package.test_suite_basename)
     if test_data:
       # Make sure SD card is ready.
       self.adb.WaitForSdCardReady(20)
       for data in test_data:
         self.CopyTestData([data], self.adb.GetExternalStorage())
     optional_test_data = _GetOptionalDataFilesForTestSuite(
         self.test_package.test_suite_basename)
     if optional_test_data:
       self.adb.WaitForSdCardReady(20)
       for data in optional_test_data:
@@ skipping 73 matching lines @@
       self.LaunchChromeTestServerSpawner()
     self.tool.SetupEnvironment()
 
   #override
   def TearDown(self):
     """Cleans up the test enviroment for the test suite."""
     self.tool.CleanUpEnvironment()
     if self._cleanup_test_files:
       self.adb.RemovePushedFiles()
     super(TestRunner, self).TearDown()