Making CloseWindow and CloseTab automation API...

chrome/test/automated_ui_tests/automated_ui_tests.cc

 // Copyright (c) 2006-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 <fstream>
 
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/logging.h"
 #include "base/path_service.h"
@@ skipping 400 matching lines @@
   // The vector will contain mostly IDC values for encoding commands plus a few
   // menu separators (0 values). If we hit a separator we just retry.
   int index = base::RandInt(0, len);
   while ((*encodings)[index].encoding_id == 0) {
     index = base::RandInt(0, len);
   }
 
   return RunCommandAsync((*encodings)[index].encoding_id);
 }
 
-bool AutomatedUITest::CloseActiveTab() {
-  bool return_value = false;
-  BrowserProxy* browser = active_browser();
-  if (browser == NULL) {
-    AddErrorAttribute("browser_window_not_found");
-    return false;
-  }
-  int browser_windows_count;
-  int tab_count;
-  bool is_timeout;
-  browser->GetTabCountWithTimeout(&tab_count,
-                                  action_max_timeout_ms(),
-                                  &is_timeout);
-  automation()->GetBrowserWindowCount(&browser_windows_count);
-  // Avoid quitting the application by not closing the last window.
-  if (tab_count > 1) {
-    return_value = browser->RunCommandAsync(IDC_CLOSE_TAB);
-    // Wait for the tab to close before we continue.
-    if (!browser->WaitForTabCountToBecome(tab_count - 1,
-                                          action_max_timeout_ms())) {
-      AddWarningAttribute("tab_count_failed_to_change");
-      return false;
-    }
-  } else if (tab_count == 1 && browser_windows_count > 1) {
-    return_value = browser->RunCommandAsync(IDC_CLOSE_TAB);
-    // Wait for the window to close before we continue.
-    if (!automation()->WaitForWindowCountToBecome(browser_windows_count - 1,
-                                                  action_max_timeout_ms())) {
-      AddWarningAttribute("window_count_failed_to_change");
-      return false;
-    }
-  } else {
-    AddInfoAttribute("would_have_exited_application");
-    return false;
-  }
-  return return_value;
-}
-
 bool AutomatedUITest::FindInPage() {
   return RunCommandAsync(IDC_FIND);
 }
 
 bool AutomatedUITest::ForwardButton() {
   return RunCommandAsync(IDC_FORWARD);
 }
 
 bool AutomatedUITest::GoOffTheRecord() {
   return RunCommandAsync(IDC_NEW_INCOGNITO_WINDOW);
 }
 
 bool AutomatedUITest::Home() {
   return RunCommandAsync(IDC_HOME);
 }
 
 bool AutomatedUITest::JavaScriptConsole() {
   return RunCommandAsync(IDC_JS_CONSOLE);
 }
 
 bool AutomatedUITest::JavaScriptDebugger() {
   return RunCommandAsync(IDC_DEBUGGER);
 }
 
 bool AutomatedUITest::Navigate() {
-  BrowserProxy* browser = active_browser();
-  if (browser == NULL) {
-    AddErrorAttribute("browser_window_not_found");
-    return false;
-  }
-  bool did_timeout;
-  scoped_ptr<TabProxy> tab(
-      browser->GetActiveTabWithTimeout(action_max_timeout_ms(), &did_timeout));
-  // TODO(devint): This might be masking a bug. I can't think of many
-  // valid cases where we would get a browser window, but not be able
-  // to return an active tab. Yet this has happened and has triggered crashes.
-  // Investigate this.
+  scoped_ptr<TabProxy> tab(GetActiveTab());
   if (tab.get() == NULL) {
     AddErrorAttribute("active_tab_not_found");
     return false;
   }
   std::string url = "about:blank";
   if (init_reader_.NodeAttribute("url", &url)) {
     xml_writer_.AddAttribute("url", url);
   }
   GURL test_url(url);
-  did_timeout = false;
+  bool did_timeout = false;
   tab->NavigateToURLWithTimeout(test_url,
                                 command_execution_timeout_ms(),
                                 &did_timeout);
 
   if (did_timeout) {
     AddWarningAttribute("timeout");
     return false;
   }
   return true;
 }
@@ skipping 156 matching lines @@
                                          - 1);
     return_value = return_value &&
                    DoAction(kTestDialogPossibleActions[action_index]);
     if (DidCrash(false))
       break;
   }
   return DoAction("PressEscapeKey") && return_value;
 }
 
 bool AutomatedUITest::ForceCrash() {
-  BrowserProxy* browser = active_browser();
-  if (browser == NULL) {
-    AddErrorAttribute("browser_window_not_found");
-    return false;
-  }
-  scoped_ptr<TabProxy> tab(browser->GetActiveTab());
+  scoped_ptr<TabProxy> tab(GetActiveTab());
   GURL test_url("about:crash");
   bool did_timeout;
   tab->NavigateToURLWithTimeout(test_url, kDebuggingTimeoutMsec, &did_timeout);
   if (!did_timeout) {
     AddInfoAttribute("expected_crash");
     return false;
   }
   return true;
 }
 
@@ skipping 261 matching lines @@
   }
 }
 
 TEST_F(AutomatedUITest, TheOneAndOnlyTest) {
   const CommandLine& parsed_command_line = *CommandLine::ForCurrentProcess();
   if (parsed_command_line.HasSwitch(kReproSwitch))
     RunReproduction();
   else
     RunAutomatedUITest();
 }