Allow webdriver users to choose between sending the key events 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_manager.h"
 
 #include <sstream>
 #include <vector>
 
 #include "base/command_line.h"
@@ skipping 41 matching lines @@
 FrameId& FrameId::operator=(const FrameId& other) {
   window_id = other.window_id;
   frame_path = other.frame_path;
   return *this;
 }
 
 Session::Session()
     : id_(GenerateRandomID()),
       thread_(id_.c_str()),
       implicit_wait_(0),
-      current_target_(FrameId(0, FramePath())) {
+      current_target_(FrameId(0, FramePath())),
+      use_native_events_(false) {
   SessionManager::GetInstance()->Add(this);
 }
 
 Session::~Session() {
   SessionManager::GetInstance()->Remove(id_);
 }
 
 bool Session::Init(const FilePath& browser_dir) {
   bool success = false;
   if (thread_.Start()) {
@@ skipping 93 matching lines @@
   return static_cast<ErrorCode>(status);
 }
 
 ErrorCode Session::ExecuteScript(const std::string& script,
                                  const ListValue* const args,
                                  Value** value) {
   return ExecuteScript(current_target_, script, args, value);
 }
 
 ErrorCode Session::SendKeys(const WebElementId& element, const string16& keys) {
+  // This method will first check if the element we want to send the keys to is
+  // already focused, if not it will try to focus on it first.
   ListValue args;
   args.Append(element.ToValue());
   // TODO(jleyba): Update this to use the correct atom.
-  std::string script = "document.activeElement.blur();arguments[0].focus();";
+  std::string script = "if(document.activeElement!=arguments[0]){"
+                       "  if(document.activeElement)"
+                       "    document.activeElement.blur();"
+                       "  arguments[0].focus();"
+                       "}";
   Value* unscoped_result = NULL;
   ErrorCode code = ExecuteScript(script, &args, &unscoped_result);
-  scoped_ptr<Value> result(unscoped_result);
   if (code != kSuccess) {
-    LOG(ERROR) << "Failed to focus element before sending keys";
+    LOG(ERROR) << "Failed to get or set focus element before sending keys";
     return code;
   }
-
   bool success = false;
   RunSessionTask(NewRunnableMethod(
       this,
       &Session::SendKeysOnSessionThread,
       keys,
       &success));
   if (!success)
     return kUnknownError;
   return kSuccess;
 }
@@ skipping 509 matching lines @@
   automation_.reset();
 }
 
 void Session::SendKeysOnSessionThread(const string16& keys,
                                       bool* success) {
   *success = true;
   std::vector<WebKeyEvent> key_events;
   ConvertKeysToWebKeyEvents(keys, &key_events);
   for (size_t i = 0; i < key_events.size(); ++i) {
     bool key_success = false;
-    automation_->SendWebKeyEvent(
-        current_target_.window_id, key_events[i], &key_success);
+    if (use_native_events_) {
+      // The automation provider will generate up/down events for us, we
+      // only need to call it once as compared to the WebKeyEvent method.
+      // Hence we filter events by their types, keeping only rawkeydown.
+      if (key_events[i].type != automation::kRawKeyDownType)
+        continue;
+      automation_->SendNativeKeyEvent(
+          current_target_.window_id,
+          key_events[i].key_code,
+          key_events[i].modifiers,
+          &key_success);
+    } else {
+      automation_->SendWebKeyEvent(
+          current_target_.window_id, key_events[i], &key_success);
+    }
     if (!key_success) {
       LOG(ERROR) << "Failed to send key event. Event details:\n"
                  << "Type: " << key_events[i].type << "\n"
                  << "KeyCode: " << key_events[i].key_code << "\n"
                  << "UnmodifiedText: " << key_events[i].unmodified_text << "\n"
                  << "ModifiedText: " << key_events[i].modified_text << "\n"
                  << "Modifiers: " << key_events[i].modifiers << "\n";
       *success = false;
     }
   }
@@ skipping 133 matching lines @@
   if (!loc_dict->GetInteger("x", &x) ||
       !loc_dict->GetInteger("y", &y)) {
     LOG(ERROR) << "Location atom returned bad coordinate dictionary";
     code = kUnknownError;
   }
   *location = gfx::Point(x, y);
   return kSuccess;
 }
 
 }  // namespace webdriver