[Android] Redesigns the sharder to allow replicated vs distributed tests

build/android/pylib/instrumentation/setup.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.
+
+"""Runs the instrumentation tests."""
+
+import logging
+import os
+
+from pylib import android_commands
+from pylib import constants
+from pylib.base import base_test_result
+from pylib.utils import report_results
+
+import test_package
+import test_runner
+
+
+def Setup(test_apk_path, test_apk_jar_path, annotations, exclude_annotations,
+          test_filter, build_type, test_data, install_apk, save_perf_json,
+          screenshot_failures, tool, wait_for_debugger, disable_assertions,
+          push_deps):
+  """Create and return the test runner factory and tests.
+
+  Args:
+    test_apk_path: Path to the test apk file.
+    test_apk_jar_path: Path to the jar associated with the test apk.
+    annotations: Annotations for the tests.
+    exclude_annotations: Any annotations to exclude from running.
+    test_filter: Filter string for tests.
+    build_type: 'Release' or 'Debug'.
+    test_data: Location of the test data.
+    install_apk: Re-installs the apk if opted.
+    save_perf_json: Whether or not to save the JSON file from UI perf tests.
+    screenshot_failures: Take a screenshot for a test failure
+    tool: Name of the Valgrind tool.
+    wait_for_debugger: blocks until the debugger is connected.
+    disable_assertions: Whether to disable java assertions on the device.
+    push_deps: If True, push all dependencies to the device.
+
+  Returns:
+    A tuple of (TestRunnerFactory, tests).
+  """
+  test_pkg = test_package.TestPackage(test_apk_path, test_apk_jar_path)
+  tests = test_pkg._GetAllMatchingTests(annotations, exclude_annotations,
+                                        test_filter)
+  if not tests:
+    logging.error('No instrumentation tests to run with current args.')
+
+  def TestRunnerFactory(device, shard_index):
+    return test_runner.TestRunner(
+        build_type, test_data, install_apk, save_perf_json, screenshot_failures,
+        tool, wait_for_debugger, disable_assertions, push_deps, device,
+        shard_index, test_pkg, [])
+
+  return (TestRunnerFactory, tests)