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: PrintPreviewTestBrowserWindow()28 PrintPreviewTestBrowserWindow() {} 29 30 // BrowserWindow overrides GetWebContentsModalDialogHost()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. GetHostView() const39 virtual gfx::NativeView GetHostView() const OVERRIDE { 40 return FindBrowser()->tab_strip_model()->GetActiveWebContents()-> 41 GetNativeView(); 42 } 43 GetDialogPosition(const gfx::Size & size)44 virtual gfx::Point GetDialogPosition(const gfx::Size& size) OVERRIDE { 45 return gfx::Point(); 46 } 47 GetMaximumDialogSize()48 virtual gfx::Size GetMaximumDialogSize() OVERRIDE { 49 return gfx::Size(); 50 } 51 AddObserver(ModalDialogHostObserver * observer)52 virtual void AddObserver( 53 ModalDialogHostObserver* observer) OVERRIDE {} RemoveObserver(ModalDialogHostObserver * observer)54 virtual void RemoveObserver( 55 ModalDialogHostObserver* observer) OVERRIDE {} 56 57 private: FindBrowser() const58 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 PrintPreviewTest()72PrintPreviewTest::PrintPreviewTest() {} ~PrintPreviewTest()73PrintPreviewTest::~PrintPreviewTest() {} 74 SetUp()75void 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 CreateBrowserWindow()86BrowserWindow* PrintPreviewTest::CreateBrowserWindow() { 87 return new PrintPreviewTestBrowserWindow; 88 } 89