Addin a time-out to view base tests to prevent them from...

chrome/test/interactive_ui/view_event_test_base.cc

 // Copyright (c) 2006-2008 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 "chrome/test/interactive_ui/view_event_test_base.h"
 
 #if defined(OS_WIN)
 #include <ole2.h>
 #endif
 
+#include "base/compiler_specific.h"
 #include "base/message_loop.h"
+#include "base/string_util.h"
 #include "chrome/browser/automation/ui_controls.h"
 #include "views/view.h"
 #include "views/window/window.h"
 
 namespace {
 
+// Default delay for the time-out at which we stop message loop.
+const int kTimeoutInMS = 20000;
+
 // View subclass that allows you to specify the preferred size.
 class TestView : public views::View {
  public:
   TestView() {}
 
   void SetPreferredSize(const gfx::Size& size) {
     preferred_size_ = size;
     PreferredSizeChanged();
   }
 
@@ skipping 12 matching lines @@
   gfx::Size preferred_size_;
 
   DISALLOW_COPY_AND_ASSIGN(TestView);
 };
 
 // Delay in background thread before posting mouse move.
 const int kMouseMoveDelayMS = 200;
 
 }  // namespace
 
-ViewEventTestBase::ViewEventTestBase() : window_(NULL), content_view_(NULL) { }
+ViewEventTestBase::ViewEventTestBase()
+  : window_(NULL),
+    content_view_(NULL),
+    ALLOW_THIS_IN_INITIALIZER_LIST(method_factory_(this)) {
+}
 
 void ViewEventTestBase::Done() {
+  // Cancel the pending time-out.
+  method_factory_.RevokeAll();
+
   MessageLoop::current()->Quit();
 
 #if defined(OS_WIN)
   // We need to post a message to tickle the Dispatcher getting called and
   // exiting out of the nested loop. Without this the quit never runs.
   PostMessage(window_->GetNativeWindow(), WM_USER, 0, 0);
 #endif
 
   // If we're in a nested message loop, as is the case with menus, we need
   // to quit twice. The second quit does that for us.
-  MessageLoop::current()->PostDelayedTask(
-      FROM_HERE, new MessageLoop::QuitTask(), 0);
+  MessageLoop::current()->PostTask(FROM_HERE, new MessageLoop::QuitTask());
 }
 
 void ViewEventTestBase::SetUp() {
 #if defined(OS_WIN)
   OleInitialize(NULL);
 #endif
   window_ = views::Window::CreateChromeWindow(NULL, gfx::Rect(), this);
 }
 
 void ViewEventTestBase::TearDown() {
@@ skipping 28 matching lines @@
   // mouse events are going to be targeted correctly.
 #if defined(OS_WIN)
   SetForegroundWindow(window_->GetNativeWindow());
 #endif
 
   // Flush any pending events to make sure we start with a clean slate.
   MessageLoop::current()->RunAllPending();
 
   // Schedule a task that starts the test. Need to do this as we're going to
   // run the message loop.
-  MessageLoop::current()->PostDelayedTask(
+  MessageLoop::current()->PostTask(
       FROM_HERE,
-      NewRunnableMethod(this, &ViewEventTestBase::DoTestOnMessageLoop), 0);
+      NewRunnableMethod(this, &ViewEventTestBase::DoTestOnMessageLoop));
+
+  // Start the timeout timer to prevent hangs.
+  MessageLoop::current()->PostDelayedTask(FROM_HERE,
+      method_factory_.NewRunnableMethod(&ViewEventTestBase::TimedOut),
+      kTimeoutInMS);
 
   MessageLoop::current()->Run();
 }
 
 gfx::Size ViewEventTestBase::GetPreferredSize() {
   return gfx::Size();
 }
 
 void ViewEventTestBase::ScheduleMouseMoveInBackground(int x, int y) {
   if (!dnd_thread_.get()) {
@@ skipping 10 matching lines @@
 }
 
 void ViewEventTestBase::RunTestMethod(Task* task) {
   StopBackgroundThread();
 
   scoped_ptr<Task> task_deleter(task);
   task->Run();
   if (HasFatalFailure())
     Done();
 }
+
+void ViewEventTestBase::TimedOut() {
+  std::string error_message = "Test timed out. Each test runs for a max of ";
+  error_message += IntToString(kTimeoutInMS);
+  error_message += " ms (kTimeoutInMS).";
+
+  GTEST_NONFATAL_FAILURE_(error_message.c_str());
+
+  Done();
+}