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/test/ui/ui_test.h" 6 7 #include "base/strings/utf_string_conversions.h" 8 #include "chrome/browser/profiles/profile.h" 9 #include "chrome/browser/ui/browser.h" 10 #include "chrome/browser/ui/tabs/tab_strip_model.h" 11 #include "chrome/browser/ui/webui/constrained_web_dialog_ui.h" 12 #include "chrome/common/url_constants.h" 13 #include "chrome/test/base/in_process_browser_test.h" 14 #include "chrome/test/base/ui_test_utils.h" 15 #include "components/web_modal/web_contents_modal_dialog_manager.h" 16 #include "content/public/browser/web_contents.h" 17 #include "content/public/browser/web_contents_observer.h" 18 #include "ui/web_dialogs/test/test_web_dialog_delegate.h" 19 20 using content::WebContents; 21 using ui::WebDialogDelegate; 22 using web_modal::WebContentsModalDialogManager; 23 24 namespace { 25 26 class ConstrainedWebDialogBrowserTestObserver 27 : public content::WebContentsObserver { 28 public: 29 explicit ConstrainedWebDialogBrowserTestObserver(WebContents* contents) 30 : content::WebContentsObserver(contents), 31 contents_destroyed_(false) { 32 } 33 virtual ~ConstrainedWebDialogBrowserTestObserver() {} 34 35 bool contents_destroyed() { return contents_destroyed_; } 36 37 private: 38 virtual void WebContentsDestroyed(WebContents* tab) OVERRIDE { 39 contents_destroyed_ = true; 40 } 41 42 bool contents_destroyed_; 43 }; 44 45 } // namespace 46 47 class ConstrainedWebDialogBrowserTest : public InProcessBrowserTest { 48 public: 49 ConstrainedWebDialogBrowserTest() {} 50 51 protected: 52 bool IsShowingWebContentsModalDialog(WebContents* web_contents) const { 53 WebContentsModalDialogManager* web_contents_modal_dialog_manager = 54 WebContentsModalDialogManager::FromWebContents(web_contents); 55 return web_contents_modal_dialog_manager->IsShowingDialog(); 56 } 57 }; 58 59 // Tests that opening/closing the constrained window won't crash it. 60 IN_PROC_BROWSER_TEST_F(ConstrainedWebDialogBrowserTest, BasicTest) { 61 // The delegate deletes itself. 62 WebDialogDelegate* delegate = new ui::test::TestWebDialogDelegate( 63 GURL(chrome::kChromeUIConstrainedHTMLTestURL)); 64 WebContents* web_contents = 65 browser()->tab_strip_model()->GetActiveWebContents(); 66 ASSERT_TRUE(web_contents); 67 68 ConstrainedWebDialogDelegate* dialog_delegate = 69 CreateConstrainedWebDialog(browser()->profile(), 70 delegate, 71 NULL, 72 web_contents); 73 ASSERT_TRUE(dialog_delegate); 74 EXPECT_TRUE(dialog_delegate->GetNativeDialog()); 75 EXPECT_TRUE(IsShowingWebContentsModalDialog(web_contents)); 76 } 77 78 // Tests that ReleaseWebContentsOnDialogClose() works. 79 IN_PROC_BROWSER_TEST_F(ConstrainedWebDialogBrowserTest, 80 ReleaseWebContentsOnDialogClose) { 81 // The delegate deletes itself. 82 WebDialogDelegate* delegate = new ui::test::TestWebDialogDelegate( 83 GURL(chrome::kChromeUIConstrainedHTMLTestURL)); 84 WebContents* web_contents = 85 browser()->tab_strip_model()->GetActiveWebContents(); 86 ASSERT_TRUE(web_contents); 87 88 ConstrainedWebDialogDelegate* dialog_delegate = 89 CreateConstrainedWebDialog(browser()->profile(), 90 delegate, 91 NULL, 92 web_contents); 93 ASSERT_TRUE(dialog_delegate); 94 scoped_ptr<WebContents> new_tab(dialog_delegate->GetWebContents()); 95 ASSERT_TRUE(new_tab.get()); 96 ASSERT_TRUE(IsShowingWebContentsModalDialog(web_contents)); 97 98 ConstrainedWebDialogBrowserTestObserver observer(new_tab.get()); 99 dialog_delegate->ReleaseWebContentsOnDialogClose(); 100 dialog_delegate->OnDialogCloseFromWebUI(); 101 102 ASSERT_FALSE(observer.contents_destroyed()); 103 EXPECT_FALSE(IsShowingWebContentsModalDialog(web_contents)); 104 new_tab.reset(); 105 EXPECT_TRUE(observer.contents_destroyed()); 106 } 107