[tools/android/loading] Helper script running Chrome on device with WPR

tools/android/loading/wpr_helper.py

Index: tools/android/loading/wpr_helper.py
diff --git a/tools/android/loading/wpr_helper.py b/tools/android/loading/wpr_helper.py
new file mode 100755
index 0000000000000000000000000000000000000000..525848e9d4fa2a1247b1d41554811c929570d02f
--- /dev/null
+++ b/tools/android/loading/wpr_helper.py
@@ -0,0 +1,154 @@
+#!/usr/bin/python
+#
+# Copyright 2015 The Chromium Authors. All rights reserved.

[alexilin, 2017/02/10 14:37:22]

nit:
s/2015/2017/

+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""Helper script to launch Chrome on device and WebPageReplay on host."""
+
+import logging
+import optparse
+import os
+import subprocess
+import sys
+import time
+
+_SRC_PATH = os.path.abspath(os.path.join(
+    os.path.dirname(__file__), os.pardir, os.pardir, os.pardir))
+
+sys.path.append(os.path.join(_SRC_PATH, 'third_party', 'catapult', 'devil'))
+from devil.android import device_utils
+from devil.android.constants import chrome
+from devil.android.perf import cache_control
+from devil.android.sdk import intent
+
+sys.path.append(os.path.join(_SRC_PATH, 'build', 'android'))
+import devil_chromium
+
+sys.path.append(os.path.join(_SRC_PATH, 'tools', 'android', 'loading'))
+import device_setup
+
+
+# Command line arguments for Chrome.
+_CHROME_ARGS = [
+    # Disable backgound network requests that may pollute WPR archive, pollute
+    # HTTP cache generation, and introduce noise in loading performance.
+    '--disable-background-networking',
+    '--disable-default-apps',
+    '--no-proxy-server',
+    # TODO(droger): Remove once crbug.com/354743 is fixed.
+    '--safebrowsing-disable-auto-update',
+
+    # Disables actions that chrome performs only on first run or each launches,
+    # which can interfere with page load performance, or even block its
+    # execution by waiting for user input.
+    '--disable-fre',
+    '--no-default-browser-check',
+    '--no-first-run',
+]
+
+
+def ResetChromeLocalState(device, package):
+  """Remove the Chrome Profile and the various disk caches."""
+  profile_dirs = ['app_chrome/Default', 'cache', 'app_chrome/ShaderCache',
+                  'app_tabs']
+  cmd = ['rm', '-rf']
+  cmd.extend(
+      '/data/data/{}/{}'.format(package, d) for d in profile_dirs)
+  device.adb.Shell(subprocess.list2cmdline(cmd))
+
+
+def RunChrome(device, cold, chrome_args, package_info):
+  """Runs Chrome on the device.
+
+  Args:
+    device: (DeviceUtils) device to run the tests on.
+    cold: (bool) Whether caches should be dropped.
+    chrome_args: ([str]) List of arguments to pass to Chrome.
+    package_info: (PackageInfo) Chrome package info.
+  """
+  if not device.HasRoot():
+    device.EnableRoot()
+
+  cmdline_file = package_info.cmdline_file
+  package = package_info.package
+  with device_setup.FlagReplacer(device, cmdline_file, chrome_args):
+    device.ForceStop(package)
+
+    if cold:
+      ResetChromeLocalState(device, package)
+      cache_control.CacheControl(device).DropRamCaches()
+
+    start_intent = intent.Intent(package=package, data='about:blank',
+                                 activity=package_info.activity)
+    try:
+      device.StartActivity(start_intent, blocking=True)
+      print(
+          '\n\n'
+          '   +---------------------------------------------+\n'
+          '   | Chrome launched, press Ctrl-C to interrupt. |\n'
+          '   +---------------------------------------------+')
+      while True:
+        time.sleep(1)
+    except KeyboardInterrupt:
+      pass
+    finally:
+      device.ForceStop(package)
+
+
+def _CreateOptionParser():
+  description = 'Launches Chrome on a device, connected to a WebPageReplay ' \
+                'instance running on the host. The WPR archive must be ' \
+                'passed as parameter.'
+  parser = optparse.OptionParser(description=description,
+                                 usage='Usage: %prog [options] wpr_archive')
+
+  # Device-related options.
+  d = optparse.OptionGroup(parser, 'Device options')
+  d.add_option('--device', help='Device ID')
+  d.add_option('--cold', help='Purge all caches before running Chrome.',
+               default=False, action='store_true')
+  d.add_option('--chrome_package_name',
+               help='Chrome package name (e.g. "chrome" or "chromium") '
+               '[default: %default].', default='chrome')
+  parser.add_option_group(d)
+
+  # WebPageReplay-related options.
+  w = optparse.OptionGroup(parser, 'WebPageReplay options')
+  w.add_option('--record',
+               help='Enable this to record a new WPR archive.',
+               action='store_true', default=False)
+  w.add_option('--wpr_log', help='WPR log path.')
+  w.add_option('--network_condition', help='Network condition for emulation.')
+  parser.add_option_group(w)
+
+  return parser
+
+
+def main():
+  parser = _CreateOptionParser()
+  options, args = parser.parse_args()
+  if len(args) != 1:
+    parser.error("Incorrect number of arguments.")
+  devil_chromium.Initialize()
+  devices = device_utils.DeviceUtils.HealthyDevices()
+  device = devices[0]
+  if len(devices) != 1 and options.device is None:
+    logging.error('Several devices attached, must specify one with --device.')
+    sys.exit(0)
+  if options.device is not None:
+    matching_devices = [d for d in devices if str(d) == options.device]
+    if not matching_devices:
+      logging.error('Device not found.')
+      sys.exit(0)
+    device = matching_devices[0]
+
+  with device_setup.RemoteWprHost(device, args[0], options.record,
+                                  options.network_condition,
+                                  out_log_path=options.wpr_log) as wpr_attr:
+    RunChrome(device, options.cold, _CHROME_ARGS + wpr_attr.chrome_args,
+              chrome.PACKAGE_INFO[options.chrome_package_name])
+
+
+if __name__ == '__main__':
+  main()