Emulation.setVirtualTimePolicy to have an optional virtual time budge

content/browser/devtools/protocol/devtools_protocol_browsertest.cc

 // Copyright 2013 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 <stddef.h>
 #include <utility>
 
 #include "base/base64.h"
 #include "base/command_line.h"
 #include "base/json/json_reader.h"
 #include "base/json/json_writer.h"
 #include "base/memory/ptr_util.h"
 #include "base/run_loop.h"
+#include "base/strings/utf_string_conversions.h"
 #include "base/values.h"
 #include "build/build_config.h"
 #include "content/public/browser/devtools_agent_host.h"
 #include "content/public/browser/javascript_dialog_manager.h"
 #include "content/public/browser/navigation_handle.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/url_constants.h"
 #include "content/public/test/browser_test_utils.h"
@@ skipping 72 matching lines @@
 
  private:
   DialogClosedCallback callback_;
   bool handle_;
   DISALLOW_COPY_AND_ASSIGN(TestJavaScriptDialogManager);
 };
 
 }
 
 class DevToolsProtocolTest : public ContentBrowserTest,
-                             public DevToolsAgentHostClient {
+                             public DevToolsAgentHostClient,
+                             public WebContentsDelegate {
  public:
   DevToolsProtocolTest()
       : last_sent_id_(0),
         waiting_for_command_result_id_(0),
         in_dispatch_(false) {
   }
 
  protected:
+  // WebContentsDelegate method:
+  bool AddMessageToConsole(WebContents* source,
+                           int32_t level,
+                           const base::string16& message,
+                           int32_t line_no,
+                           const base::string16& source_id) override {
+    console_messages_.push_back(base::UTF16ToUTF8(message));
+    return true;
+  }
+
   void SendCommand(const std::string& method,
                    std::unique_ptr<base::DictionaryValue> params) {
     SendCommand(method, std::move(params), true);
   }
 
   void SendCommand(const std::string& method,
                    std::unique_ptr<base::DictionaryValue> params,
                    bool wait) {
     in_dispatch_ = true;
     base::DictionaryValue command;
@@ skipping 35 matching lines @@
         return false;
       if (id == value)
         return true;
     }
     return false;
   }
 
   void Attach() {
     agent_host_ = DevToolsAgentHost::GetOrCreateFor(shell()->web_contents());
     agent_host_->AttachClient(this);
+    shell()->web_contents()->SetDelegate(this);
   }
 
   void TearDownOnMainThread() override {
     if (agent_host_) {
       agent_host_->DetachClient(this);
       agent_host_ = nullptr;
     }
   }
 
   void WaitForNotification(const std::string& notification) {
@@ skipping 70 matching lines @@
       urls.push_back(RemovePort(render_frame_host->GetLastCommittedURL()));
     }
     return urls;
   }
 
   std::unique_ptr<base::DictionaryValue> result_;
   scoped_refptr<DevToolsAgentHost> agent_host_;
   int last_sent_id_;
   std::vector<int> result_ids_;
   std::vector<std::string> notifications_;
+  std::vector<std::string> console_messages_;
   std::unique_ptr<base::DictionaryValue> requested_notification_params_;
 
  private:
   void DispatchProtocolMessage(DevToolsAgentHost* agent_host,
                                const std::string& message) override {
     std::unique_ptr<base::DictionaryValue> root(
         static_cast<base::DictionaryValue*>(
             base::JSONReader::Read(message).release()));
     int id;
     if (root->GetInteger("id", &id)) {
@@ skipping 555 matching lines @@
 
   // Make sure each frame has the expected url.
   EXPECT_THAT(
       GetAllFrameUrls(),
       ElementsAre("http://127.0.0.1/devtools/control_navigations/"
                   "iframe_navigation.html",
                   "http://a.com/devtools/navigation.html",
                   "http://b.com/devtools/control_navigations/meta_tag.html"));
 }
 
+IN_PROC_BROWSER_TEST_F(DevToolsProtocolTest, VirtualTimeTest) {
+  NavigateToURLBlockUntilNavigationsComplete(shell(), GURL("about:blank"), 1);
+  Attach();
+
+  std::unique_ptr<base::DictionaryValue> params(new base::DictionaryValue());
+  params->SetString("policy", "pause");
+  SendCommand("Emulation.setVirtualTimePolicy", std::move(params), true);
+
+  params.reset(new base::DictionaryValue());
+  params->SetString("expression",
+                    "setTimeout(function(){console.log('before')}, 1000);"
+                    "setTimeout(function(){console.log('after')}, 1001);");
+  SendCommand("Runtime.evaluate", std::move(params), true);
+
+  // Let virtual time advance for one second.
+  params.reset(new base::DictionaryValue());
+  params->SetString("policy", "advance");
+  params->SetInteger("budget", 1000);
+  SendCommand("Emulation.setVirtualTimePolicy", std::move(params), true);
+
+  WaitForNotification("Emulation.virtualTimeBudgetExpired");
+
+  params.reset(new base::DictionaryValue());
+  params->SetString("expression", "console.log('done')");
+  SendCommand("Runtime.evaluate", std::move(params), true);
+
+  // The second timer shold not fire.
+  EXPECT_THAT(console_messages_, ElementsAre("before", "done"));
+
+  // Let virtual time advance for another second, which should make the second
+  // timer fire.
+  params.reset(new base::DictionaryValue());
+  params->SetString("policy", "advance");
+  params->SetInteger("budget", 1000);
+  SendCommand("Emulation.setVirtualTimePolicy", std::move(params), true);
+
+  WaitForNotification("Emulation.virtualTimeBudgetExpired");
+
+  EXPECT_THAT(console_messages_, ElementsAre("before", "done", "after"));
+}
+
 }  // namespace content