Add shared preference file modification option to instrumentation test runner

build/android/pylib/local/device/local_device_instrumentation_test_run.py

 # 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.
 
 import logging
 import os
 import posixpath
 import re
 import time
 
 from devil.android import device_errors
 from devil.android import flag_changer
+from devil.android.sdk import shared_prefs
 from devil.utils import reraiser_thread
 from pylib import valgrind_tools
 from pylib.android import logdog_logcat_monitor
 from pylib.base import base_test_result
 from pylib.instrumentation import instrumentation_test_instance
 from pylib.local.device import local_device_environment
 from pylib.local.device import local_device_test_run
 from pylib.utils import logdog_helper
 from py_trace_event import trace_event
 from py_utils import contextlib_ext
@@ skipping 79 matching lines @@
         if self._test_instance.flags:
           if not self._test_instance.package_info:
             logging.error("Couldn't set debug app: no package info")
           elif not self._test_instance.package_info.package:
             logging.error("Couldn't set debug app: no package defined")
           else:
             dev.RunShellCommand(['am', 'set-debug-app', '--persistent',
                                   self._test_instance.package_info.package],
                                 check_return=True)
 
+      def edit_shared_prefs():
+        for pref in self._test_instance.edit_shared_prefs:
+          prefs = shared_prefs.SharedPrefs(dev, pref['package'],
+                                           pref['filename'])
+          prefs.Load()
+          for key in pref.get('remove', []):
+            try:
+              prefs.Remove(key)
+            except KeyError:
+              logging.warning("Attempted to remove non-existent key %s", key)
+          for key, value in pref.get('set', {}).iteritems():
+            if isinstance(value, bool):
+              prefs.SetBoolean(key, value)
+            elif isinstance(value, basestring):
+              prefs.SetString(key, value)
+            elif isinstance(value, long) or isinstance(value, int):
+              prefs.SetLong(key, value)
+            elif isinstance(value, list):
+              prefs.SetStringSet(key, value)
+            else:
+              raise ValueError("Given invalid value type %s for key %s" % (
+                  str(type(value)), key))
+          prefs.Commit()
+
       def push_test_data():
         device_root = posixpath.join(dev.GetExternalStoragePath(),
                                      'chromium_tests_root')
         host_device_tuples_substituted = [
             (h, local_device_test_run.SubstituteDeviceRoot(d, device_root))
             for h, d in host_device_tuples]
         logging.info('instrumentation data deps:')
         for h, d in host_device_tuples_substituted:
           logging.info('%r -> %r', h, d)
         dev.PushChangedFiles(host_device_tuples_substituted,
@@ skipping 10 matching lines @@
             logging.error("Couldn't set flags: no cmdline_file")
           else:
             self._CreateFlagChangerIfNeeded(dev)
             logging.debug('Attempting to set flags: %r',
                           self._test_instance.flags)
             self._flag_changers[str(dev)].AddFlags(self._test_instance.flags)
 
         valgrind_tools.SetChromeTimeoutScale(
             dev, self._test_instance.timeout_scale)
 
-      steps = (install_apk, push_test_data, create_flag_changer)
+      steps = (install_apk, edit_shared_prefs, push_test_data,
+               create_flag_changer)
       if self._env.concurrent_adb:
         reraiser_thread.RunAsync(steps)
       else:
         for step in steps:
           step()
       if self._test_instance.store_tombstones:
         tombstones.ClearAllTombstones(dev)
 
     self._env.parallel_devices.pMap(
         individual_device_set_up,
@@ skipping 246 matching lines @@
         timeout = v
         break
     else:
       logging.warning('Using default 1 minute timeout for %s', test_name)
       timeout = 60
 
     timeout *= cls._GetTimeoutScaleFromAnnotations(annotations)
 
     return timeout