Implement per-test timeout for UI tests.

chrome/browser/automation/automation_provider.cc

 // Copyright (c) 2009 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/browser/automation/automation_provider.h"
 
 #include <set>
 
 #include "app/l10n_util.h"
 #include "app/message_box_flags.h"
@@ skipping 410 matching lines @@
         WaitForAppModalDialogToBeShown)
     IPC_MESSAGE_HANDLER_DELAY_REPLY(
         AutomationMsg_GoBackBlockUntilNavigationsComplete,
         GoBackBlockUntilNavigationsComplete)
     IPC_MESSAGE_HANDLER_DELAY_REPLY(
         AutomationMsg_GoForwardBlockUntilNavigationsComplete,
         GoForwardBlockUntilNavigationsComplete)
 #if defined(OS_WIN)
     IPC_MESSAGE_HANDLER(AutomationMsg_ConnectExternalTab, ConnectExternalTab)
 #endif
+    IPC_MESSAGE_HANDLER(AutomationMsg_QuitAfterTimeout, QuitAfterTimeout)
   IPC_END_MESSAGE_MAP()
 }
 
 void AutomationProvider::ActivateTab(int handle, int at_index, int* status) {
   *status = -1;
   if (browser_tracker_->ContainsHandle(handle) && at_index > -1) {
     Browser* browser = browser_tracker_->GetResource(handle);
     if (at_index >= 0 && at_index < browser->tab_count()) {
       browser->SelectTabContentsAt(at_index, true);
       *status = 0;
@@ skipping 1572 matching lines @@
       browser->GoForward(CURRENT_TAB);
       return;
     }
   }
 
   AutomationMsg_GoForwardBlockUntilNavigationsComplete::WriteReplyParams(
       reply_message, AUTOMATION_MSG_NAVIGATION_ERROR);
   Send(reply_message);
 }
 
+void AutomationProvider::QuitAfterTimeout(int timeout_ms) {
+  MessageLoop::current()->PostDelayedTask(FROM_HERE,
+                                          new MessageLoop::QuitTask(),
+                                          timeout_ms);
+}
+
 RenderViewHost* AutomationProvider::GetViewForTab(int tab_handle) {
   if (tab_tracker_->ContainsHandle(tab_handle)) {
     NavigationController* tab = tab_tracker_->GetResource(tab_handle);
     if (!tab) {
       NOTREACHED();
       return NULL;
     }
 
     TabContents* tab_contents = tab->tab_contents();
     if (!tab_contents) {
@@ skipping 22 matching lines @@
   for (;iter != BrowserList::end(); ++iter) {
     gfx::NativeWindow this_window = (*iter)->window()->GetNativeHandle();
     if (window == this_window) {
       // Add() returns the existing handle for the resource if any.
       *browser_handle = browser_tracker_->Add(*iter);
       *success = true;
       return;
     }
   }
 }