Revert of [DevTools] Make DevTools clients talk directly to DevToolsAgentHost instead of using DevToolsManage…

content/browser/devtools/devtools_manager_unittest.cc

Index: content/browser/devtools/devtools_manager_unittest.cc
diff --git a/content/browser/devtools/devtools_manager_unittest.cc b/content/browser/devtools/devtools_manager_unittest.cc
index 588e5309bcfa0a49a1b0e57050a72684a9c0278d..79b76542e00dd6b5bd16e56728675badb8392c9d 100644
--- a/content/browser/devtools/devtools_manager_unittest.cc
+++ b/content/browser/devtools/devtools_manager_unittest.cc
@@ -10,6 +10,7 @@
 #include "content/common/view_messages.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/browser/devtools_agent_host.h"
+#include "content/public/browser/devtools_client_host.h"
 #include "content/public/browser/devtools_external_agent_proxy.h"
 #include "content/public/browser/devtools_external_agent_proxy_delegate.h"
 #include "content/public/browser/web_contents_delegate.h"
@@ -23,7 +24,7 @@
 namespace content {
 namespace {
 
-class TestDevToolsClientHost : public DevToolsAgentHostClient {
+class TestDevToolsClientHost : public DevToolsClientHost {
  public:
   TestDevToolsClientHost()
       : last_sent_message(NULL),
@@ -34,29 +35,23 @@
     EXPECT_TRUE(closed_);
   }
 
-  void Close() {
+  virtual void Close(DevToolsManager* manager) {
     EXPECT_FALSE(closed_);
     close_counter++;
-    agent_host_->DetachClient();
+    manager->ClientHostClosing(this);
     closed_ = true;
   }
-
-  virtual void AgentHostClosed(
-      DevToolsAgentHost* agent_host, bool replaced) OVERRIDE {
+  virtual void InspectedContentsClosing() OVERRIDE {
     FAIL();
   }
 
-  virtual void DispatchProtocolMessage(
-      DevToolsAgentHost* agent_host, const std::string& message) OVERRIDE {
+  virtual void DispatchOnInspectorFrontend(
+      const std::string& message) OVERRIDE {
     last_sent_message = &message;
   }
 
-  void InspectAgentHost(DevToolsAgentHost* agent_host) {
-    agent_host_ = agent_host;
-    agent_host_->AttachClient(this);
-  }
-
-  DevToolsAgentHost* agent_host() { return agent_host_.get(); }
+  virtual void ReplacedWithAnotherClient() OVERRIDE {
+  }
 
   static void ResetCounters() {
     close_counter = 0;
@@ -68,7 +63,6 @@
 
  private:
   bool closed_;
-  scoped_refptr<DevToolsAgentHost> agent_host_;
 
   DISALLOW_COPY_AND_ASSIGN(TestDevToolsClientHost);
 };
@@ -104,19 +98,40 @@
 };
 
 TEST_F(DevToolsManagerTest, OpenAndManuallyCloseDevToolsClientHost) {
+  DevToolsManager* manager = DevToolsManager::GetInstance();
+
   scoped_refptr<DevToolsAgentHost> agent(
       DevToolsAgentHost::GetOrCreateFor(web_contents()));
   EXPECT_FALSE(agent->IsAttached());
 
   TestDevToolsClientHost client_host;
-  client_host.InspectAgentHost(agent.get());
+  manager->RegisterDevToolsClientHostFor(agent.get(), &client_host);
   // Test that the connection is established.
   EXPECT_TRUE(agent->IsAttached());
+  EXPECT_EQ(agent, manager->GetDevToolsAgentHostFor(&client_host));
   EXPECT_EQ(0, TestDevToolsClientHost::close_counter);
 
-  client_host.Close();
+  client_host.Close(manager);
   EXPECT_EQ(1, TestDevToolsClientHost::close_counter);
   EXPECT_FALSE(agent->IsAttached());
+}
+
+TEST_F(DevToolsManagerTest, ForwardMessageToClient) {
+  DevToolsManagerImpl* manager = DevToolsManagerImpl::GetInstance();
+
+  TestDevToolsClientHost client_host;
+  scoped_refptr<DevToolsAgentHost> agent_host(
+      DevToolsAgentHost::GetOrCreateFor(web_contents()));
+  manager->RegisterDevToolsClientHostFor(agent_host.get(), &client_host);
+  EXPECT_EQ(0, TestDevToolsClientHost::close_counter);
+
+  std::string m = "test message";
+  agent_host = DevToolsAgentHost::GetOrCreateFor(web_contents());
+  manager->DispatchOnInspectorFrontend(agent_host.get(), m);
+  EXPECT_TRUE(&m == client_host.last_sent_message);
+
+  client_host.Close(manager);
+  EXPECT_EQ(1, TestDevToolsClientHost::close_counter);
 }
 
 TEST_F(DevToolsManagerTest, NoUnresponsiveDialogInInspectedContents) {
@@ -129,7 +144,8 @@
   TestDevToolsClientHost client_host;
   scoped_refptr<DevToolsAgentHost> agent_host(DevToolsAgentHost::GetOrCreateFor(
       WebContents::FromRenderViewHost(inspected_rvh)));
-  client_host.InspectAgentHost(agent_host.get());
+  DevToolsManager::GetInstance()->RegisterDevToolsClientHostFor(
+      agent_host.get(), &client_host);
 
   // Start with a short timeout.
   inspected_rvh->StartHangMonitorTimeout(TimeDelta::FromMilliseconds(10));
@@ -142,7 +158,7 @@
   EXPECT_FALSE(delegate.renderer_unresponsive_received());
 
   // Now close devtools and check that the notification is delivered.
-  client_host.Close();
+  client_host.Close(DevToolsManager::GetInstance());
   // Start with a short timeout.
   inspected_rvh->StartHangMonitorTimeout(TimeDelta::FromMilliseconds(10));
   // Wait long enough for first timeout and see if it fired.
@@ -165,25 +181,26 @@
   EXPECT_FALSE(contents()->cross_navigation_pending());
 
   TestDevToolsClientHost client_host;
-  client_host.InspectAgentHost(
-      DevToolsAgentHost::GetOrCreateFor(web_contents()).get());
+  DevToolsManager* devtools_manager = DevToolsManager::GetInstance();
+  devtools_manager->RegisterDevToolsClientHostFor(
+      DevToolsAgentHost::GetOrCreateFor(web_contents()).get(), &client_host);
 
   // Navigate to new site which should get a new RenderViewHost.
   const GURL url2("http://www.yahoo.com");
   controller().LoadURL(
       url2, Referrer(), PAGE_TRANSITION_TYPED, std::string());
   EXPECT_TRUE(contents()->cross_navigation_pending());
-  EXPECT_EQ(client_host.agent_host(),
-      DevToolsAgentHost::GetOrCreateFor(web_contents()));
+  EXPECT_EQ(devtools_manager->GetDevToolsAgentHostFor(&client_host),
+            DevToolsAgentHost::GetOrCreateFor(web_contents()));
 
   // Interrupt pending navigation and navigate back to the original site.
   controller().LoadURL(
       url, Referrer(), PAGE_TRANSITION_TYPED, std::string());
   contents()->TestDidNavigate(rvh(), 1, url, PAGE_TRANSITION_TYPED);
   EXPECT_FALSE(contents()->cross_navigation_pending());
-  EXPECT_EQ(client_host.agent_host(),
+  EXPECT_EQ(devtools_manager->GetDevToolsAgentHostFor(&client_host),
             DevToolsAgentHost::GetOrCreateFor(web_contents()));
-  client_host.Close();
+  client_host.Close(DevToolsManager::GetInstance());
 }
 
 class TestExternalAgentDelegate: public DevToolsExternalAgentProxyDelegate {
@@ -228,13 +245,16 @@
       DevToolsAgentHost::Create(delegate);
   EXPECT_EQ(agent_host, DevToolsAgentHost::GetForId(agent_host->GetId()));
 
-  TestDevToolsClientHost client_host;
-  client_host.InspectAgentHost(agent_host.get());
-  agent_host->DispatchProtocolMessage("message1");
-  agent_host->DispatchProtocolMessage("message2");
-  agent_host->DispatchProtocolMessage("message2");
-
-  client_host.Close();
+  DevToolsManager* manager = DevToolsManager::GetInstance();
+
+  TestDevToolsClientHost client_host;
+  manager->RegisterDevToolsClientHostFor(agent_host.get(), &client_host);
+
+  manager->DispatchOnInspectorBackend(&client_host, "message1");
+  manager->DispatchOnInspectorBackend(&client_host, "message2");
+  manager->DispatchOnInspectorBackend(&client_host, "message2");
+
+  client_host.Close(manager);
 }
 
 }  // namespace content