Convert printing ui_tests to browser_tests.

chrome/browser/printing/printing_layout_browsertest.cc

 // Copyright (c) 2012 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/command_line.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
+#include "base/message_loop.h"
+#include "base/path_service.h"
 #include "base/process_util.h"
 #include "base/string_util.h"
 #include "base/test/test_file_util.h"
 #include "base/threading/simple_thread.h"
 #include "base/utf_string_conversions.h"
-#include "chrome/test/automation/tab_proxy.h"
-#include "chrome/test/ui/ui_test.h"
+#include "chrome/browser/printing/print_job.h"
+#include "chrome/browser/printing/print_view_manager.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
+#include "chrome/common/chrome_paths.h"
+#include "chrome/common/chrome_switches.h"
+#include "chrome/common/chrome_notification_types.h"
+#include "chrome/test/base/in_process_browser_test.h"
+#include "chrome/test/base/ui_test_utils.h"
+#include "content/public/browser/notification_observer.h"
+#include "content/public/browser/notification_registrar.h"
+#include "content/public/browser/notification_service.h"
 #include "net/test/test_server.h"
 #include "printing/image.h"
 #include "printing/printing_test.h"
 
 namespace {
 
 using printing::Image;
 
 const char kGenerateSwitch[] = "print-layout-generate";
 const FilePath::CharType kDocRoot[] = FILE_PATH_LITERAL("chrome/test/data");
 
-class PrintingLayoutTest : public PrintingTest<UITest> {
+class PrintingLayoutTest : public PrintingTest<InProcessBrowserTest>,
+                           public content::NotificationObserver {
  public:
   PrintingLayoutTest() {
-    emf_path_ = browser_directory_.AppendASCII("metafile_dumps");
-    launch_arguments_.AppendSwitchPath("debug-print", emf_path_);
-    show_window_ = true;
+    FilePath browser_directory;
+    PathService::Get(chrome::DIR_APP, &browser_directory);
+    emf_path_ = browser_directory.AppendASCII("metafile_dumps");
   }
 
-  virtual void SetUp() {
+  virtual void SetUp() OVERRIDE {
     // Make sure there is no left overs.
     CleanupDumpDirectory();
-    UITest::SetUp();
+    InProcessBrowserTest::SetUp();
   }
 
-  virtual void TearDown() {
-    UITest::TearDown();
+  virtual void TearDown() OVERRIDE {
+    InProcessBrowserTest::TearDown();
     file_util::Delete(emf_path_, true);
   }
 
+  virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
+    command_line->AppendSwitchPath(switches::kDebugPrint, emf_path_);
+  }
+
  protected:
   void PrintNowTab() {
-    scoped_refptr<TabProxy> tab_proxy(GetActiveTab());
-    ASSERT_TRUE(tab_proxy.get());
-    ASSERT_TRUE(tab_proxy->PrintNow());
+    registrar_.Add(this, chrome::NOTIFICATION_PRINT_JOB_EVENT,
+                   content::NotificationService::AllSources());
+
+    content::WebContents* web_contents = browser()->GetSelectedWebContents();

[Lei Zhang, 2012/04/09 19:44:44]

Just call browser()->GetSelectedTabContentsWrapper() instead?

[jam, 2012/04/09 20:16:22]

Done.

+    TabContentsWrapper* wrapper =
+        TabContentsWrapper::GetCurrentWrapperForContents(web_contents);
+    wrapper->print_view_manager()->PrintNow();
+    ui_test_utils::RunMessageLoop();
+    registrar_.RemoveAll();
+  }
+
+  virtual void Observe(int type,
+                       const content::NotificationSource& source,
+                       const content::NotificationDetails& details) {
+    DCHECK(type == chrome::NOTIFICATION_PRINT_JOB_EVENT);
+    switch (content::Details<printing::JobEventDetails>(details)->type()) {
+      case printing::JobEventDetails::JOB_DONE: {
+        // Succeeded.
+        MessageLoop::current()->PostTask(FROM_HERE, MessageLoop::QuitClosure());
+        break;
+      }
+      case printing::JobEventDetails::USER_INIT_CANCELED:
+      case printing::JobEventDetails::FAILED: {
+        // Failed.
+        ASSERT_TRUE(false);
+        MessageLoop::current()->PostTask(FROM_HERE, MessageLoop::QuitClosure());
+        break;
+      }
+      case printing::JobEventDetails::NEW_DOC:
+      case printing::JobEventDetails::USER_INIT_DONE:
+      case printing::JobEventDetails::DEFAULT_INIT_DONE:
+      case printing::JobEventDetails::NEW_PAGE:
+      case printing::JobEventDetails::PAGE_DONE:
+      case printing::JobEventDetails::DOC_DONE:
+      case printing::JobEventDetails::ALL_PAGES_REQUESTED: {
+        // Don't care.
+        break;
+      }
+      default: {
+        NOTREACHED();
+        break;
+      }
+    }
   }
 
   // Finds the dump for the last print job and compares it to the data named
   // |verification_name|. Compares the saved printed job pixels with the test
   // data pixels and returns the percentage of different pixels; 0 for success,
   // [0, 100] for failure.
   double CompareWithResult(const std::wstring& verification_name) {
     FilePath test_result(ScanFiles(verification_name));
     if (test_result.value().empty()) {
       // 100% different, the print job buffer is not there.
       return 100.;
     }
 
-    FilePath base_path(test_data_directory_.AppendASCII("printing"));
+    FilePath base_path(ui_test_utils::GetTestFilePath(
+        FilePath().AppendASCII("printing"), FilePath()));

[Lei Zhang, 2012/04/09 19:44:44]

FilePath().AppendASCII("printing") -> FilePath(L"printing") ?

[jam, 2012/04/09 20:16:22]

I didn't do it that way because that would make porting the test to non-windows
more work

     FilePath emf(base_path.Append(verification_name + L".emf"));
     FilePath png(base_path.Append(verification_name + L".png"));
 
     FilePath cleartype(base_path.Append(verification_name + L"_cleartype.png"));
     // Looks for Cleartype override.
     if (file_util::PathExists(cleartype) && IsClearTypeEnabled())
       png = cleartype;
 
     if (GenerateFiles()) {
       // Copy the .emf and generate an .png.
@@ skipping 29 matching lines @@
         FilePath rendering(
             base_path.Append(verification_name + L"_rendering.png"));
         emf_content.SaveToPng(rendering);
       }
       return std::max(diff_png, diff_emf);
     }
   }
 
   // Makes sure the directory exists and is empty.
   void CleanupDumpDirectory() {
-    EXPECT_TRUE(file_util::DieFileDie(emf_path(), true));
-    EXPECT_TRUE(file_util::CreateDirectory(emf_path()));
+    EXPECT_TRUE(file_util::DieFileDie(emf_path_, true));
+    EXPECT_TRUE(file_util::CreateDirectory(emf_path_));
   }
 
   // Returns if Clear Type is currently enabled.
   static bool IsClearTypeEnabled() {
     BOOL ct_enabled = 0;
     if (SystemParametersInfo(SPI_GETCLEARTYPE, 0, &ct_enabled, 0) && ct_enabled)
       return true;
     UINT smoothing = 0;
     if (SystemParametersInfo(SPI_GETFONTSMOOTHINGTYPE, 0, &smoothing, 0) &&
         smoothing == FE_FONTSMOOTHINGCLEARTYPE)
       return true;
     return false;
   }
 
  private:
   // Verifies that there is one .emf and one .prn file in the dump directory.
   // Returns the path of the .emf file and deletes the .prn file.
   std::wstring ScanFiles(const std::wstring& verification_name) {
     // Try to 10 seconds.
     std::wstring emf_file;
     std::wstring prn_file;
     bool found_emf = false;
     bool found_prn = false;
     for (int i = 0; i < 100; ++i) {
-      file_util::FileEnumerator enumerator(emf_path(), false,
+      file_util::FileEnumerator enumerator(emf_path_, false,
           file_util::FileEnumerator::FILES);
       emf_file.clear();
       prn_file.clear();
       found_emf = false;
       found_prn = false;
       FilePath file;
       while (!(file = enumerator.Next()).empty()) {
         std::wstring ext = file.Extension();
         if (base::strcasecmp(WideToUTF8(ext).c_str(), ".emf") == 0) {
           EXPECT_FALSE(found_emf) << "Found a leftover .EMF file: \"" <<
@@ skipping 20 matching lines @@
     }
     EXPECT_TRUE(found_emf) << ".PRN file is: " << prn_file;
     EXPECT_TRUE(found_prn) << ".EMF file is: " << emf_file;
     return emf_file;
   }
 
   static bool GenerateFiles() {
     return CommandLine::ForCurrentProcess()->HasSwitch(kGenerateSwitch);
   }
 
-  const FilePath& emf_path() const { return emf_path_; }
-
   FilePath emf_path_;
+  content::NotificationRegistrar registrar_;
 
   DISALLOW_COPY_AND_ASSIGN(PrintingLayoutTest);
 };
 
-// Tests that don't need UI access.
-class PrintingLayoutTestHidden : public PrintingLayoutTest {
- public:
-  PrintingLayoutTestHidden() {
-    show_window_ = false;
-  }
-};
-
 class PrintingLayoutTextTest : public PrintingLayoutTest {
   typedef PrintingLayoutTest Parent;
  public:
   // Returns if the test is disabled.
   // http://crbug.com/64869 Until the issue is fixed, disable the test if
   // ClearType is enabled.
   static bool IsTestCaseDisabled() {
     return Parent::IsTestCaseDisabled() || IsClearTypeEnabled();
   }
 };
@@ skipping 30 matching lines @@
       WM_COMMAND,
       print_button_id,
       reinterpret_cast<LPARAM>(GetDlgItem(dialog_window, print_button_id)));
   return res == 0;
 }
 
 // Dismiss the first dialog box owned by owner_process by "executing" the
 // default button.
 class DismissTheWindow : public base::DelegateSimpleThread::Delegate {
  public:
-  explicit DismissTheWindow(DWORD owner_process)
-      : owner_process_(owner_process) {
+  DismissTheWindow()
+      : owner_process_(base::Process::Current().pid()) {
   }
 
   virtual void Run() {
     HWND dialog_window;
     for (;;) {
       // First enumerate the windows.
       dialog_window = FindDialogWindow(owner_process_);
 
       // Try to close it.
       if (dialog_window) {
@@ skipping 13 matching lines @@
 
   DWORD owner_process() { return owner_process_; }
 
  private:
   DWORD owner_process_;
 };
 
 }  // namespace
 
 // Fails, see http://crbug.com/7721.
-TEST_F(PrintingLayoutTextTest, DISABLED_Complex) {
+IN_PROC_BROWSER_TEST_F(PrintingLayoutTextTest, DISABLED_Complex) {
   if (IsTestCaseDisabled())
     return;
 
-  DismissTheWindow dismisser(base::GetProcId(process()));
+  DismissTheWindow dismisser;
   base::DelegateSimpleThread close_printdlg_thread(&dismisser,
                                                    "close_printdlg_thread");
 
   // Print a document, check its output.
-  net::TestServer test_server(net::TestServer::TYPE_HTTP,
-                              net::TestServer::kLocalhost,
-                              FilePath(kDocRoot));
-  ASSERT_TRUE(test_server.Start());
+  ASSERT_TRUE(test_server()->Start());
 
-  NavigateToURL(test_server.GetURL("files/printing/test1.html"));
+  ui_test_utils::NavigateToURL(
+      browser(), test_server()->GetURL("files/printing/test1.html"));
   close_printdlg_thread.Start();
   PrintNowTab();
   close_printdlg_thread.Join();
   EXPECT_EQ(0., CompareWithResult(L"test1"));
 }
 
 struct TestPool {
   const char* source;
   const wchar_t* result;
 };
 
 const TestPool kTestPool[] = {
   // ImagesB&W
   "files/printing/test2.html", L"test2",
   // ImagesTransparent
   "files/printing/test3.html", L"test3",
   // ImageColor
   "files/printing/test4.html", L"test4",
 };
 
 // http://crbug.com/7721
-TEST_F(PrintingLayoutTestHidden, DISABLED_ManyTimes) {
+IN_PROC_BROWSER_TEST_F(PrintingLayoutTest, DISABLED_ManyTimes) {
   if (IsTestCaseDisabled())
     return;
 
-  net::TestServer test_server(net::TestServer::TYPE_HTTP,
-                              net::TestServer::kLocalhost,
-                              FilePath(kDocRoot));
-  ASSERT_TRUE(test_server.Start());
+  ASSERT_TRUE(test_server()->Start());
 
-  DismissTheWindow dismisser(base::GetProcId(process()));
+  DismissTheWindow dismisser;
 
   ASSERT_GT(arraysize(kTestPool), 0u);
   for (int i = 0; i < arraysize(kTestPool); ++i) {
     if (i)
       CleanupDumpDirectory();
     const TestPool& test = kTestPool[i % arraysize(kTestPool)];
-    NavigateToURL(test_server.GetURL(test.source));
+    ui_test_utils::NavigateToURL(browser(), test_server()->GetURL(test.source));
     base::DelegateSimpleThread close_printdlg_thread1(&dismisser,
                                                       "close_printdlg_thread");
     EXPECT_EQ(NULL, FindDialogWindow(dismisser.owner_process()));
     close_printdlg_thread1.Start();
     PrintNowTab();
     close_printdlg_thread1.Join();
     EXPECT_EQ(0., CompareWithResult(test.result)) << test.result;
     CleanupDumpDirectory();
     base::DelegateSimpleThread close_printdlg_thread2(&dismisser,
                                                       "close_printdlg_thread");
@@ skipping 15 matching lines @@
                                                       "close_printdlg_thread");
     EXPECT_EQ(NULL, FindDialogWindow(dismisser.owner_process()));
     close_printdlg_thread4.Start();
     PrintNowTab();
     close_printdlg_thread4.Join();
     EXPECT_EQ(0., CompareWithResult(test.result)) << test.result;
   }
 }
 
 // Prints a popup and immediately closes it. Disabled because it crashes.
-TEST_F(PrintingLayoutTest, DISABLED_Delayed) {
+IN_PROC_BROWSER_TEST_F(PrintingLayoutTest, DISABLED_Delayed) {
   if (IsTestCaseDisabled())
     return;
 
-  net::TestServer test_server(net::TestServer::TYPE_HTTP,
-                              net::TestServer::kLocalhost,
-                              FilePath(kDocRoot));
-  ASSERT_TRUE(test_server.Start());
+  ASSERT_TRUE(test_server()->Start());
 
   {
-    scoped_refptr<TabProxy> tab_proxy(GetActiveTab());
-    ASSERT_TRUE(tab_proxy.get());
     bool is_timeout = true;
-    GURL url = test_server.GetURL("files/printing/popup_delayed_print.htm");
-    EXPECT_EQ(AUTOMATION_MSG_NAVIGATION_SUCCESS,
-              tab_proxy->NavigateToURL(url));
+    GURL url = test_server()->GetURL("files/printing/popup_delayed_print.htm");
+    ui_test_utils::NavigateToURL(browser(), url);
 
-    DismissTheWindow dismisser(base::GetProcId(process()));
+    DismissTheWindow dismisser;
     base::DelegateSimpleThread close_printdlg_thread(&dismisser,
                                                      "close_printdlg_thread");
     close_printdlg_thread.Start();
     close_printdlg_thread.Join();
 
     // Force a navigation elsewhere to verify that it's fine with it.
-    url = test_server.GetURL("files/printing/test1.html");
-    EXPECT_EQ(AUTOMATION_MSG_NAVIGATION_SUCCESS,
-              tab_proxy->NavigateToURL(url));
+    url = test_server()->GetURL("files/printing/test1.html");
+    ui_test_utils::NavigateToURL(browser(), url);
   }
-  CloseBrowserAndServer();
+  browser()->CloseWindow();
+  ui_test_utils::RunAllPendingInMessageLoop();
 
   EXPECT_EQ(0., CompareWithResult(L"popup_delayed_print"))
       << L"popup_delayed_print";
 }
 
 // Prints a popup and immediately closes it. http://crbug.com/7721
-TEST_F(PrintingLayoutTest, DISABLED_IFrame) {
+IN_PROC_BROWSER_TEST_F(PrintingLayoutTest, DISABLED_IFrame) {
   if (IsTestCaseDisabled())
     return;
 
-  net::TestServer test_server(net::TestServer::TYPE_HTTP,
-                              net::TestServer::kLocalhost,
-                              FilePath(kDocRoot));
-  ASSERT_TRUE(test_server.Start());
+  ASSERT_TRUE(test_server()->Start());
 
   {
-    scoped_refptr<TabProxy> tab_proxy(GetActiveTab());
-    ASSERT_TRUE(tab_proxy.get());
-    GURL url = test_server.GetURL("files/printing/iframe.htm");
-    EXPECT_EQ(AUTOMATION_MSG_NAVIGATION_SUCCESS,
-              tab_proxy->NavigateToURL(url));
+    GURL url = test_server()->GetURL("files/printing/iframe.htm");
+    ui_test_utils::NavigateToURL(browser(), url);
 
-    DismissTheWindow dismisser(base::GetProcId(process()));
+    DismissTheWindow dismisser;
     base::DelegateSimpleThread close_printdlg_thread(&dismisser,
                                                      "close_printdlg_thread");
     close_printdlg_thread.Start();
     close_printdlg_thread.Join();
 
     // Force a navigation elsewhere to verify that it's fine with it.
-    url = test_server.GetURL("files/printing/test1.html");
-    EXPECT_EQ(AUTOMATION_MSG_NAVIGATION_SUCCESS,
-              tab_proxy->NavigateToURL(url));
+    url = test_server()->GetURL("files/printing/test1.html");
+    ui_test_utils::NavigateToURL(browser(), url);
   }
-  CloseBrowserAndServer();
+  browser()->CloseWindow();
+  ui_test_utils::RunAllPendingInMessageLoop();
 
   EXPECT_EQ(0., CompareWithResult(L"iframe")) << L"iframe";
 }