chrome/browser/printing/print_preview_test.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 "chrome/browser/printing/print_preview_test.h"
   6
   7 #include "chrome/browser/ui/browser.h"
   8 #include "chrome/browser/ui/browser_iterator.h"
   9 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  10 #include "chrome/test/base/test_browser_window.h"
  11 #include "components/web_modal/web_contents_modal_dialog_host.h"
  12 #include "content/public/browser/plugin_service.h"
  13 #include "content/public/browser/web_contents.h"
  14
  15 using web_modal::WebContentsModalDialogHost;
  16 using web_modal::ModalDialogHostObserver;
  17
  18 namespace {
  19
  20 // Use a custom test browser window to provide a parent view to the print
  21 // preview dialog.
  22 class PrintPreviewTestBrowserWindow
  23     : public TestBrowserWindow,
  24       public WebContentsModalDialogHost {
  25  public:
  26   PrintPreviewTestBrowserWindow() {}
  27
  28   // BrowserWindow overrides
  29   virtual WebContentsModalDialogHost* GetWebContentsModalDialogHost() OVERRIDE {
  30     return this;
  31   }
  32
  33   // WebContentsModalDialogHost overrides
  34
  35   // The web contents modal dialog must be parented to *something*; use the
  36   // WebContents window since there is no true browser window for unit tests.
  37   virtual gfx::NativeView GetHostView() const OVERRIDE {
  38     return FindBrowser()->tab_strip_model()->GetActiveWebContents()->
  39         GetNativeView();
  40   }
  41
  42   virtual gfx::Point GetDialogPosition(const gfx::Size& size) OVERRIDE {
  43     return gfx::Point();
  44   }
  45
  46   virtual gfx::Size GetMaximumDialogSize() OVERRIDE {
  47     return gfx::Size();
  48   }
  49
  50   virtual void AddObserver(
  51       ModalDialogHostObserver* observer) OVERRIDE {}
  52   virtual void RemoveObserver(
  53       ModalDialogHostObserver* observer) OVERRIDE {}
  54
  55  private:
  56   Browser* FindBrowser() const {
  57     for (chrome::BrowserIterator it; !it.done(); it.Next()) {
  58       Browser* browser = *it;
  59       if (browser->window() == this)
  60         return browser;
  61     }
  62     NOTREACHED();
  63     return NULL;
  64   }
  65
  66   DISALLOW_COPY_AND_ASSIGN(PrintPreviewTestBrowserWindow);
  67 };
  68 }  // namespace
  69
  70 PrintPreviewTest::PrintPreviewTest() {}
  71 PrintPreviewTest::~PrintPreviewTest() {}
  72
  73 void PrintPreviewTest::SetUp() {
  74   BrowserWithTestWindowTest::SetUp();
  75
  76   // The PluginService will be destroyed at the end of the test (due to the
  77   // ShadowingAtExitManager in our base class).
  78   content::PluginService::GetInstance()->Init();
  79   content::PluginService::GetInstance()->DisablePluginsDiscoveryForTesting();
  80 }
  81
  82 BrowserWindow* PrintPreviewTest::CreateBrowserWindow() {
  83   return new PrintPreviewTestBrowserWindow;
  84 }