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

chrome/test/chromedriver/devtools_state.h

Index: chrome/test/chromedriver/devtools_state.h
diff --git a/chrome/test/chromedriver/dom_tracker.h b/chrome/test/chromedriver/devtools_state.h
similarity index 56%
copy from chrome/test/chromedriver/dom_tracker.h
copy to chrome/test/chromedriver/devtools_state.h
index 3de98853f9d1b8420f0d9d6c5de7c63b9af17014..e76bd0ad90d3a21974b8300fd6e1c8ee862267f8 100644
--- a/chrome/test/chromedriver/dom_tracker.h
+++ b/chrome/test/chromedriver/devtools_state.h
@@ -2,8 +2,8 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-#ifndef CHROME_TEST_CHROMEDRIVER_DOM_TRACKER_H_
-#define CHROME_TEST_CHROMEDRIVER_DOM_TRACKER_H_
+#ifndef CHROME_TEST_CHROMEDRIVER_DEVTOOLS_STATE_H_
+#define CHROME_TEST_CHROMEDRIVER_DEVTOOLS_STATE_H_
 
 #include <map>
 #include <string>
@@ -14,30 +14,26 @@
 
 namespace base {
 class DictionaryValue;
-class Value;
 }
 
+class DevToolsClient;
 class Status;
 
 // Tracks the state of the DOM and execution context creation.
-class DomTracker : public DevToolsEventListener {
+class DevToolsState : public DevToolsEventListener {
  public:
-  DomTracker();
-  virtual ~DomTracker();
+  explicit DevToolsState(DevToolsClient* client);
+  virtual ~DevToolsState();
 
-  Status GetFrameIdForNode(int node_id, std::string* frame_id);
+  Status Init();
 
   // Overridden from DevToolsEventListener:
   virtual void OnEvent(const std::string& method,
                        const base::DictionaryValue& params) OVERRIDE;
 
  private:
-  bool ProcessNodeList(const base::Value* nodes);
-  bool ProcessNode(const base::Value* node);
-
-  std::map<int, std::string> node_to_frame_map_;
-
-  DISALLOW_COPY_AND_ASSIGN(DomTracker);
+  DevToolsClient* client_;
+  DISALLOW_COPY_AND_ASSIGN(DevToolsState);
 };
 
-#endif  // CHROME_TEST_CHROMEDRIVER_DOM_TRACKER_H_
+#endif  // CHROME_TEST_CHROMEDRIVER_DEVTOOLS_STATE_H_