[DevTools] New harness for inspector-protocol layout tests

content/shell/browser/layout_test/devtools_protocol_test_bindings.cc

Index: content/shell/browser/layout_test/devtools_protocol_test_bindings.cc
diff --git a/content/shell/browser/layout_test/devtools_protocol_test_bindings.cc b/content/shell/browser/layout_test/devtools_protocol_test_bindings.cc
new file mode 100644
index 0000000000000000000000000000000000000000..5b8ae60ac53c088803bac21a91baa7f02756f8ff
--- /dev/null
+++ b/content/shell/browser/layout_test/devtools_protocol_test_bindings.cc
@@ -0,0 +1,141 @@
+// Copyright 2017 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 "content/shell/browser/layout_test/devtools_protocol_test_bindings.h"
+
+#include "base/json/json_reader.h"
+#include "base/json/string_escape.h"
+#include "base/strings/string_number_conversions.h"
+#include "base/strings/utf_string_conversions.h"
+#include "base/values.h"
+#include "content/public/browser/devtools_agent_host.h"
+#include "content/public/browser/navigation_handle.h"
+#include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/web_contents.h"
+
+#if !defined(OS_ANDROID)
+#include "content/public/browser/devtools_frontend_host.h"
+#endif
+
+namespace content {
+
+namespace {
+// This constant should be in sync with
+// the constant at devtools_ui_bindings.cc.
+const size_t kMaxMessageChunkSize = IPC::Channel::kMaximumMessageSize / 4;
+}
+
+DevToolsProtocolTestBindings::DevToolsProtocolTestBindings(
+    WebContents* devtools)
+    : WebContentsObserver(devtools),
+      agent_host_(DevToolsAgentHost::CreateForDiscovery()) {
+#if !defined(OS_ANDROID)
+  frontend_host_.reset(DevToolsFrontendHost::Create(
+      devtools->GetMainFrame(),
+      base::Bind(&DevToolsProtocolTestBindings::HandleMessageFromTest,
+                 base::Unretained(this))));
+#endif
+  agent_host_->AttachClient(this);
+}
+
+DevToolsProtocolTestBindings::~DevToolsProtocolTestBindings() {
+  if (agent_host_)
+    agent_host_->DetachClient(this);
+}
+
+// static
+GURL DevToolsProtocolTestBindings::GetProtocolTestURL(const GURL& test_url) {
+  std::string spec = test_url.spec();
+  std::string dir = "/inspector-protocol-2/";

[chenwilliam, 2017/06/14 22:06:41]

let's make this directory name more descriptive? (e.g.
/inspector-protocol-multi-client/)

[dgozman, 2017/06/19 17:48:59]

This was a temporary name :)

+  size_t pos = spec.find(dir);
+  if (pos == std::string::npos)
+    return GURL();
+  spec = spec.substr(0, pos + dir.length()) +
+         "resources/inspector-protocol-test.html?" + spec;
+  return GURL(spec);
+}
+
+void DevToolsProtocolTestBindings::ReadyToCommitNavigation(
+    NavigationHandle* navigation_handle) {
+#if !defined(OS_ANDROID)
+  content::RenderFrameHost* frame = navigation_handle->GetRenderFrameHost();
+  if (frame->GetParent())
+    return;
+  frontend_host_.reset(DevToolsFrontendHost::Create(
+      frame, base::Bind(&DevToolsProtocolTestBindings::HandleMessageFromTest,
+                        base::Unretained(this))));
+#endif
+}
+
+void DevToolsProtocolTestBindings::WebContentsDestroyed() {
+  if (agent_host_) {
+    agent_host_->DetachClient(this);
+    agent_host_ = nullptr;
+  }
+}
+
+void DevToolsProtocolTestBindings::HandleMessageFromTest(
+    const std::string& message) {
+  if (!agent_host_)
+    return;
+  std::string method;
+  base::ListValue* params = NULL;
+  base::DictionaryValue* dict = NULL;

[chenwilliam, 2017/06/14 22:06:41]

nullptr? :)

[dgozman, 2017/06/19 17:48:59]

Done.

+  std::unique_ptr<base::Value> parsed_message = base::JSONReader::Read(message);
+  if (!parsed_message || !parsed_message->GetAsDictionary(&dict) ||
+      !dict->GetString("method", &method)) {
+    return;
+  }
+
+  int request_id = 0;
+  dict->GetInteger("id", &request_id);
+  dict->GetList("params", &params);
+
+  if (method == "dispatchProtocolMessage" && params && params->GetSize() == 1) {
+    if (!agent_host_ || !agent_host_->IsAttached())
+      return;
+    std::string protocol_message;
+    if (!params->GetString(0, &protocol_message))
+      return;
+    agent_host_->DispatchProtocolMessage(this, protocol_message);
+    if (request_id) {
+      std::string code = "DevToolsAPI.embedderMessageAck(" +
+                         base::IntToString(request_id) + ");";
+      base::string16 javascript = base::UTF8ToUTF16(code);
+      web_contents()->GetMainFrame()->ExecuteJavaScriptForTests(javascript);
+    }
+  }
+}
+
+void DevToolsProtocolTestBindings::DispatchProtocolMessage(
+    DevToolsAgentHost* agent_host,
+    const std::string& message) {
+  if (message.length() < kMaxMessageChunkSize) {
+    std::string param;
+    base::EscapeJSONString(message, true, &param);
+    std::string code = "DevToolsAPI.dispatchMessage(" + param + ");";
+    base::string16 javascript = base::UTF8ToUTF16(code);
+    web_contents()->GetMainFrame()->ExecuteJavaScriptForTests(javascript);
+    return;
+  }
+
+  size_t total_size = message.length();
+  for (size_t pos = 0; pos < message.length(); pos += kMaxMessageChunkSize) {
+    std::string param;
+    base::EscapeJSONString(message.substr(pos, kMaxMessageChunkSize), true,
+                           &param);
+    std::string code = "DevToolsAPI.dispatchMessageChunk(" + param + "," +
+                       std::to_string(pos ? 0 : total_size) + ");";
+    base::string16 javascript = base::UTF8ToUTF16(code);
+    web_contents()->GetMainFrame()->ExecuteJavaScriptForTests(javascript);
+  }
+}
+
+void DevToolsProtocolTestBindings::AgentHostClosed(
+    DevToolsAgentHost* agent_host,
+    bool replaced) {
+  agent_host_ = nullptr;
+}
+
+}  // namespace content