[ChromeDriver] Send DOM.getDocument after each DOM.documentUpdated.

chrome/test/chromedriver/chrome_impl.cc

 // Copyright (c) 2012 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/chrome_impl.h"
 
 #include "base/bind.h"
 #include "base/json/json_reader.h"
 #include "base/json/json_writer.h"
 #include "base/logging.h"
 #include "base/process_util.h"
 #include "base/stringprintf.h"
 #include "base/threading/platform_thread.h"
 #include "base/time.h"
 #include "base/values.h"
 #include "chrome/test/chromedriver/devtools_client_impl.h"
+#include "chrome/test/chromedriver/devtools_state.h"
 #include "chrome/test/chromedriver/dom_tracker.h"
 #include "chrome/test/chromedriver/frame_tracker.h"
 #include "chrome/test/chromedriver/js.h"
 #include "chrome/test/chromedriver/navigation_tracker.h"
 #include "chrome/test/chromedriver/net/net_util.h"
 #include "chrome/test/chromedriver/net/sync_websocket_impl.h"
 #include "chrome/test/chromedriver/net/url_request_context_getter.h"
 #include "chrome/test/chromedriver/status.h"
 #include "googleurl/src/gurl.h"
 
@@ skipping 86 matching lines @@
     FetchPagesInfo(context_getter_, port_, &debugger_urls);
     if (debugger_urls.empty())
       base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(100));
     else
       break;
   }
   if (debugger_urls.empty())
     return Status(kUnknownError, "unable to discover open pages");
   client_.reset(new DevToolsClientImpl(
       socket_factory_, debugger_urls.front()));
+  client_state_.reset(new DevToolsState(client_.get()));
+  Status status = client_state_->Init();
+  if (status.IsError())
+    return status;
   client_->AddListener(dom_tracker_.get());
   client_->AddListener(frame_tracker_.get());
   client_->AddListener(navigation_tracker_.get());
-
-  // Perform necessary configuration of the DevTools client.
-  // Fetch the root document node so that Inspector will push DOM node
-  // information to the client.
-  base::DictionaryValue params;
-  Status status = client_->SendCommand("DOM.getDocument", params);
-  if (status.IsError())
-    return status;
-  // Enable page domain notifications to allow tracking navigation state.
-  status = client_->SendCommand("Page.enable", params);
-  if (status.IsError())
-    return status;
-  // Enable runtime events to allow tracking execution context creation.
-  return client_->SendCommand("Runtime.enable", params);
+  client_->AddListener(client_state_.get());
+  return Status(kOk);
 }
 
 Status ChromeImpl::Load(const std::string& url) {
   base::DictionaryValue params;
   params.SetString("url", url);
   return client_->SendCommand("Page.navigate", params);
 }
 
 Status ChromeImpl::Reload() {
   base::DictionaryValue params;
@@ skipping 245 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