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

tools/android/customtabs_benchmark/scripts/customtabs_benchmark.py

 #!/usr/bin/python
 #
 # Copyright 2015 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.
 
 """Loops Custom Tabs tests and outputs the results into a CSV file."""
 
 import collections
 import contextlib
 import logging
 import optparse
 import os
 import random
 import re
-import subprocess
 import sys
 import time
 
 _SRC_PATH = os.path.abspath(os.path.join(
     os.path.dirname(__file__), os.pardir, os.pardir, os.pardir, os.pardir))
 
 sys.path.append(os.path.join(_SRC_PATH, 'third_party', 'catapult', 'devil'))
 from devil.android import device_errors
 from devil.android import device_utils
 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 chrome_setup
 import device_setup
 
 
 # Local build of Chrome (not Chromium).
 _CHROME_PACKAGE = 'com.google.android.apps.chrome'
 _COMMAND_LINE_PATH = '/data/local/tmp/chrome-command-line'
 _TEST_APP_PACKAGE_NAME = 'org.chromium.customtabsclient.test'
 _INVALID_VALUE = -1
 
 
-# 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):
-  """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(_CHROME_PACKAGE, d) for d in profile_dirs)
-  device.adb.Shell(subprocess.list2cmdline(cmd))
-
-
 def RunOnce(device, url, warmup, speculation_mode, delay_to_may_launch_url,
             delay_to_launch_url, cold, chrome_args, reset_chrome_state):
   """Runs a test on a device once.
 
   Args:
     device: (DeviceUtils) device to run the tests on.
     url: (str) URL to load.
     warmup: (bool) Whether to call warmup.
     speculation_mode: (str) Speculation Mode.
     delay_to_may_launch_url: (int) Delay to mayLaunchUrl() in ms.
@@ skipping 27 matching lines @@
                 'delay_to_may_launch_url': delay_to_may_launch_url,
                 'delay_to_launch_url': delay_to_launch_url,
                 'timeout': timeout_s})
     result_line_re = re.compile(r'CUSTOMTABSBENCH.*: (.*)')
     logcat_monitor = device.GetLogcatMonitor(clear=True)
     logcat_monitor.Start()
     device.ForceStop(_CHROME_PACKAGE)
     device.ForceStop(_TEST_APP_PACKAGE_NAME)
 
     if reset_chrome_state:
-      ResetChromeLocalState(device)
+      chrome_setup.ResetChromeLocalState(device, _CHROME_PACKAGE)
 
     if cold:
       cache_control.CacheControl(device).DropRamCaches()
 
     device.StartActivity(launch_intent, blocking=True)
 
     match = None
     try:
       match = logcat_monitor.WaitFor(result_line_re, timeout=logcat_timeout)
     except device_errors.CommandTimeoutError as _:
@@ skipping 43 matching lines @@
     once: (bool) Run only once.
     should_stop: (threading.Event or None) When the event is set, stop looping.
   """
   with SetupWpr(device, wpr_archive_path, wpr_record, network_condition,
                 wpr_log_path) as wpr_attributes:
     to_stdout = output_filename == '-'
     out = sys.stdout if to_stdout else open(output_filename, 'a')
     try:
       while should_stop is None or not should_stop.is_set():
         config = configs[random.randint(0, len(configs) - 1)]
-        chrome_args = CHROME_ARGS + wpr_attributes.chrome_args
+        chrome_args = chrome_setup.CHROME_ARGS + wpr_attributes.chrome_args
         if config['speculation_mode'] == 'no_state_prefetch':
           # NoStatePrefetch is enabled through an experiment.
           chrome_args.extend([
               '--force-fieldtrials=trial/group',
               '--force-fieldtrial-params=trial.group:mode/no_state_prefetch',
               '--enable-features="NoStatePrefetch<trial"'])
         elif config['speculation_mode'] == 'speculative_prefetch':
           # Speculative Prefetch is enabled through an experiment.
           chrome_args.extend([
               '--force-fieldtrials=trial/group',
@@ skipping 129 matching lines @@
       'delay_to_launch_url': options.delay_to_launch_url,
       'cold': options.cold,
   }
   LoopOnDevice(device, [config], options.output_file, options.wpr_archive,
                options.record, options.network_condition, options.wpr_log,
                once=options.once)
 
 
 if __name__ == '__main__':
   main()