[chromedriver] Support clicking an element in sub frames.

chrome/test/chromedriver/window_commands.cc

 // Copyright (c) 2013 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/chromedriver/window_commands.h"
 
 #include <list>
 
 #include "base/callback.h"
 #include "base/string_number_conversions.h"
 #include "base/stringprintf.h"
 #include "base/values.h"
 #include "chrome/test/chromedriver/basic_types.h"
 #include "chrome/test/chromedriver/chrome.h"
 #include "chrome/test/chromedriver/element_util.h"
 #include "chrome/test/chromedriver/js.h"
 #include "chrome/test/chromedriver/session.h"
 #include "chrome/test/chromedriver/status.h"
 #include "chrome/test/chromedriver/ui_events.h"
+#include "chrome/test/chromedriver/util.h"
 #include "chrome/test/chromedriver/web_view.h"
 
 namespace {
 
 Status GetMouseButton(const base::DictionaryValue& params,
                       MouseButton* button) {
   int button_num;
   if (!params.GetInteger("button", &button_num)) {
     button_num = 0;  // Default to left mouse button.
   } else if (button_num < 0 || button_num > 2) {
@@ skipping 107 matching lines @@
   status = web_view->ConnectIfNecessary();
   if (status.IsError())
     return status;
 
   Status nav_status = web_view->WaitForPendingNavigations(session->frame);
   if (nav_status.IsError())
     return nav_status;
   status = command.Run(session, web_view, params, value);
   // Switch to main frame and retry command if subframe no longer exists.
   if (status.code() == kNoSuchFrame) {
-    session->frame = "";
+    session->SetToFrame("", "");
     nav_status = web_view->WaitForPendingNavigations(session->frame);
     if (nav_status.IsError())
       return nav_status;
     status = command.Run(session, web_view, params, value);
   }
   nav_status = web_view->WaitForPendingNavigations(session->frame);
   if (status.IsOk() && nav_status.IsError() &&
       nav_status.code() != kDisconnected)
     return nav_status;
   return status;
@@ skipping 29 matching lines @@
 Status ExecuteSwitchToFrame(
     Session* session,
     WebView* web_view,
     const base::DictionaryValue& params,
     scoped_ptr<base::Value>* value) {
   const base::Value* id;
   if (!params.Get("id", &id))
     return Status(kUnknownError, "missing 'id'");
 
   if (id->IsType(base::Value::TYPE_NULL)) {
-    session->frame = "";
+    session->SetToFrame("", "");
     return Status(kOk);
   }
 
   std::string script;
   base::ListValue args;
   const base::DictionaryValue* id_dict;
   if (id->GetAsDictionary(&id_dict)) {
     script = "function(elem) { return elem; }";
     args.Append(id_dict->DeepCopy());
   } else {
@@ skipping 13 matching lines @@
     } else {
       return Status(kUnknownError, "invalid 'id'");
     }
     args.Append(new base::StringValue(xpath));
   }
   std::string frame;
   Status status = web_view->GetFrameByFunction(
       session->frame, script, args, &frame);
   if (status.IsError())
     return status;
-  session->frame = frame;
+
+  scoped_ptr<base::Value> result;
+  status = web_view->CallFunction(
+      session->frame, script, args, &result);
+  if (status.IsError())
+    return status;
+  const base::DictionaryValue* element;
+  if (!result->GetAsDictionary(&element))
+    return Status(kUnknownError, "fail to locate the sub frame element");
+
+  std::string fake_id = GenerateId();
+  const char* kSetFrameIdentifier =
+      "function(frame, id) {"
+      "  frame.setAttribute('cd_frame_id_', id);"
+      "}";
+  base::ListValue new_args;
+  new_args.Append(element->DeepCopy());
+  new_args.AppendString(fake_id);
+  result.reset(NULL);
+  status = web_view->CallFunction(
+      session->frame, kSetFrameIdentifier, new_args, &result);
+  if (status.IsError())
+    return status;
+  session->SetToFrame(
+      frame, base::StringPrintf("//*[@cd_frame_id_ = '%s']", fake_id.c_str()));
   return Status(kOk);
 }
 
 Status ExecuteGetTitle(
     Session* session,
     WebView* web_view,
     const base::DictionaryValue& params,
     scoped_ptr<base::Value>* value) {
   const char* kGetTitleScript =
       "function() {"
@@ skipping 93 matching lines @@
   if (has_element) {
     Status status = ScrollElementIntoView(
         session, web_view, element_id, &location);
     if (status.IsError())
       return status;
   } else {
     location = session->mouse_position;
   }
 
   if (has_offset) {
-    location.offset(x_offset, y_offset);
+    location.Offset(x_offset, y_offset);
   } else {
     WebSize size;
     Status status = GetElementSize(session, web_view, element_id, &size);
     if (status.IsError())
       return status;
-    location.offset(size.width / 2, size.height / 2);
+    location.Offset(size.width / 2, size.height / 2);
   }
 
   std::list<MouseEvent> events;
   events.push_back(
       MouseEvent(kMovedMouseEventType, kNoneMouseButton,
                  location.x, location.y, 0));
   Status status = web_view->DispatchMouseEvents(events);
   if (status.IsOk())
     session->mouse_position = location;
   return status;
@@ skipping 287 matching lines @@
     for (std::list<Cookie>::const_iterator it = cookies.begin();
          it != cookies.end(); ++it) {
       status = web_view->DeleteCookie(it->name, url);
       if (status.IsError())
         return status;
     }
   }
 
   return Status(kOk);
 }