Emulation.setVirtualTimePolicy to have an optional virtual time budge

third_party/WebKit/Source/web/InspectorEmulationAgent.cpp

 // Copyright 2015 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 "web/InspectorEmulationAgent.h"
 
 #include "core/frame/FrameHost.h"
 #include "core/frame/FrameView.h"
 #include "core/frame/Settings.h"
 #include "core/page/Page.h"
 #include "platform/geometry/DoubleRect.h"
+#include "public/platform/Platform.h"
+#include "public/platform/WebThread.h"
 #include "public/platform/WebViewScheduler.h"
 #include "web/DevToolsEmulator.h"
 #include "web/WebLocalFrameImpl.h"
 #include "web/WebViewImpl.h"
 
 namespace blink {
 
 namespace EmulationAgentState {
 static const char scriptExecutionDisabled[] = "scriptExecutionDisabled";
 static const char touchEventEmulationEnabled[] = "touchEventEmulationEnabled";
 static const char emulatedMedia[] = "emulatedMedia";
 }
 
 InspectorEmulationAgent* InspectorEmulationAgent::create(WebLocalFrameImpl* webLocalFrameImpl, Client* client)
 {
     return new InspectorEmulationAgent(webLocalFrameImpl, client);
 }
 
 InspectorEmulationAgent::InspectorEmulationAgent(WebLocalFrameImpl* webLocalFrameImpl, Client* client)
     : m_webLocalFrameImpl(webLocalFrameImpl)
     , m_client(client)
+    , m_virtualTimeBudgetExpiredTask(CancellableTaskFactory::create(this, &InspectorEmulationAgent::virtualTimeBudgetExpired))
 {
 }
 
 InspectorEmulationAgent::~InspectorEmulationAgent()
 {
 }
 
 WebViewImpl* InspectorEmulationAgent::webViewImpl()
 {
     return m_webLocalFrameImpl->viewImpl();
@@ skipping 43 matching lines @@
 {
     m_state->setString(EmulationAgentState::emulatedMedia, media);
     webViewImpl()->page()->settings().setMediaTypeOverride(media);
 }
 
 void InspectorEmulationAgent::setCPUThrottlingRate(ErrorString*, double throttlingRate)
 {
     m_client->setCPUThrottlingRate(throttlingRate);
 }
 
-void InspectorEmulationAgent::setVirtualTimePolicy(ErrorString*, const String& in_policy)
+void InspectorEmulationAgent::setVirtualTimePolicy(ErrorString*, const String& in_policy, const Maybe<int>& in_budget)
 {
     if (protocol::Emulation::VirtualTimePolicyEnum::Advance == in_policy) {
-        m_webLocalFrameImpl->view()->scheduler()->enableVirtualTime();
         m_webLocalFrameImpl->view()->scheduler()->setVirtualTimePolicy(WebViewScheduler::VirtualTimePolicy::ADVANCE);
     } else if (protocol::Emulation::VirtualTimePolicyEnum::Pause == in_policy) {
-        m_webLocalFrameImpl->view()->scheduler()->enableVirtualTime();
         m_webLocalFrameImpl->view()->scheduler()->setVirtualTimePolicy(WebViewScheduler::VirtualTimePolicy::PAUSE);
     } else if (protocol::Emulation::VirtualTimePolicyEnum::PauseIfNetworkFetchesPending == in_policy) {
-        m_webLocalFrameImpl->view()->scheduler()->enableVirtualTime();
         m_webLocalFrameImpl->view()->scheduler()->setVirtualTimePolicy(WebViewScheduler::VirtualTimePolicy::DETERMINISTIC_LOADING);
     }
+    m_webLocalFrameImpl->view()->scheduler()->enableVirtualTime();
+
+    if (in_budget.isJust()) {
+        WebTaskRunner* taskRunner = Platform::current()->currentThread()->getWebTaskRunner();
+        long long delayMillis = static_cast<long long>(in_budget.fromJust());
+        taskRunner->postDelayedTask(BLINK_FROM_HERE, m_virtualTimeBudgetExpiredTask->cancelAndCreate(), delayMillis);
+    }
 }
 
+void InspectorEmulationAgent::virtualTimeBudgetExpired()
+{
+    m_webLocalFrameImpl->view()->scheduler()->setVirtualTimePolicy(WebViewScheduler::VirtualTimePolicy::PAUSE);
+    frontend()->virtualTimeBudgetExpired();
+}
+
 DEFINE_TRACE(InspectorEmulationAgent)
 {
     visitor->trace(m_webLocalFrameImpl);
     InspectorBaseAgent::trace(visitor);
 }
 
 } // namespace blink