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

tools/android/loading/controller.py

 # Copyright 2016 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.
 
 """Controller objects that control the context in which chrome runs.
 
 This is responsible for the setup necessary for launching chrome, and for
 creating a DevToolsConnection. There are remote device and local
 desktop-specific versions.
 """
 
 import contextlib
 import copy
 import datetime
 import errno
 import logging
 import os
 import platform
 import shutil
 import socket
 import subprocess
 import sys
 import tempfile
 import time
 import traceback
 
 import psutil
 
 import chrome_cache
+import chrome_setup
 import common_util
 import device_setup
 import devtools_monitor
 import emulation
 from options import OPTIONS
 
 _SRC_DIR = os.path.abspath(os.path.join(
     os.path.dirname(__file__), '..', '..', '..'))
 _CATAPULT_DIR = os.path.join(_SRC_DIR, 'third_party', 'catapult')
 
@@ skipping 99 matching lines @@
 class ChromeControllerBase(object):
   """Base class for all controllers.
 
   Defines common operations but should not be created directly.
   """
   METADATA_GATHERER = ChromeControllerMetadataGatherer()
   DEVTOOLS_CONNECTION_ATTEMPTS = 10
   DEVTOOLS_CONNECTION_ATTEMPT_INTERVAL_SECONDS = 1
 
   def __init__(self):
-    self._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(gabadie): Remove once crbug.com/354743 done.
-        '--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',
-
+    self._chrome_args = chrome_setup.CHROME_ARGS + [
         # Tests & dev-tools related stuff.
         '--enable-test-events',
         '--remote-debugging-port=%d' % OPTIONS.devtools_port,
 
         # Detailed log.
         '--enable-logging=stderr',
         '--v=1',
     ]
     self._wpr_attributes = None
     self._metadata = {}
@@ skipping 223 matching lines @@
       except Exception:
         logcat = ''.join([l + '\n' for l in self._device.adb.Logcat(dump=True)])
         raise ChromeControllerError(log=logcat)
       finally:
         self._device.ForceStop(package_info.package)
         self._DismissCrashDialogIfNeeded()
 
   def ResetBrowserState(self):
     """Override resetting Chrome local state."""
     logging.info('Resetting Chrome local state')
-    package = OPTIONS.ChromePackage().package
-    # Remove the Chrome Profile and the various disk caches. Other parts
-    # theoretically should not affect loading performance. Also remove the tab
-    # state to prevent it from growing infinitely. [:D]
-    for directory in ['app_chrome/Default', 'cache', 'app_chrome/ShaderCache',
-                      'app_tabs']:
-      cmd = ['rm', '-rf', '/data/data/{}/{}'.format(package, directory)]
-      self._device.adb.Shell(subprocess.list2cmdline(cmd))
+    chrome_setup.ResetChromeLocalState(self._device,
+                                       OPTIONS.ChromePackage().package)
+
 
   def RebootDevice(self):
     """Reboot the remote device."""
     assert self._wpr_attributes is None, 'WPR should be closed before rebooting'
     logging.warning('Rebooting the device')
     device_setup.Reboot(self._device)
     self._InitDevice()
 
   def PushBrowserCache(self, cache_path):
     """Override for chrome cache pushing."""
@@ skipping 217 matching lines @@
     if (not os.path.isdir(self._profile_dir) or
         os.listdir(self._profile_dir) == []):
       # Launch chrome so that it populates the profile directory.
       with self.Open():
         pass
     assert os.path.isdir(self._profile_dir)
     assert os.path.isdir(os.path.dirname(self._GetCacheDirectoryPath()))
 
   def _GetCacheDirectoryPath(self):
     return os.path.join(self._profile_dir, 'Default', 'Cache')