Add pyauto tests for the New Tab page.

chrome/test/pyautolib/pyautolib.cc

 // Copyright (c) 2010 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 "base/string_number_conversions.h"
 #include "base/string_util.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/test/automation/extension_proxy.h"
 #include "chrome/test/automation/tab_proxy.h"
@@ skipping 153 matching lines @@
   if (!tab_proxy.get())
     return download_dir;
   EXPECT_TRUE(tab_proxy->GetDownloadDirectory(&download_dir));
   return download_dir;
 }
 
 bool PyUITestBase::OpenNewBrowserWindow(bool show) {
   return automation()->OpenNewBrowserWindow(Browser::TYPE_NORMAL, show);
 }
 
+bool PyUITestBase::CloseBrowserWindow(int window_index) {
+  scoped_refptr<BrowserProxy> browser_proxy =
+      automation()->GetBrowserWindow(window_index);
+  if (!browser_proxy.get())
+    return false;
+  bool app_closed;
+  return CloseBrowser(browser_proxy.get(), &app_closed);
+}
+
 int PyUITestBase::GetBrowserWindowCount() {
   int num_windows = 0;
   EXPECT_TRUE(automation()->GetBrowserWindowCount(&num_windows));
   return num_windows;
 }
 
 bool PyUITestBase::InstallExtension(const FilePath& crx_file, bool with_ui) {
   scoped_refptr<ExtensionProxy> proxy =
       automation()->InstallExtension(crx_file, with_ui);
   return proxy.get() != NULL;
@@ skipping 180 matching lines @@
   // TODO(phadjan.jr): figure out a way to unambiguously report error.
   if (!browser_proxy.get())
     return cookie_val;
   scoped_refptr<TabProxy> tab_proxy = browser_proxy->GetTab(tab_index);
   EXPECT_TRUE(tab_proxy.get());
   if (!tab_proxy.get())
     return cookie_val;
   EXPECT_TRUE(tab_proxy->GetCookies(cookie_url, &cookie_val));
   return cookie_val;
 }