chrome/browser/printing/print_preview_dialog_controller_browsertest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/memory/scoped_ptr.h"
   6 #include "base/run_loop.h"
   7 #include "chrome/browser/printing/print_preview_dialog_controller.h"
   8 #include "chrome/browser/ui/browser.h"
   9 #include "chrome/browser/ui/browser_commands.h"
  10 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  11 #include "chrome/common/print_messages.h"
  12 #include "chrome/common/url_constants.h"
  13 #include "chrome/test/base/in_process_browser_test.h"
  14 #include "chrome/test/base/ui_test_utils.h"
  15 #include "content/public/browser/web_contents_observer.h"
  16 #include "content/public/test/browser_test_utils.h"
  17 #include "url/gurl.h"
  18 #include "ipc/ipc_message_macros.h"
  19
  20 using content::WebContents;
  21 using content::WebContentsObserver;
  22
  23 class RequestPrintPreviewObserver : public WebContentsObserver {
  24  public:
  25   explicit RequestPrintPreviewObserver(WebContents* dialog)
  26       : WebContentsObserver(dialog) {
  27   }
  28   virtual ~RequestPrintPreviewObserver() {}
  29
  30   void set_quit_closure(const base::Closure& quit_closure) {
  31     quit_closure_ = quit_closure;
  32   }
  33
  34  private:
  35   // content::WebContentsObserver implementation.
  36   virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE {
  37     IPC_BEGIN_MESSAGE_MAP(RequestPrintPreviewObserver, message)
  38       IPC_MESSAGE_HANDLER(PrintHostMsg_RequestPrintPreview,
  39                           OnRequestPrintPreview)
  40       IPC_MESSAGE_UNHANDLED(break;)
  41     IPC_END_MESSAGE_MAP();
  42     return false;  // Report not handled so the real handler receives it.
  43   }
  44
  45   void OnRequestPrintPreview(
  46       const PrintHostMsg_RequestPrintPreview_Params& /* params */) {
  47     base::MessageLoop::current()->PostTask(FROM_HERE, quit_closure_);
  48   }
  49
  50   base::Closure quit_closure_;
  51
  52   DISALLOW_COPY_AND_ASSIGN(RequestPrintPreviewObserver);
  53 };
  54
  55 class PrintPreviewDialogClonedObserver : public WebContentsObserver {
  56  public:
  57   explicit PrintPreviewDialogClonedObserver(WebContents* dialog)
  58       : WebContentsObserver(dialog) {
  59   }
  60   virtual ~PrintPreviewDialogClonedObserver() {}
  61
  62   RequestPrintPreviewObserver* request_preview_dialog_observer() {
  63     return request_preview_dialog_observer_.get();
  64   }
  65
  66  private:
  67   // content::WebContentsObserver implementation.
  68   virtual void DidCloneToNewWebContents(
  69       WebContents* old_web_contents,
  70       WebContents* new_web_contents) OVERRIDE {
  71     request_preview_dialog_observer_.reset(
  72         new RequestPrintPreviewObserver(new_web_contents));
  73   }
  74
  75   scoped_ptr<RequestPrintPreviewObserver> request_preview_dialog_observer_;
  76
  77   DISALLOW_COPY_AND_ASSIGN(PrintPreviewDialogClonedObserver);
  78 };
  79
  80 class PrintPreviewDialogDestroyedObserver : public WebContentsObserver {
  81  public:
  82   explicit PrintPreviewDialogDestroyedObserver(WebContents* dialog)
  83       : WebContentsObserver(dialog),
  84         dialog_destroyed_(false) {
  85   }
  86   virtual ~PrintPreviewDialogDestroyedObserver() {}
  87
  88   bool dialog_destroyed() const { return dialog_destroyed_; }
  89
  90  private:
  91   // content::WebContentsObserver implementation.
  92   virtual void WebContentsDestroyed() OVERRIDE {
  93     dialog_destroyed_ = true;
  94   }
  95
  96   bool dialog_destroyed_;
  97
  98   DISALLOW_COPY_AND_ASSIGN(PrintPreviewDialogDestroyedObserver);
  99 };
 100
 101 class PrintPreviewDialogControllerBrowserTest : public InProcessBrowserTest {
 102  public:
 103   PrintPreviewDialogControllerBrowserTest() : initiator_(NULL) {}
 104   virtual ~PrintPreviewDialogControllerBrowserTest() {}
 105
 106   WebContents* initiator() {
 107     return initiator_;
 108   }
 109
 110   void PrintPreview() {
 111     base::RunLoop run_loop;
 112     request_preview_dialog_observer()->set_quit_closure(run_loop.QuitClosure());
 113     chrome::Print(browser());
 114     run_loop.Run();
 115   }
 116
 117   WebContents* GetPrintPreviewDialog() {
 118     printing::PrintPreviewDialogController* dialog_controller =
 119         printing::PrintPreviewDialogController::GetInstance();
 120     return dialog_controller->GetPrintPreviewForContents(initiator_);
 121   }
 122
 123  private:
 124   virtual void SetUpOnMainThread() OVERRIDE {
 125     WebContents* first_tab =
 126         browser()->tab_strip_model()->GetActiveWebContents();
 127     ASSERT_TRUE(first_tab);
 128
 129     // Open a new tab so |cloned_tab_observer_| can see it first and attach a
 130     // RequestPrintPreviewObserver to it before the real
 131     // PrintPreviewMessageHandler gets created. Thus enabling
 132     // RequestPrintPreviewObserver to get messages first for the purposes of
 133     // this test.
 134     cloned_tab_observer_.reset(new PrintPreviewDialogClonedObserver(first_tab));
 135     chrome::DuplicateTab(browser());
 136
 137     initiator_ = browser()->tab_strip_model()->GetActiveWebContents();
 138     ASSERT_TRUE(initiator_);
 139     ASSERT_NE(first_tab, initiator_);
 140   }
 141
 142   virtual void CleanUpOnMainThread() OVERRIDE {
 143     cloned_tab_observer_.reset();
 144     initiator_ = NULL;
 145   }
 146
 147   RequestPrintPreviewObserver* request_preview_dialog_observer() {
 148     return cloned_tab_observer_->request_preview_dialog_observer();
 149   }
 150
 151   scoped_ptr<PrintPreviewDialogClonedObserver> cloned_tab_observer_;
 152   WebContents* initiator_;
 153
 154   DISALLOW_COPY_AND_ASSIGN(PrintPreviewDialogControllerBrowserTest);
 155 };
 156
 157 // Test to verify that when a initiator navigates, we can create a new preview
 158 // dialog for the new tab contents.
 159 // http://crbug.com/377337
 160 #if defined(OS_WIN)
 161 #define MAYBE_NavigateFromInitiatorTab DISABLED_NavigateFromInitiatorTab
 162 #else
 163 #define MAYBE_NavigateFromInitiatorTab NavigateFromInitiatorTab
 164 #endif
 165 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest,
 166                        MAYBE_NavigateFromInitiatorTab) {
 167   // print for the first time.
 168   PrintPreview();
 169
 170   // Get the preview dialog for the initiator tab.
 171   WebContents* preview_dialog = GetPrintPreviewDialog();
 172
 173   // Check a new print preview dialog got created.
 174   ASSERT_TRUE(preview_dialog);
 175   ASSERT_NE(initiator(), preview_dialog);
 176
 177   // Navigate in the initiator tab. Make sure navigating destroys the print
 178   // preview dialog.
 179   PrintPreviewDialogDestroyedObserver dialog_destroyed_observer(preview_dialog);
 180   ui_test_utils::NavigateToURL(browser(), GURL(chrome::kChromeUINewTabURL));
 181   ASSERT_TRUE(dialog_destroyed_observer.dialog_destroyed());
 182
 183   // Try printing again.
 184   PrintPreview();
 185
 186   // Get the print preview dialog for the initiator tab.
 187   WebContents* new_preview_dialog = GetPrintPreviewDialog();
 188
 189   // Check a new preview dialog got created.
 190   EXPECT_TRUE(new_preview_dialog);
 191 }
 192
 193 // Test to verify that after reloading the initiator, it creates a new print
 194 // preview dialog.
 195 // http://crbug.com/377337
 196 #if defined(OS_WIN)
 197 #define MAYBE_ReloadInitiatorTab DISABLED_ReloadInitiatorTab
 198 #else
 199 #define MAYBE_ReloadInitiatorTab ReloadInitiatorTab
 200 #endif
 201 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest,
 202                        MAYBE_ReloadInitiatorTab) {
 203   // print for the first time.
 204   PrintPreview();
 205
 206   WebContents* preview_dialog = GetPrintPreviewDialog();
 207
 208   // Check a new print preview dialog got created.
 209   ASSERT_TRUE(preview_dialog);
 210   ASSERT_NE(initiator(), preview_dialog);
 211
 212   // Reload the initiator. Make sure reloading destroys the print preview
 213   // dialog.
 214   PrintPreviewDialogDestroyedObserver dialog_destroyed_observer(preview_dialog);
 215   chrome::Reload(browser(), CURRENT_TAB);
 216   content::WaitForLoadStop(
 217       browser()->tab_strip_model()->GetActiveWebContents());
 218   ASSERT_TRUE(dialog_destroyed_observer.dialog_destroyed());
 219
 220   // Try printing again.
 221   PrintPreview();
 222
 223   // Create a preview dialog for the initiator tab.
 224   WebContents* new_preview_dialog = GetPrintPreviewDialog();
 225   EXPECT_TRUE(new_preview_dialog);
 226 }