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/url_constants.h"
  12 #include "chrome/test/base/in_process_browser_test.h"
  13 #include "chrome/test/base/ui_test_utils.h"
  14 #include "components/printing/common/print_messages.h"
  15 #include "content/public/browser/web_contents_observer.h"
  16 #include "content/public/test/browser_test_utils.h"
  17 #include "ipc/ipc_message_macros.h"
  18 #include "url/gurl.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   ~RequestPrintPreviewObserver() override {}
  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   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   ~PrintPreviewDialogClonedObserver() override {}
  61
  62   RequestPrintPreviewObserver* request_preview_dialog_observer() {
  63     return request_preview_dialog_observer_.get();
  64   }
  65
  66  private:
  67   // content::WebContentsObserver implementation.
  68   void DidCloneToNewWebContents(WebContents* old_web_contents,
  69                                 WebContents* new_web_contents) override {
  70     request_preview_dialog_observer_.reset(
  71         new RequestPrintPreviewObserver(new_web_contents));
  72   }
  73
  74   scoped_ptr<RequestPrintPreviewObserver> request_preview_dialog_observer_;
  75
  76   DISALLOW_COPY_AND_ASSIGN(PrintPreviewDialogClonedObserver);
  77 };
  78
  79 class PrintPreviewDialogDestroyedObserver : public WebContentsObserver {
  80  public:
  81   explicit PrintPreviewDialogDestroyedObserver(WebContents* dialog)
  82       : WebContentsObserver(dialog),
  83         dialog_destroyed_(false) {
  84   }
  85   ~PrintPreviewDialogDestroyedObserver() override {}
  86
  87   bool dialog_destroyed() const { return dialog_destroyed_; }
  88
  89  private:
  90   // content::WebContentsObserver implementation.
  91   void WebContentsDestroyed() override { dialog_destroyed_ = true; }
  92
  93   bool dialog_destroyed_;
  94
  95   DISALLOW_COPY_AND_ASSIGN(PrintPreviewDialogDestroyedObserver);
  96 };
  97
  98 class PrintPreviewDialogControllerBrowserTest : public InProcessBrowserTest {
  99  public:
 100   PrintPreviewDialogControllerBrowserTest() : initiator_(NULL) {}
 101   ~PrintPreviewDialogControllerBrowserTest() override {}
 102
 103   WebContents* initiator() {
 104     return initiator_;
 105   }
 106
 107   void PrintPreview() {
 108     base::RunLoop run_loop;
 109     request_preview_dialog_observer()->set_quit_closure(run_loop.QuitClosure());
 110     chrome::Print(browser());
 111     run_loop.Run();
 112   }
 113
 114   WebContents* GetPrintPreviewDialog() {
 115     printing::PrintPreviewDialogController* dialog_controller =
 116         printing::PrintPreviewDialogController::GetInstance();
 117     return dialog_controller->GetPrintPreviewForContents(initiator_);
 118   }
 119
 120  private:
 121   void SetUpOnMainThread() override {
 122     WebContents* first_tab =
 123         browser()->tab_strip_model()->GetActiveWebContents();
 124     ASSERT_TRUE(first_tab);
 125
 126     // Open a new tab so |cloned_tab_observer_| can see it first and attach a
 127     // RequestPrintPreviewObserver to it before the real
 128     // PrintPreviewMessageHandler gets created. Thus enabling
 129     // RequestPrintPreviewObserver to get messages first for the purposes of
 130     // this test.
 131     cloned_tab_observer_.reset(new PrintPreviewDialogClonedObserver(first_tab));
 132     chrome::DuplicateTab(browser());
 133
 134     initiator_ = browser()->tab_strip_model()->GetActiveWebContents();
 135     ASSERT_TRUE(initiator_);
 136     ASSERT_NE(first_tab, initiator_);
 137   }
 138
 139   void TearDownOnMainThread() override {
 140     cloned_tab_observer_.reset();
 141     initiator_ = NULL;
 142   }
 143
 144   RequestPrintPreviewObserver* request_preview_dialog_observer() {
 145     return cloned_tab_observer_->request_preview_dialog_observer();
 146   }
 147
 148   scoped_ptr<PrintPreviewDialogClonedObserver> cloned_tab_observer_;
 149   WebContents* initiator_;
 150
 151   DISALLOW_COPY_AND_ASSIGN(PrintPreviewDialogControllerBrowserTest);
 152 };
 153
 154 // Test to verify that when a initiator navigates, we can create a new preview
 155 // dialog for the new tab contents.
 156 // http://crbug.com/377337
 157 #if defined(OS_WIN)
 158 #define MAYBE_NavigateFromInitiatorTab DISABLED_NavigateFromInitiatorTab
 159 #else
 160 #define MAYBE_NavigateFromInitiatorTab NavigateFromInitiatorTab
 161 #endif
 162 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest,
 163                        MAYBE_NavigateFromInitiatorTab) {
 164   // print for the first time.
 165   PrintPreview();
 166
 167   // Get the preview dialog for the initiator tab.
 168   WebContents* preview_dialog = GetPrintPreviewDialog();
 169
 170   // Check a new print preview dialog got created.
 171   ASSERT_TRUE(preview_dialog);
 172   ASSERT_NE(initiator(), preview_dialog);
 173
 174   // Navigate in the initiator tab. Make sure navigating destroys the print
 175   // preview dialog.
 176   PrintPreviewDialogDestroyedObserver dialog_destroyed_observer(preview_dialog);
 177   ui_test_utils::NavigateToURL(browser(), GURL(chrome::kChromeUINewTabURL));
 178   ASSERT_TRUE(dialog_destroyed_observer.dialog_destroyed());
 179
 180   // Try printing again.
 181   PrintPreview();
 182
 183   // Get the print preview dialog for the initiator tab.
 184   WebContents* new_preview_dialog = GetPrintPreviewDialog();
 185
 186   // Check a new preview dialog got created.
 187   EXPECT_TRUE(new_preview_dialog);
 188 }
 189
 190 // Test to verify that after reloading the initiator, it creates a new print
 191 // preview dialog.
 192 // http://crbug.com/377337
 193 #if defined(OS_WIN)
 194 #define MAYBE_ReloadInitiatorTab DISABLED_ReloadInitiatorTab
 195 #else
 196 #define MAYBE_ReloadInitiatorTab ReloadInitiatorTab
 197 #endif
 198 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest,
 199                        MAYBE_ReloadInitiatorTab) {
 200   // print for the first time.
 201   PrintPreview();
 202
 203   WebContents* preview_dialog = GetPrintPreviewDialog();
 204
 205   // Check a new print preview dialog got created.
 206   ASSERT_TRUE(preview_dialog);
 207   ASSERT_NE(initiator(), preview_dialog);
 208
 209   // Reload the initiator. Make sure reloading destroys the print preview
 210   // dialog.
 211   PrintPreviewDialogDestroyedObserver dialog_destroyed_observer(preview_dialog);
 212   chrome::Reload(browser(), CURRENT_TAB);
 213   content::WaitForLoadStop(
 214       browser()->tab_strip_model()->GetActiveWebContents());
 215   ASSERT_TRUE(dialog_destroyed_observer.dialog_destroyed());
 216
 217   // Try printing again.
 218   PrintPreview();
 219
 220   // Create a preview dialog for the initiator tab.
 221   WebContents* new_preview_dialog = GetPrintPreviewDialog();
 222   EXPECT_TRUE(new_preview_dialog);
 223 }