Allow chromedriver to set local state preferences.

chrome/test/webdriver/webdriver_automation.cc

 // Copyright (c) 2011 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.
 
 #include "chrome/test/webdriver/webdriver_automation.h"
 
 #if defined(OS_WIN)
 #include <windows.h>
 #endif
 
@@ skipping 167 matching lines @@
   return CheckForChromeExe(browser_exes, locations, browser_exe) ||
       CheckForChromeExe(browser_exes, chromium_locations, browser_exe);
 }
 
 }  // namespace
 
 namespace webdriver {
 
 Automation::BrowserOptions::BrowserOptions()
     : command(CommandLine::NO_PROGRAM),
-      detach_process(false) {}
+      detach_process(false),
+      cert_revocation_checking(true) {}
 
 Automation::BrowserOptions::~BrowserOptions() {}
 
 Automation::Automation() {}
 
 Automation::~Automation() {}
 
 void Automation::Init(const BrowserOptions& options, Error** error) {
   // Prepare Chrome's command line.
   CommandLine command(CommandLine::NO_PROGRAM);
@@ skipping 537 matching lines @@
   *error = CheckNewExtensionInterfaceSupported();
   if (*error)
     return;
 
   std::string error_msg;
   if (!SendInstallExtensionJSONRequest(
           automation(), path, false /* with_ui */, extension_id, &error_msg))
     *error = new Error(kUnknownError, error_msg);
 }
 
+void Automation::SetLocalStatePreference(const std::string& pref,
+                                         base::Value* value,
+                                         Error** error) {
+  scoped_ptr<Value> scoped_value(value);
+  bool has_new_local_state_api = false;
+  *error = CompareVersion(927, 0, &has_new_local_state_api);

[Huyen, 2011/12/09 23:56:55]

maybe move the version # out into a constant and/or add comment stating this
version supports new local state api

[kkania, 2011/12/10 00:37:44]

Done.

+  if (*error)
+    return;
+
+  if (has_new_local_state_api) {
+    std::string error_msg;
+    if (!SendSetLocalStatePreferenceJSONRequest(
+            automation(), pref, scoped_value.release(), &error_msg))
+      *error = new Error(kUnknownError, error_msg);
+  } else {
+    std::string request, response;
+    DictionaryValue request_dict;
+    request_dict.SetString("command", "SetLocalStatePrefs");
+    request_dict.SetString("path", pref);
+    request_dict.Set("value", scoped_value.release());
+    base::JSONWriter::Write(&request_dict, false, &request);
+    if (!automation()->GetBrowserWindow(0)->SendJSONRequest(
+            request, -1, &response)) {
+      *error = new Error(kUnknownError, base::StringPrintf(
+          "Failed to set local state pref '%s'", pref.c_str()));
+    }
+  }
+}
+
+void Automation::SetPreference(const std::string& pref,
+                               base::Value* value,
+                               Error** error) {
+  scoped_ptr<Value> scoped_value(value);
+  bool has_new_pref_api = false;
+  *error = CompareVersion(964, 0, &has_new_pref_api);

[Huyen, 2011/12/09 23:56:55]

same comment as above

[kkania, 2011/12/10 00:37:44]

Done.

+  if (*error)
+    return;
+
+  if (has_new_pref_api) {
+    std::string error_msg;
+    if (!SendSetPreferenceJSONRequest(
+            automation(), pref, scoped_value.release(), &error_msg))
+      *error = new Error(kUnknownError, error_msg);
+  } else {
+    std::string request, response;
+    DictionaryValue request_dict;
+    request_dict.SetString("command", "SetPrefs");
+    request_dict.SetString("path", pref);
+    request_dict.Set("value", scoped_value.release());
+    base::JSONWriter::Write(&request_dict, false, &request);
+    if (!automation()->GetBrowserWindow(0)->SendJSONRequest(
+            request, -1, &response)) {
+      *error = new Error(kUnknownError, base::StringPrintf(
+          "Failed to set pref '%s'", pref.c_str()));
+    }
+  }
+}
+
 AutomationProxy* Automation::automation() const {
   return launcher_->automation();
 }
 
 Error* Automation::GetIndicesForTab(
     int tab_id, int* browser_index, int* tab_index) {
   std::string error_msg;
   if (!SendGetIndicesFromTabIdJSONRequest(
           automation(), tab_id, browser_index, tab_index, &error_msg)) {
     return new Error(kUnknownError, error_msg);
@@ skipping 50 matching lines @@
   return CheckVersion(750, 0, message);
 }
 
 Error* Automation::CheckNewExtensionInterfaceSupported() {
   const char* message =
       "Extension interface is not supported for this version of Chrome";
   return CheckVersion(947, 0, message);
 }
 
 }  // namespace webdriver