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