Implementation of TestRunner::Sleep

content/shell/renderer/test_runner/test_runner.cc

 // Copyright 2014 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 "content/shell/renderer/test_runner/test_runner.h"
 
 #include <limits>
 
 #include "base/logging.h"
+#include "base/run_loop.h"
 #include "content/public/test/layouttest_support.h"
 #include "content/shell/common/test_runner/test_preferences.h"
 #include "content/shell/renderer/binding_helpers.h"
 #include "content/shell/renderer/test_runner/mock_credential_manager_client.h"
 #include "content/shell/renderer/test_runner/mock_web_push_client.h"
 #include "content/shell/renderer/test_runner/mock_web_speech_recognizer.h"
 #include "content/shell/renderer/test_runner/notification_presenter.h"
 #include "content/shell/renderer/test_runner/test_interfaces.h"
 #include "content/shell/renderer/test_runner/web_permissions.h"
 #include "content/shell/renderer/test_runner/web_test_delegate.h"
@@ skipping 278 matching lines @@
                                 const std::string& registration_id);
   void SetMockPushClientError(const std::string& message);
   void SetBluetoothMockDataSet(const std::string& dataset_name);
 
   std::string PlatformName();
   std::string TooltipText();
   bool DisableNotifyDone();
   int WebHistoryItemCount();
   bool InterceptPostMessage();
   void SetInterceptPostMessage(bool value);
+  void Sleep(int milliseconds);
 
   void NotImplemented(const gin::Arguments& args);
 
   void ForceNextWebGLContextCreationToFail();
 
   base::WeakPtr<TestRunner> runner_;
 
   DISALLOW_COPY_AND_ASSIGN(TestRunnerBindings);
 };
 
@@ skipping 135 matching lines @@
                  &TestRunnerBindings::DumpUserGestureInFrameLoadCallbacks)
       .SetMethod("dumpTitleChanges", &TestRunnerBindings::DumpTitleChanges)
       .SetMethod("dumpCreateView", &TestRunnerBindings::DumpCreateView)
       .SetMethod("setCanOpenWindows", &TestRunnerBindings::SetCanOpenWindows)
       .SetMethod("dumpResourceLoadCallbacks",
                  &TestRunnerBindings::DumpResourceLoadCallbacks)
       .SetMethod("dumpResourceRequestCallbacks",
                  &TestRunnerBindings::DumpResourceRequestCallbacks)
       .SetMethod("dumpResourceResponseMIMETypes",
                  &TestRunnerBindings::DumpResourceResponseMIMETypes)
+      .SetMethod("sleep", &TestRunnerBindings::Sleep)
       .SetMethod("setImagesAllowed", &TestRunnerBindings::SetImagesAllowed)
       .SetMethod("setMediaAllowed", &TestRunnerBindings::SetMediaAllowed)
       .SetMethod("setScriptsAllowed", &TestRunnerBindings::SetScriptsAllowed)
       .SetMethod("setStorageAllowed", &TestRunnerBindings::SetStorageAllowed)
       .SetMethod("setPluginsAllowed", &TestRunnerBindings::SetPluginsAllowed)
       .SetMethod("setAllowDisplayOfInsecureContent",
                  &TestRunnerBindings::SetAllowDisplayOfInsecureContent)
       .SetMethod("setAllowRunningOfInsecureContent",
                  &TestRunnerBindings::SetAllowRunningOfInsecureContent)
       .SetMethod("dumpPermissionClientCallbacks",
@@ skipping 636 matching lines @@
 void TestRunnerBindings::DumpResourceRequestCallbacks() {
   if (runner_)
     runner_->DumpResourceRequestCallbacks();
 }
 
 void TestRunnerBindings::DumpResourceResponseMIMETypes() {
   if (runner_)
     runner_->DumpResourceResponseMIMETypes();
 }
 
+void TestRunnerBindings::Sleep(int milliseconds) {
+  if (runner_)
+    runner_->Sleep(milliseconds);
+}
+
 void TestRunnerBindings::SetImagesAllowed(bool allowed) {
   if (runner_)
     runner_->SetImagesAllowed(allowed);
 }
 
 void TestRunnerBindings::SetMediaAllowed(bool allowed) {
   if (runner_)
     runner_->SetMediaAllowed(allowed);
 }
 
@@ skipping 1481 matching lines @@
 }
 
 void TestRunner::DumpResourceRequestCallbacks() {
   dump_resource_request_callbacks_ = true;
 }
 
 void TestRunner::DumpResourceResponseMIMETypes() {
   dump_resource_reqponse_mime_types_ = true;
 }
 
+void TestRunner::Sleep(int milliseconds) {
+  web_view_->setDefersLoading(true);
+
+  base::MessageLoop* ml = base::MessageLoop::current();
+  base::MessageLoop::ScopedNestableTaskAllower allow_nested_tasks(ml);
+  base::RunLoop run_loop;
+  ml->PostDelayedTask(FROM_HERE, run_loop.QuitClosure(),
+                      base::TimeDelta::FromMilliseconds(milliseconds));
+  run_loop.Run();
+
+  web_view_->setDefersLoading(false);
+}
+
 void TestRunner::SetImagesAllowed(bool allowed) {
   web_permissions_->SetImagesAllowed(allowed);
 }
 
 void TestRunner::SetMediaAllowed(bool allowed) {
   web_permissions_->SetMediaAllowed(allowed);
 }
 
 void TestRunner::SetScriptsAllowed(bool allowed) {
   web_permissions_->SetScriptsAllowed(allowed);
@@ skipping 345 matching lines @@
 }
 
 void TestRunner::DidLosePointerLockInternal() {
   bool was_locked = pointer_locked_;
   pointer_locked_ = false;
   if (was_locked)
     web_view_->didLosePointerLock();
 }
 
 }  // namespace content