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