[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"
@@ skipping 76 matching lines @@
 
 ChromeImpl::ChromeImpl(base::ProcessHandle process,
                        URLRequestContextGetter* context_getter,
                        base::ScopedTempDir* user_data_dir,
                        int port,
                        const SyncWebSocketFactory& socket_factory)
     : process_(process),
       context_getter_(context_getter),
       port_(port),
       socket_factory_(socket_factory),
-      dom_tracker_(new DomTracker()),
       frame_tracker_(new FrameTracker()),
       navigation_tracker_(new NavigationTracker()) {
   if (user_data_dir->IsValid()) {
     CHECK(user_data_dir_.Set(user_data_dir->Take()));
   }
 }
 
 ChromeImpl::~ChromeImpl() {
   base::CloseProcessHandle(process_);
 }
 
 Status ChromeImpl::Init() {
   base::Time deadline = base::Time::Now() + base::TimeDelta::FromSeconds(20);
   std::list<std::string> debugger_urls;
   while (base::Time::Now() < deadline) {
     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()));
+  dom_tracker_.reset(new DomTracker(client_.get()));
+  Status status = dom_tracker_->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);
+  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