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

chrome/test/chromedriver/devtools_client_impl.h

Index: chrome/test/chromedriver/devtools_client_impl.h
diff --git a/chrome/test/chromedriver/devtools_client_impl.h b/chrome/test/chromedriver/devtools_client_impl.h
index 2b2a83122be7f8ed521ed3b826266d578092b9c3..4d03e78f2e7225fef3358c4936d081403ad9c0b5 100644
--- a/chrome/test/chromedriver/devtools_client_impl.h
+++ b/chrome/test/chromedriver/devtools_client_impl.h
@@ -6,6 +6,7 @@
 #define CHROME_TEST_CHROMEDRIVER_DEVTOOLS_CLIENT_IMPL_H_
 
 #include <list>
+#include <map>
 #include <string>
 
 #include "base/basictypes.h"
@@ -50,7 +51,6 @@ class SyncWebSocket;
 
 class DevToolsClientImpl : public DevToolsClient {
  public:
-  // Listener may be NULL.
   DevToolsClientImpl(const SyncWebSocketFactory& factory,
                      const std::string& url);
 
@@ -66,6 +66,8 @@ class DevToolsClientImpl : public DevToolsClient {
 
   virtual ~DevToolsClientImpl();
 
+  void SetParserFuncForTesting(const ParserFunc& parser_func);
+
   // Overridden from DevToolsClient:
   virtual Status SendCommand(const std::string& method,
                              const base::DictionaryValue& params) OVERRIDE;
@@ -96,6 +98,8 @@ class DevToolsClientImpl : public DevToolsClient {
   GURL url_;
   ParserFunc parser_func_;
   std::list<DevToolsEventListener*> listeners_;
+  typedef std::map<int, base::DictionaryValue*> ResponseMap;
+  ResponseMap cmd_response_map_;
   bool connected_;
   int next_id_;