Making CloseWindow and CloseTab automation API...

chrome/test/automated_ui_tests/automated_ui_test_base.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 "base/scoped_ptr.h"
 #include "chrome/app/chrome_dll_resource.h"
 #include "chrome/test/automated_ui_tests/automated_ui_test_base.h"
 #include "chrome/test/automation/browser_proxy.h"
 #include "chrome/test/automation/tab_proxy.h"
 #include "chrome/test/automation/window_proxy.h"
 #include "chrome/test/ui/ui_test.h"
 
 AutomatedUITestBase::AutomatedUITestBase() {}
 
 AutomatedUITestBase::~AutomatedUITestBase() {}
 
 void AutomatedUITestBase::LogErrorMessage(const std::string& error) {}
 
 void AutomatedUITestBase::LogWarningMessage(const std::string& warning) {}
 
 void AutomatedUITestBase::LogInfoMessage(const std::string& info) {}
 
 void AutomatedUITestBase::SetUp() {
   UITest::SetUp();
   set_active_browser(automation()->GetBrowserWindow(0));
 }
 
+bool AutomatedUITestBase::CloseActiveTab() {
+  BrowserProxy* browser = active_browser();
+  int tab_count;
+  bool is_timeout;
+  browser->GetTabCountWithTimeout(&tab_count,
+                                  action_max_timeout_ms(),
+                                  &is_timeout);
+
+  if (is_timeout) {
+    LogInfoMessage("get_tab_count_time_out");

[Finnur, 2009/05/01 04:05:49]

nit: add a 'd' after time (timed_out)

+    return false;
+  }
+
+  if (tab_count > 1) {
+    scoped_ptr<TabProxy> tab(GetActiveTab());
+    // Wait until tab is closed

[Finnur, 2009/05/01 04:05:49]

nit: End the comment in a period.

+    return tab->Close(true);
+  } else if (tab_count == 1) {
+    // Synchronously close the window if it is not the last window.
+    return CloseActiveWindow();
+  } else {
+    LogInfoMessage("invalid_tab_count");
+    return false;
+  }
+}
+
+bool AutomatedUITestBase::CloseActiveWindow() {
+  int browser_windows_count = 0;
+  if (!automation()->GetNormalBrowserWindowCount(&browser_windows_count))
+    return false;
+  // Avoid quitting the application by not closing the last window.
+  if (browser_windows_count < 2)
+    return false;
+  bool application_closed;
+  CloseBrowser(active_browser(), &application_closed);
+  if (application_closed) {
+    LogErrorMessage("Application closed unexpectedly.");
+    return false;
+  }
+  BrowserProxy* browser = automation()->FindNormalBrowserWindow();
+  if (browser == NULL) {
+    LogErrorMessage("Can't find browser window.");
+    return false;
+  }
+  set_active_browser(browser);
+  return true;
+}
+
 bool AutomatedUITestBase::DuplicateTab() {
   return RunCommand(IDC_DUPLICATE_TAB);
 }
 
 bool AutomatedUITestBase::OpenAndActivateNewBrowserWindow(
     BrowserProxy** previous_browser) {
   if (!automation()->OpenNewBrowserWindow(SW_SHOWNORMAL)) {
     LogWarningMessage("failed_to_open_new_browser_window");
     return false;
   }
@@ skipping 46 matching lines @@
     LogErrorMessage("browser_window_not_found");
     return false;
   }
 
   if (!browser->RunCommand(browser_command)) {
     LogWarningMessage("failure_running_browser_command");
     return false;
   }
   return true;
 }
+
+TabProxy* AutomatedUITestBase::GetActiveTab() {
+  BrowserProxy* browser = active_browser();
+  if (browser == NULL) {
+    LogErrorMessage("browser_window_not_found");
+    return false;
+  }
+
+  bool did_timeout;
+  TabProxy* tab =
+      browser->GetActiveTabWithTimeout(action_max_timeout_ms(), &did_timeout);
+  if (did_timeout)
+    return NULL;
+  return tab;
+}