Toggle a multiple select option instead of setting it as selected when

chrome/test/webdriver/session.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/session.h"
 
 #include <sstream>
 #include <vector>
 
 #include "base/command_line.h"
@@ skipping 900 matching lines @@
   Error* error = ExecuteScript(frame_id, script, &args, &unscoped_result);
   scoped_ptr<Value> result(unscoped_result);
   if (error)
     return error;
   if (!result->GetAsBoolean(is_enabled))
     return new Error(kUnknownError, "IsEnabled atom returned non-boolean: " +
                          JsonStringify(result.get()));
   return NULL;
 }
 
-Error* Session::SelectOptionElement(const FrameId& frame_id,
-                                    const WebElementId& element) {
+Error* Session::IsOptionElementSelected(const FrameId& frame_id,
+                                        const WebElementId& element,
+                                        bool* is_selected) {
   ListValue args;
   args.Append(element.ToValue());
-  args.Append(Value::CreateBooleanValue(true));
+
+  std::string script = base::StringPrintf(
+      "return (%s).apply(null, arguments);", atoms::IS_SELECTED);
+
+  Value* result = NULL;
+  Error* error = ExecuteScript(frame_id, script, &args, &result);
+  if (error)
+    return error;
+  scoped_ptr<Value> scoped_result(result);
+  if (!result->GetAsBoolean(is_selected)) {
+    return new Error(kUnknownError, "isSelected atom returned non-boolean: " +
+                         JsonStringify(result));
+  }
+  return NULL;
+}
+
+Error* Session::SetOptionElementSelected(const FrameId& frame_id,
+                                         const WebElementId& element,
+                                         bool selected) {
+  ListValue args;
+  args.Append(element.ToValue());
+  args.Append(Value::CreateBooleanValue(selected));
 
   std::string script = base::StringPrintf(
       "return (%s).apply(null, arguments);", atoms::SET_SELECTED);
 
   Value* unscoped_result = NULL;
   Error* error = ExecuteScript(frame_id, script, &args, &unscoped_result);
   scoped_ptr<Value> result(unscoped_result);
   return error;
 }
 
+Error* Session::ToggleOptionElement(const FrameId& frame_id,
+                                    const WebElementId& element) {
+  bool is_selected;
+  Error* error = IsOptionElementSelected(frame_id, element, &is_selected);
+  if (error)
+    return error;
+
+  return SetOptionElementSelected(frame_id, element, !is_selected);
+}
+
 Error* Session::GetElementTagName(const FrameId& frame_id,
                                   const WebElementId& element,
                                   std::string* tag_name) {
   ListValue args;
   args.Append(element.ToValue());
 
   std::string script = "return arguments[0].tagName.toLocaleLowerCase();";
 
   Value* unscoped_result = NULL;
   Error* error = ExecuteScript(frame_id, script, &args, &unscoped_result);
@@ skipping 496 matching lines @@
       path,
       &error));
   if (error)
     return error;
   if (!file_util::ReadFileToString(path, png))
     return new Error(kUnknownError, "Could not read screenshot file");
   return NULL;
 }
 
 }  // namespace webdriver