[chromedriver] Support clicking an element in sub frames.

chrome/test/chromedriver/web_view_impl.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/web_view_impl.h"
 
 #include "base/bind.h"
 #include "base/json/json_writer.h"
 #include "base/logging.h"
 #include "base/stringprintf.h"
 #include "base/values.h"
+#include "chrome/test/chromedriver/basic_types.h"
 #include "chrome/test/chromedriver/devtools_client_impl.h"
 #include "chrome/test/chromedriver/dom_tracker.h"
 #include "chrome/test/chromedriver/frame_tracker.h"
 #include "chrome/test/chromedriver/javascript_dialog_manager.h"
 #include "chrome/test/chromedriver/js.h"
 #include "chrome/test/chromedriver/navigation_tracker.h"
 #include "chrome/test/chromedriver/status.h"
 #include "chrome/test/chromedriver/ui_events.h"
 #include "chrome/test/chromedriver/web_view_delegate.h"
 
@@ skipping 53 matching lines @@
       return "keyUp";
     case kRawKeyDownEventType:
       return "rawKeyDown";
     case kCharEventType:
       return "char";
     default:
       return "";
   }
 }
 
+Status ParseSubFrame(const base::DictionaryValue* dict, Frame* frame) {
+  std::string id;
+  if (!dict->GetString("id", &id))
+    return Status(kUnknownError, "no id in frame object from DevTools");
+  std::string name;
+  dict->GetString("name", &name);
+  frame->id = id;
+  frame->name = name;
+  return Status(kOk);
+}
+
+Status SearchForFrame(const std::string& frame_id,
+                      const base::DictionaryValue* frame_resource_tree,
+                      std::list<Frame>* out_frames,
+                      bool* out_found) {
+  const base::DictionaryValue* current_frame_dict;
+  if (!frame_resource_tree->GetDictionary("frame", &current_frame_dict))
+    return Status(kUnknownError, "no frame in frame resource tree");
+  Frame current_frame;
+  Status status = ParseSubFrame(current_frame_dict, &current_frame);
+  if (status.IsError())
+    return status;
+  if (current_frame.id == frame_id) {
+    out_frames->push_back(current_frame);
+    *out_found = true;
+    return Status(kOk);
+  }
+
+  const base::ListValue* child_frames;
+  if (frame_resource_tree->GetList("childFrames", &child_frames)) {
+    for (size_t i = 0; i < child_frames->GetSize(); ++i) {
+      const base::DictionaryValue* sub_tree;
+      if (!child_frames->GetDictionary(i, &sub_tree))
+        return Status(kUnknownError, "frame should be a dictionary");
+      bool found = false;
+      status = SearchForFrame(frame_id, sub_tree, out_frames, &found);
+      if (status.IsError())
+        return status;
+      if (found) {
+        out_frames->back().index = i;
+        out_frames->back().parent_id = current_frame.id;
+        out_frames->push_back(current_frame);
+        *out_found = true;
+        break;
+      }
+    }
+  }
+  return Status(kOk);
+}
+
 }  // namespace
 
 WebViewImpl::WebViewImpl(const std::string& id,
                          DevToolsClient* client,
                          WebViewDelegate* delegate,
                          const CloserFunc& closer_func)
     : id_(id),
       dom_tracker_(new DomTracker(client)),
       frame_tracker_(new FrameTracker(client)),
       navigation_tracker_(new NavigationTracker(client)),
@@ skipping 173 matching lines @@
   scoped_ptr<base::DictionaryValue> result;
   Status status = client_->SendCommandAndGetResult(
       "Page.captureScreenshot", params, &result);
   if (status.IsError())
     return status;
   if (!result->GetString("data", screenshot))
     return Status(kUnknownError, "expected string 'data' in response");
   return Status(kOk);
 }
 
+Status WebViewImpl::GetFramePath(const std::string& frame_id,
+                                 std::list<Frame>* frames) {
+  base::DictionaryValue params;
+  scoped_ptr<base::DictionaryValue> result;
+  Status status = client_->SendCommandAndGetResult(
+      "Page.getResourceTree", params, &result);
+  if (status.IsError())
+    return status;
+  const base::DictionaryValue* root_frame_tree;
+  if (!result->GetDictionary("frameTree", &root_frame_tree))
+    return Status(kUnknownError, "no frame tree returned by DevTools");
+  bool found = false;
+  status = SearchForFrame(frame_id, root_frame_tree, frames, &found);
+  if (status.IsError())
+    return status;
+  if (!found)
+    return Status(kUnknownError, "frame not found in the frame tree");
+  return Status(kOk);
+}
+
 namespace internal {
 
 Status EvaluateScript(DevToolsClient* client,
                       int context_id,
                       const std::string& expression,
                       EvaluateScriptReturnType return_type,
                       scoped_ptr<base::DictionaryValue>* result) {
   base::DictionaryValue params;
   params.SetString("expression", expression);
   if (context_id)
@@ skipping 124 matching lines @@
   }
   if (status.IsError())
     return status;
 
   if (!cmd_result->GetInteger("nodeId", node_id))
     return Status(kUnknownError, "DOM.requestNode missing int 'nodeId'");
   return Status(kOk);
 }
 
 }  // namespace internal