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